def test_outputs(): actual_tree = generate_improved_xmas_tree().strip("\n").split("\n") expected_tree = default_tree.strip("\n").split("\n") for i, j in zip(actual_tree, expected_tree): assert i.rstrip() == j.rstrip() actual_tree = generate_improved_xmas_tree(3).strip("\n").split("\n") expected_tree = smaller_tree.strip("\n").split("\n") for i, j in zip(actual_tree, expected_tree): assert i.rstrip() == j.rstrip()
def test_num_leafs_used(size, expected): assert generate_improved_xmas_tree(size).count("*") == expected
def test_trunk_used(size, expected): assert generate_improved_xmas_tree(size).count("|") == expected
def test_height_xmas_tree(size, expected): actual = len(generate_improved_xmas_tree(size).rstrip().splitlines()) assert actual == expected