def test_ngrams_for_tokens(self): self.assertEqual(text.ngrams_for_tokens(['one', 'two', 'three'], 1), [('one',), ('two',), ('three',)]) self.assertEqual(text.ngrams_for_tokens(['one', 'two', 'three'], 2), [('one', 'two'), ('two', 'three')]) self.assertEqual(text.ngrams_for_tokens(['one', 'two', 'three'], 3), [('one', 'two', 'three')])
def ngrams_for_text(self, s, n): tokens = text.title_tokens(s) return text.ngrams_for_tokens(tokens, n)