def test_row_drop_xor_reducer():
"""Testing the ColDrop pipeline stage."""
res_df = RowDrop([lambda x: x < 3]).apply(DF3)
assert 1 not in res_df.index
assert 2 not in res_df.index
assert 3 in res_df.index
res_df = RowDrop([lambda x: x < 3], reduce='xor').apply(DF3)
assert 1 in res_df.index
assert 2 not in res_df.index
assert 3 in res_df.index
def test_row_drop_columns():
"""Testing the ColDrop pipeline stage."""
res_df = RowDrop([lambda x: x < 2]).apply(DF2)
assert 1 not in res_df.index
assert 2 not in res_df.index
assert 3 in res_df.index
res_df = RowDrop([lambda x: x < 2], columns=['a', 'b']).apply(DF2)
assert 1 not in res_df.index
assert 2 in res_df.index
assert 3 in res_df.index
def test_row_droo_bad_columns():
"""Testing the ColDrop pipeline stage."""
with pytest.raises(FailedPreconditionError):
RowDrop([lambda x: x < 2], columns=['d']).apply(DF1)
def test_row_drop_bad_condition_in_list():
"""Testing the ColDrop pipeline stage."""
with pytest.raises(ValueError):
RowDrop(['bad'])
def test_row_drop_bad_reducer():
"""Testing the ColDrop pipeline stage."""
with pytest.raises(ValueError):
RowDrop([lambda x: x < 2], reduce='al')
def test_row_drop_verbose():
"""Testing the ColDrop pipeline stage."""
res_df = RowDrop([lambda x: x < 2]).apply(DF1, verbose=True)
assert 1 not in res_df.index
assert 2 in res_df.index
assert 3 in res_df.index