class TestPrefixTrees(unittest.TestCase):
    def setUp(self):
        self.treeD = PrefixTree()
        self.treeLL = PrefixTreeLinkedList()

    def tearDown(self):
        self.treeD = None
        self.treeLL = None

    def test_basic(self):
        """Basic test."""
        words = [randomString(6) for _ in range(999)]

        # make sure unique, just in case...
        uniq = []
        for _ in words:
            if _ not in self.treeD:
                uniq.append(_)
            self.treeD.add(_)
            self.treeLL.add(_)

        for _ in uniq:
            self.assertTrue(_ in self.treeD)
            self.assertTrue(_ in self.treeLL)

        random.shuffle(uniq)
        for _ in uniq:
            self.assertTrue(self.treeD.remove(_))
            self.assertTrue(self.treeLL.remove(_))

        self.assertEqual(0, len(self.treeD))
        self.assertEqual(0, len(self.treeLL))
class TestPrefixTrees(unittest.TestCase):
    
    def setUp(self):
        self.treeD = PrefixTree()
        self.treeLL = PrefixTreeLinkedList()
        
    def tearDown(self):
        self.treeD = None
        self.treeLL = None
        
    def test_basic(self):
        """Basic test."""
        words = [randomString(6) for _ in range(999)]

        # make sure unique, just in case...
        uniq = []
        for _ in words:
            if _ not in self.treeD:
                uniq.append(_)
            self.treeD.add(_)
            self.treeLL.add(_)

        for _ in uniq:
            self.assertTrue(_ in self.treeD)
            self.assertTrue(_ in self.treeLL)

        random.shuffle(uniq)
        for _ in uniq:
            self.assertTrue(self.treeD.remove(_))
            self.assertTrue(self.treeLL.remove(_))

        self.assertEqual(0, len(self.treeD))
        self.assertEqual(0, len(self.treeLL))
 def setUp(self):
     self.treeD = PrefixTree()
     self.treeLL = PrefixTreeLinkedList()
 def setUp(self):
     self.treeD = PrefixTree()
     self.treeLL = PrefixTreeLinkedList()