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, {})
