def test_method_double(self):
trie = graph.Graph()
trie.node('A')
trie.node('B')
trie.edge('f', 'A', 'B', 'g')
val = trie_methods.num_leaves(trie)
self.assertEqual(val, 1)
def test_method_complex(self):
trie = graph.Graph()
trie.node('A')
trie.node('B')
trie.edge('f', 'A', 'B', 'g')
trie.node('C')
trie.node('D')
trie.node('E')
trie.edge('g', 'B', 'C', 't')
trie.edge('h', 'B', 'D', 'a')
trie.edge('i', 'A', 'E', 'c')
val = trie_methods.num_leaves(trie)
self.assertEqual(val, 3)
def test_method_single(self):
trie = graph.Graph()
trie.node('A')
val = trie_methods.num_leaves(trie)
self.assertEqual(val, 1)
