def test_insert_unique(self): """insert_unique should insert only unique words.""" data = self.data t = Trie() for (label, seq) in data.items(): t._insert_unique(seq, label) self.assertEqual(t.find("ab"), []) self.assertEqual(t.find("cacgchagc"), []) self.assertEqual(t.find("abababa"), ["1"])
def test_insert_find(self): """An added key should be found by find.""" data = self.data t = Trie() for (label, seq) in data.items(): t.insert(seq, label) for (label, seq) in data.items(): self.assertEqual(label in t.find(seq), True) self.assertEqual(t.find("cacgchagc"), []) self.assertEqual(t.find("abababa"), ["1", "6"])
def test_insert_unique(self): """insert_unique should insert only unique words.""" data = self.data t = Trie() for (label, seq) in data.iteritems(): t._insert_unique(seq, label) self.assertEqual(t.find("ab"), []) self.assertEqual(t.find("cacgchagc"), []) self.assertEqual(t.find("abababa"), ["1"])
def test_insert_find(self): """An added key should be found by find.""" data = self.data t = Trie() for (label, seq) in data.iteritems(): t.insert(seq, label) for (label, seq) in data.iteritems(): self.assertEqual(label in t.find(seq), True) self.assertEqual(t.find("cacgchagc"), []) self.assertEqual(t.find("abababa"), ["1","6"])
def test_init(self): """Trie init should create an empty trie.""" t = Trie() self.assertEqual(t.root.labels, []) self.assertEqual(t.root.children, {})