From c766b8b0e05eda764fd2ebe535be32d2effa7f91 Mon Sep 17 00:00:00 2001 From: Filippo Utro <137520996+f-utro@users.noreply.github.com> Date: Wed, 15 Nov 2023 11:35:52 -0500 Subject: [PATCH] Update test_ReVeaL.py --- tests/test_ReVeaL.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_ReVeaL.py b/tests/test_ReVeaL.py index 5470bf7..a8c09ca 100644 --- a/tests/test_ReVeaL.py +++ b/tests/test_ReVeaL.py @@ -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)