Python generate_improved_xmas_tree Examples

Example #1

File: test_improved_xmas_tree.py Project: alehpineda/bitesofpy

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

Example #2

File: test_improved_xmas_tree.py Project: alehpineda/bitesofpy

def test_num_leafs_used(size, expected):
    assert generate_improved_xmas_tree(size).count("*") == expected

Example #3

File: test_improved_xmas_tree.py Project: alehpineda/bitesofpy

def test_trunk_used(size, expected):
    assert generate_improved_xmas_tree(size).count("|") == expected

Example #4

File: test_improved_xmas_tree.py Project: alehpineda/bitesofpy

def test_height_xmas_tree(size, expected):
    actual = len(generate_improved_xmas_tree(size).rstrip().splitlines())
    assert actual == expected