def test_check_n_categories_ser():
# verifies that check_n_categories generates the expected output
assert good_result.loc[1, 'n_categories'] == cf.check_n_categories(good['g2']).values
def test_check_n_categories_no_dropna_ser():
# verifies that check_n_categories generates the expected output with nulls
assert nan_result.loc[0, 'n_categories'] == cf.check_n_categories(nan['g1'], dropna=False).values
def test_check_n_categories_df():
# verifies that check_n_categories generates the expected output
assert good_result.equals(cf.check_n_categories(good))
def test_check_n_categories_no_dropna_df():
# verifies that check_n_categories generates the expected output with nulls
assert nan_result.equals(cf.check_n_categories(nan, dropna=False))