class TestBST(unittest.TestCase):
"""Runs the simple examples from the activate state recipe."""
names = "bob joe Jane jack Mary sue Ed Zoey ann".split()
def setUp(self):
self.bst = BinarySearchTree(self.names, str.upper)
def test_insert(self):
self.assertEquals(self.bst.minimum(), "ann")
def test_minimum(self):
self.assertEquals(self.bst.maximum(), "Zoey")
def test_values(self):
expected_list = ["ann", "bob", "Ed", "jack", "Jane", "joe", "Mary", "sue", "Zoey"]
for expected, result in zip(expected_list, self.bst.values()):
self.assertEquals(expected, result)
def test_values_reversed(self):
expected_list = ["Zoey", "sue", "Mary", "joe", "Jane", "jack", "Ed", "bob", "ann"]
for expected, result in zip(expected_list, self.bst.values(True)):
self.assertEquals(expected, result)