def test_key_iterator(self):
a = AVLTree()
values = [
(1, 'a'),
(2, 'b'),
(3, 'c'),
]
for k, v in values:
a[k] = v
self.assert_list(a.keys(), [k for k, _ in values])
def test_traversal_on_balanced(self):
a = AVLTree()
a[2] = "a"
a[1] = "b"
a[3] = "c"
self.assertListEqual([1, 2, 3], list(a.keys()))