test_prefix_tree.py 文件源码

def test_tree(self):
        prefix_tree = PrefixTree()
        values = ['amy', 'ann', 'anne', 'emma', 'rob', 'roger', 'anna']

        # Test setter
        for value in values:
            prefix_tree[value] = value

        # Test getter
        result = prefix_tree['ann']
        self.assertTrue(result)
        self.assertTrue(inspect.isgenerator(result))
        # expect 'ann', 'anne' and 'anna'
        self.assertEqual(len(list(result)), 3)

        # Test containment
        self.assertTrue('amy' in prefix_tree)
        self.assertFalse('am' in prefix_tree)