def testUpdate(self):
t = main.Trie()
t.add("cat")
t.add("cats")
t.add("cattle")
for i in range(10):
t.update("cat")
self.assertEquals(t.count("cat"), 10)
for i in range(3):
t.update("cats")
self.assertEquals(t.count("cat"), 10)
self.assertEquals(t.count("cats"), 3)
for i in range(15):
t.update("cattle")
self.assertEquals(t.count("cat"), 10)
self.assertEquals(t.count("cats"), 3)
self.assertEquals(t.count("cattle"), 15)
def testAddDictionary(self):
t = main.Trie()
infile = open(dictfile)
[t.add(w.strip()) for w in infile]
[self.assertTrue(w.strip() in t) for w in infile]
def testAddSingle(self):
t = main.Trie()
t.add("test")
self.assertTrue("test" in t)