Skip to content

Commit

Permalink
Update test_ReVeaL.py
Browse files Browse the repository at this point in the history
  • Loading branch information
f-utro authored Nov 15, 2023
1 parent 0ea5847 commit c766b8b
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions tests/test_ReVeaL.py
Original file line number Diff line number Diff line change
@@ -60,7 +60,7 @@ def test_compute_mutational_load():
expected_df['count'] = [2, 0, 1, 0, 0, 1]
expected_df['samples'] = ['s1', 's1', 's2', 's2', 's4', 's4']

assert_frame_equal(expected_df.pivot('samples', 'window', 'count').fillna(0), returned_df)
assert_frame_equal(expected_df.pivot(index='samples', columns='window', values='count').fillna(0), returned_df)

returned_df = ReVeaL.compute_mutational_load(sample_data, samples.tolist(), regions_size, 'C', window_size=2)

@@ -69,7 +69,7 @@ def test_compute_mutational_load():
expected_df['count'] = [1, 0, 0, 1, 1, 0, 1, 0, 0, 0, 0, 1, 0, 0, 0]
expected_df['samples'] = ['s1','s2','s4', 's1','s2','s4', 's1','s2','s4', 's1','s2','s4', 's1','s2','s4']

assert_frame_equal(expected_df.pivot(index='samples', columns='window', value='count').fillna(0), returned_df)
assert_frame_equal(expected_df.pivot(index='samples', columns='window', values='count').fillna(0), returned_df)



0 comments on commit c766b8b

Please sign in to comment.