bites 146

Author: [email protected]

.gitignore

@@ -62,3 +62,4 @@
 /163/README.md
 /Pipfile
 /Pipfile.lock
+/146/README.md

146/rhombus.py

@@ -0,0 +1,30 @@
+STAR = '*'
+
+
+def gen_rhombus(width):
+    """Create a generator that yields the rows of a rhombus row
+       by row. So if width = 5 it should generate the following
+       rows one by one:
+
+       gen = gen_rhombus(5)
+       for row in gen:
+           print(row)
+
+        output:
+          *
+         ***
+        *****
+         ***
+          *
+    """
+
+    for rowz in range(1, (width * 2) + 1, 2):
+        if rowz-width <= 0:
+            yield '{:^{width}}'.format(STAR * rowz, width=width)
+        else:
+            multiplierOf4 = (rowz-width)//2
+            yield '{:^{width}}'.format(STAR * (rowz-4*multiplierOf4), width=width)
+
+gen = gen_rhombus(5)
+for row in gen:
+    print(row)

146/test_rhombus.py

@@ -0,0 +1,37 @@
+from rhombus import gen_rhombus
+
+
+def test_rhombus_width3():
+    # recommended: actual before expected
+    # https://twitter.com/brianokken/status/1063337328553295876
+    actual = list(gen_rhombus(3))
+    expected = [' * ', '***', ' * ']
+    assert actual == expected
+
+
+def test_rhombus_width5():
+    actual = list(gen_rhombus(5))
+    expected = ['  *  ', ' *** ', '*****',
+                ' *** ', '  *  ']
+    assert actual == expected
+
+
+def test_rhombus_width11():
+    """print('\n'.join(expected)) would give (ignore indents):
+         *
+        ***
+       *****
+      *******
+     *********
+    ***********
+     *********
+      *******
+       *****
+        ***
+         *
+    """
+    actual = list(gen_rhombus(11))
+    expected = ['     *     ', '    ***    ', '   *****   ',
+                '  *******  ', ' ********* ', '***********', ' ********* ',
+                '  *******  ', '   *****   ', '    ***    ', '     *     ']
+    assert actual == expected