def test_string_matching(self, example_df):
""" unix style matching should also work. """
# test *
out = filter_df(example_df, first_name="J*")
assert {"Jason", "Jake"} == set(example_df[out].first_name)
# test ???
out = filter_df(example_df, first_name="J???")
assert {"Jake"} == set(example_df[out].first_name)
def test_bad_parameter_raises(self, example_df):
""" ensure passing a parameter that doesn't have a column raises. """
with pytest.raises(ValueError):
filter_df(example_df, bad_column=2)
def test_non_str_sequence(self, example_df):
""" ensure sequences still work for isin style comparisons. """
out = filter_df(example_df, age={42, 52})
assert out[:2].all()
assert not out[2:].any()
def test_non_str_single_arg(self, example_df):
""" test that filter index can be used on Non-nslc columns. """
# test non strings
out = filter_df(example_df, age=42)
assert out[0]
assert not out[1:].any()
def test_str_sequence(self, example_df):
""" Test str sequences find values in sequence. """
out = filter_df(example_df, last_name={"Miller", "Jacobson"})
assert out[:2].all()
assert not out[2:].any()
def test_string_basic(self, example_df):
""" test that specifying a string with no matching works. """
out = filter_df(example_df, first_name="Jason")
assert out[0]
assert not out[1:].any()