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)