class DottedKeyTests(unittest.TestCase):
def setUp(self):
self.trie = Trie(keyFunction = KEY_DOTTED)
self.keys = ['com.example', 'com.baz', 'com.example.sub', 'org.example']
for key in self.keys:
self.trie.add(key, 1)
def test_has(self):
for key in self.keys:
self.assertTrue(self.trie.has(key), "Trie::has")
def test_paths(self):
for path in self.trie.paths():
self.assertTrue(path in self.keys, "Trie::paths")
paths = list(self.trie.paths())
self.assertTrue(len(paths) == 4, "Trie::paths length")
def test_allpathvalues(self):
values = self.trie.getAllPathValues('com.example.sub')
self.assertTrue(len(values) == 2, "Trie::getAllPathValues")
def test_subpaths(self):
subpaths = self.trie.getSubPaths('com.example.sub')
self.assertTrue(len(subpaths) == 2, "Trie::getSubPaths")
self.assertTrue(subpaths[0] == 'com.example', "Trie::getSubPaths")
self.assertTrue(subpaths[1] == 'com.example.sub', "Trie::getSubPaths")
class DottedKeyTests(unittest.TestCase):
def setUp(self):
self.trie = Trie(keyFunction=KEY_DOTTED)
self.keys = [
'com.example', 'com.baz', 'com.example.sub', 'org.example'
]
for key in self.keys:
self.trie.add(key, 1)
def test_has(self):
for key in self.keys:
self.assertTrue(self.trie.has(key), "Trie::has")
def test_paths(self):
for path in self.trie.paths():
self.assertTrue(path in self.keys, "Trie::paths")
paths = list(self.trie.paths())
self.assertTrue(len(paths) == 4, "Trie::paths length")
def test_allpathvalues(self):
values = self.trie.getAllPathValues('com.example.sub')
self.assertTrue(len(values) == 2, "Trie::getAllPathValues")
def test_subpaths(self):
subpaths = self.trie.getSubPaths('com.example.sub')
self.assertTrue(len(subpaths) == 2, "Trie::getSubPaths")
self.assertTrue(subpaths[0] == 'com.example', "Trie::getSubPaths")
self.assertTrue(subpaths[1] == 'com.example.sub', "Trie::getSubPaths")
