def test_count_1gram(self):
ngram = NGram(1, self.sents)
counts = {
(): 12,
('el',): 1,
('gato',): 1,
('come',): 2,
('pescado',): 1,
('.',): 2,
('</s>',): 2,
('la',): 1,
('gata',): 1,
('salmón',): 1,
}
for gram, c in counts.items():
self.assertEqual(ngram.count(gram), c)
def test_count_1gram(self):
ngram = NGram(1, self.sents)
counts = {
(): 12,
('el', ): 1,
('gato', ): 1,
('come', ): 2,
('pescado', ): 1,
('.', ): 2,
('</s>', ): 2,
('la', ): 1,
('gata', ): 1,
('salmón', ): 1,
}
for gram, c in counts.items():
self.assertEqual(ngram.count(gram), c)
