Merge pull request #14 from BiomedSciAI/f-utro-patch-2

Update test_ReVeaL.py
BiomedSciAI · Nov 6, 2023 · d79bb1f · d79bb1f
2 parents e76352d + 9351bc5
commit d79bb1f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_ReVeaL.py b/tests/test_ReVeaL.py
@@ -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('samples', 'window', 'count').fillna(0), returned_df)
+    assert_frame_equal(expected_df.pivot(index='samples', columns='window', value='count').fillna(0), returned_df)