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()