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

20 lines
474 B
Python

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()