def test_insert_keeps_correct_count(self):
trie = Trie()
trie.insert(1, 0, 'foo')
trie.insert(2, 0, 'bar')
trie.insert(3, 0, 'baz')
trie.insert(None, 0, 'bar')
trie.insert(4, 0, None)
trie.insert(5, None, 'biz')
self.assertEqual(trie.count, 3)
def test_insert(self):
trie = Trie()
trie.insert(1, 0, 'foo')
trie.insert(2, 0, 'bar')
trie.insert(3, 0, 'baz')
trie.insert(None, 0, 'bar')
trie.insert(4, -1, 'booz')
trie.insert(5, None, 'baj')
words = trie.root.flatten_words()
expected = [(1, 0), (2, 0), (3, 0)]
for e in expected:
self.assertTrue(e in words)
self.assertFalse((None, 0) in words)
self.assertFalse((4, -1) in words)
self.assertFalse((5, None) in words)
