def test_ngram(self):
ctx = Context(2, 2)
ctx.add(1)
ctx.add(2)
self.assertTrue(ctx.kgram == [1, 2])
ctx.reset()
self.assertTrue(ctx.kgram == [2])
ctx.add(2)
self.assertTrue(ctx.kgram == [2])
ctx.add(3)
self.assertTrue(ctx.kgram == [2, 3])
self.assertFalse(ctx.add(4))
