def test_unigram(self):
n = NGram(0)
n.add('after')
n.next_word() == 'after'
def test_trigram(self):
n = NGram(2)
n.add('after', ('before', 'other'))
assert n.next_word(('before', 'other')) == 'after'
def test_bigram(self):
n = NGram(1)
n.add('after', ('before'))
assert n.next_word(('before')) == 'after'
def test_unigram(self):
n = NGram(0)
n.add('after')
n.next_word() == 'after'
def test_trigram(self):
n = NGram(2)
n.add('after', ('before', 'other'))
assert n.next_word(('before', 'other')) == 'after'
def test_bigram(self):
n = NGram(1)
n.add('after', ('before'))
assert n.next_word(('before')) == 'after'