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