def test_normal(self):
p1 = ['ag', 'ct', 'tt']
p2 = ['tt', 'ct', 'ag']
tr1 = trie_builder.PrefixTrieConstruction(p1)
tr2 = trie_builder.PrefixTrieConstruction(p2)
val = trie_methods.trie_equivalence(tr1, tr2)
self.assertTrue(val)
def test_normal_2(self):
patterns = ["t", "a"]
trie = graph.Graph()
trie.node("A")
trie.node("B")
trie.node("C")
trie.edge("f", "A", "B", "t")
trie.edge("g", "A", "C", "a")
expected = trie_builder.PrefixTrieConstruction(patterns)
val = trie_methods.trie_equivalence(trie, expected)
self.assertTrue(val)
def test_same(self):
pattern = "cg"
trie = graph.Graph()
trie.node("A")
trie.node("B")
trie.node("C")
trie.edge("f", "A", "B", "c")
trie.edge("g", "B", "C", "g")
expected = trie_builder.incorporate(pattern, trie)
val = trie_methods.trie_equivalence(expected, trie)
self.assertTrue(val)
def test_complex(self):
patterns = ["cga", "cg", "gta"]
trie = graph.Graph()
trie.node("A")
trie.node("B")
trie.node("C")
trie.node("D")
trie.node("E")
trie.node("F")
trie.node("G")
trie.edge("f", "A", "B", "c")
trie.edge("g", "B", "C", "g")
trie.edge("h", "C", "D", "a")
trie.edge("i", "A", "E", "g")
trie.edge("j", "E", "F", "t")
trie.edge("k", "F", "G", "a")
expected = trie_builder.PrefixTrieConstruction(patterns)
val = trie_methods.trie_equivalence(trie, expected)
self.assertTrue(val)