def test_search(self): tree = TernarySearchTree() tree.add('An aapple fell off the tree') tree.add('An aardvark fell off the tree') (node, count) = tree.search('An aa') self.assertEqual(node.token, 'a') self.assertEqual(count, 2) (node, count) = tree.search('An aardvark') self.assertEqual(node.token, 'k') self.assertEqual(count, 1) (node, count) = tree.search('An aapple fell off the tree') self.assertEqual(node.token, 'e') self.assertEqual(count, 2)
def test_search(self): tree = TernarySearchTree() tree.add("An aapple fell off the tree") tree.add("An aardvark fell off the tree") (node, count) = tree.search("An aa") self.assertEqual(node.token, "a") self.assertEqual(count, 2) (node, count) = tree.search("An aardvark") self.assertEqual(node.token, "k") self.assertEqual(count, 1) (node, count) = tree.search("An aapple fell off the tree") self.assertEqual(node.token, "e") self.assertEqual(count, 2)
def test_search_not_in_tree(self): tree = TernarySearchTree() tree.add('a') node, count = tree.search('b') self.assertEqual(node, None) self.assertEqual(count, 0)
def test_search_repeated_token(self): tree = TernarySearchTree() tree.add('aaaa') node, count = tree.search('aaa') self.assertEqual(node.token, 'a') self.assertEqual(count, 3)
def test_search_single_token(self): tree = TernarySearchTree() tree.add('a') node, count = tree.search('a') self.assertEqual(node.token, 'a') self.assertEqual(count, 1)
def test_search_not_in_tree(self): tree = TernarySearchTree() tree.add("a") node, count = tree.search("b") self.assertEqual(node, None) self.assertEqual(count, 0)
def test_search_repeated_token(self): tree = TernarySearchTree() tree.add("aaaa") node, count = tree.search("aaa") self.assertEqual(node.token, "a") self.assertEqual(count, 3)
def test_search_single_token(self): tree = TernarySearchTree() tree.add("a") node, count = tree.search("a") self.assertEqual(node.token, "a") self.assertEqual(count, 1)