def test_n_grams_ignore_case_to_lower(self): """Should return all case insensitive ngrams in a string""" test_series = pd.Series(pd.Series(['aa'])) # Explicit ignore case sg = StringGrouper(test_series, ignore_case=True) expected_result = ['mcd', 'cdo', 'don', 'ona', 'nal', 'ald', 'lds'] self.assertListEqual(expected_result, sg.n_grams('McDonalds'))
def test_n_grams_ignore_case_to_lower_with_defaults(self): """Should return all case insensitive ngrams in a string""" test_series = pd.Series(pd.Series(['aa'])) # Implicit default case (i.e. default behaviour) sg = StringGrouper(test_series) expected_result = ['mcd', 'cdo', 'don', 'ona', 'nal', 'ald', 'lds'] self.assertListEqual(expected_result, sg.n_grams('McDonalds'))
def test_n_grams_case_unchanged(self): """Should return all ngrams in a string with case""" test_series = pd.Series(pd.Series(['aa'])) # Explicit do not ignore case sg = StringGrouper(test_series, ignore_case=False) expected_result = ['McD', 'cDo', 'Don', 'ona', 'nal', 'ald', 'lds'] self.assertListEqual(expected_result, sg.n_grams('McDonalds'))
def test_n_grams(self): """Should return all ngrams in a string""" test_series = pd.Series(pd.Series(['aa'])) sg = StringGrouper(test_series) expected_result = ['McD', 'cDo', 'Don', 'ona', 'nal', 'ald', 'lds'] self.assertListEqual(expected_result, sg.n_grams('McDonalds'))