def test_cohort_filter6():
filtered_data = cohort_filter(data, "Science", exclusive_search=False)
assert (filtered_data.shape[0] == 4)
def test_cohort_filter7():
filtered_data = cohort_filter(data,
"Science",
unit_list=unit_list,
exclusive_search=False)
assert (filtered_data.shape[0] == 3)
def test_cohort_filter5():
filtered_data = cohort_filter(data, "Bachelor of Science")
assert (filtered_data.shape[0] == 1)
def test_cohort_filter4():
filtered_data = cohort_filter(data, "Engineering", unit_list=unit_list)
assert (filtered_data.shape[0] == 8)
def test_cohort_filter3():
filtered_data = cohort_filter(data, "Engineering")
assert (filtered_data.shape[0] == 9)
assert (isinstance(filtered_data.outcome_date[0], datetime.date))
def test_cohort_filter2():
filtered_data = cohort_filter(data, "Engineering")
assert (isinstance(filtered_data, pd.DataFrame))
def test_cohort_filter1():
with pytest.raises(Exception):
filtered_data = cohort_filter()