interactive-coding-challenges/online_judges/utopian_tree/test_utopian_tree.py

from nose.tools import assert_equal


class TestUtopianTree(object):

    def test_utopian_tree(self):
        solution = Solution()
        assert_equal(solution.calc_utopian_tree_height(0), 1)
        assert_equal(solution.calc_utopian_tree_height(1), 2)
        assert_equal(solution.calc_utopian_tree_height(4), 7)
        print('Success: test_utopian_tree')


def main():
    test = TestUtopianTree()
    test.test_utopian_tree()


if __name__ == '__main__':
    main()