def test_q_filter_all_with_two_predicates(all_trials, trial_one):
expected = [trial_one]
results = q_filter_all(all_trials,
q_item("survived").is_true(),
q_item("date").contains("2003"))
assert list(results) == expected
def test_q_filter_all_with_three_predicates(all_trials, trial_three):
expected = [trial_three]
results = q_filter_all(all_trials,
q_item("survived").is_false(),
q_item("drug") == "tolterodine tartrate",
q_item("dose_mg") > 350)
assert list(results) == expected
def test_not_with_all(all_trials, trial_one, trial_two, trial_four,
trial_five):
expected = [trial_one, trial_two, trial_four, trial_five]
results = q_filter_all(
all_trials,
q_not(q_all(q_item("date").contains("2003"),
q_item("dose_mg") > 400)))
assert list(results) == expected
def test_not_with_any_and_two_predicates(all_trials, trial_four, trial_five):
expected = [trial_four, trial_five]
results = q_filter_all(
all_trials,
q_not(
q_any(
q_item("survived").is_true(),
q_item("date").contains("2003"))))
assert list(results) == expected
def test_k_filter_all_predicate_arg_with_keyword_arg(users, user_two):
expected = [user_two]
def id_is_even(item):
return item["id"] % 2 == 0
results = q_filter_all(users,
id_is_even,
q_item("gender") == "Male")
assert list(results) == expected
def test_q_filter_not_any_prediocate_arg_with_keyword_arg(
users, user_three, user_five
):
expected = [user_three, user_five]
def id_is_even(item):
return item["id"] % 2 == 0
results = q_filter_not_any(users,
q_item("gender") == "Female",
id_is_even)
assert list(results) == expected
def test_filter_any_prediocate_arg_with_keyword_arg(
users, user_one, user_two, user_four
):
expected = [user_one, user_two, user_four]
def id_is_even(item):
return item["id"] % 2 == 0
results = q_filter_any(users,
id_is_even,
q_item("gender") == "Female")
assert list(results) == expected
def test_not(all_trials, trial_one, trial_two):
expected = [trial_one, trial_two]
results = q_filter_all(all_trials, q_not(q_item("survived").is_false()))
assert list(results) == expected
