interactive-coding-challenges/graphs_trees/tree_height/test_height.py

from nose.tools import assert_equal


class TestHeight(object):

    def test_height(self):
        bst = BstHeight(Node(5))
        assert_equal(bst.height(bst.root), 1)
        bst.insert(2)
        bst.insert(8)
        bst.insert(1)
        bst.insert(3)
        assert_equal(bst.height(bst.root), 3)

        print('Success: test_height')


def main():
    test = TestHeight()
    test.test_height()


if __name__ == '__main__':
    main()
Added tree height challenge. 2015-08-02 22:53:35 +08:00			`from nose.tools import assert_equal`


			`class TestHeight(object):`

			`def test_height(self):`
Move tree height to a class 2016-08-14 20:29:25 +08:00			`bst = BstHeight(Node(5))`
			`assert_equal(bst.height(bst.root), 1)`
			`bst.insert(2)`
			`bst.insert(8)`
			`bst.insert(1)`
			`bst.insert(3)`
			`assert_equal(bst.height(bst.root), 3)`
Added tree height challenge. 2015-08-02 22:53:35 +08:00
			`print('Success: test_height')`


			`def main():`
			`test = TestHeight()`
			`test.test_height()`


			`if __name__ == '__main__':`
			`main()`