def test_outputs():
actual_tree = generate_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_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_stars_used():
assert generate_xmas_tree(3).count('*') == 9
assert generate_xmas_tree(5).count('*') == 25
assert generate_xmas_tree(20).count('*') == 400
def test_height_xmas_tree():
assert len(generate_xmas_tree().split('\n')) == 10 # default arg
assert len(generate_xmas_tree(5).split('\n')) == 5
assert len(generate_xmas_tree(20).split('\n')) == 20