Skip to content

Commit 0b7af81

Browse files
committed
add group_swapped column to test
1 parent 93467a6 commit 0b7af81

1 file changed

Lines changed: 4 additions & 3 deletions

File tree

tests/test_column_prep.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -51,13 +51,14 @@ def test_column_prep():
5151
"coverage_p1": [0.5, 0.5, 0.5],
5252
"coverage_p2": [0.5, 0.5, 0.5],
5353
"protein_score_p1": [0.1, 0.6, 1.1],
54-
"protein_score_p2": [0.1, 0.6, 1.1]
54+
"protein_score_p2": [0.1, 0.6, 1.1],
55+
"group_swapped": [True, False, False],
5556
})
5657

5758
df_res = prepare_columns(df)
5859
assert_frame_equal(
59-
df_res.sort(df_expect.columns),
60-
df_expect.select(df_res.columns).sort(df_expect.columns)
60+
df_res.select(df_expect.columns).sort(df_expect.columns),
61+
df_expect.sort(df_expect.columns)
6162
)
6263
# Check that the resulting state is constant
6364
df_res = prepare_columns(df)

0 commit comments

Comments
 (0)