bites 119

Author: [email protected]

.gitignore

@@ -48,3 +48,4 @@
 /203/README.md
 /191/README.md
 /113/README.md
+/119/README.md

119/test_xmas.py

@@ -0,0 +1,43 @@
+from xmas import generate_xmas_tree
+
+default_tree = """
+         *
+        ***
+       *****
+      *******
+     *********
+    ***********
+   *************
+  ***************
+ *****************
+*******************
+"""
+smaller_tree = """
+  *
+ ***
+*****
+"""
+
+
+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
+
+
+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_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()

119/xmas.py

@@ -0,0 +1,15 @@
+def generate_xmas_tree(rows=10):
+    """Generate a xmas tree of stars (*) for given rows (default 10).
+       Each row has row_number*2-1 stars, simple example: for rows=3 the
+       output would be like this (ignore docstring's indentation):
+         *
+        ***
+       *****"""
+    tree = list()
+    for row in range(rows):
+        leaves = ((row+1)*2-1) * "*"
+        tree.append(leaves.center((rows)*2))
+    tree = "\n".join(tree)
+    return tree
+
+print(generate_xmas_tree())