Problem 11 solution

lukemcguire · lukemcguire · commit ff5dfce62295 · 2024-06-29T14:47:58.000-07:00
diff --git a/11_bottles_of_beer/bottles.py b/11_bottles_of_beer/bottles.py
@@ -0,0 +1,103 @@
+#!/usr/bin/env python3
+"""
+Author : Luke McGuire <luke.mcguire@gmail.com>
+Date   : 2024-06-28
+Purpose: Bottles of beer song
+"""
+
+import argparse
+
+
+# --------------------------------------------------
+def positive_int(string):
+    """check for positive integer"""
+    try:
+        value = int(string)
+    except ValueError as e:
+        raise argparse.ArgumentTypeError(f"invalid int value: '{string}'") from e
+    if value < 1:
+        raise argparse.ArgumentTypeError(f'"{value}" must be greater than 0')
+    return value
+
+
+# --------------------------------------------------
+
+
+def get_args():
+    """Get command-line arguments"""
+
+    parser = argparse.ArgumentParser(
+        description="Bottles of beer song",
+        formatter_class=argparse.ArgumentDefaultsHelpFormatter,
+    )
+
+    parser.add_argument(
+        "-n",
+        "--num",
+        help="How many bottles",
+        metavar="int",
+        type=positive_int,
+        default=10,
+    )
+
+    return parser.parse_args()
+
+
+# --------------------------------------------------
+def verse(bottle: int) -> str:
+    """Sing a verse"""
+
+    next_bottle = bottle - 1
+    s1 = "" if bottle == 1 else "s"
+    s2 = "" if next_bottle == 1 else "s"
+
+    lyrics = [
+        f"{bottle} bottle{s1} of beer on the wall,",
+        f"{bottle} bottle{s1} of beer,",
+        "Take one down, pass it around,",
+        f"{next_bottle if next_bottle else 'No more'} bottle{s2} of beer on the wall!",
+    ]
+    return "\n".join(lyrics)
+
+
+# --------------------------------------------------
+def test_verse_1():
+    """Test verse for 1 bottle"""
+
+    one = verse(1)
+    assert one == "\n".join(
+        [
+            "1 bottle of beer on the wall,",
+            "1 bottle of beer,",
+            "Take one down, pass it around,",
+            "No more bottles of beer on the wall!",
+        ]
+    )
+
+
+def test_verse_2():
+    """Test verse for 2 bottles"""
+
+    two = verse(2)
+    assert two == "\n".join(
+        [
+            "2 bottles of beer on the wall,",
+            "2 bottles of beer,",
+            "Take one down, pass it around,",
+            "1 bottle of beer on the wall!",
+        ]
+    )
+
+
+# --------------------------------------------------
+def main():
+    """Make a jazz noise here"""
+
+    args = get_args()
+    # song = map(verse, range(args.num, 0, -1))
+    print("\n\n".join(song))
+
+
+# --------------------------------------------------
+if __name__ == "__main__":
+    main()
diff --git a/11_bottles_of_beer/test.py b/11_bottles_of_beer/test.py
@@ -8,22 +8,22 @@
 import string
 from subprocess import getstatusoutput
 
-prg = './bottles.py'
+PRG = "./bottles.py"
 
 
 # --------------------------------------------------
 def test_exists():
     """exists"""
 
-    assert os.path.isfile(prg)
+    assert os.path.isfile(PRG)
 
 
 # --------------------------------------------------
 def test_usage():
     """usage"""
 
-    for flag in ['-h', '--help']:
-        rv, out = getstatusoutput(f'{prg} {flag}')
+    for flag in ["-h", "--help"]:
+        rv, out = getstatusoutput(f"python {PRG} {flag}")
         assert rv == 0
         assert re.match("usage", out, re.IGNORECASE)
 
@@ -33,17 +33,17 @@ def test_bad_int():
     """Bad integer value"""
 
     bad = random.randint(-10, 0)
-    rv, out = getstatusoutput(f'{prg} -n {bad}')
+    rv, out = getstatusoutput(f"python {PRG} -n {bad}")
     assert rv != 0
-    assert re.search(f'--num "{bad}" must be greater than 0', out)
+    assert re.search(f'--num: "{bad}" must be greater than 0', out)
 
 
 # --------------------------------------------------
 def test_float():
     """float value"""
 
     bad = round(random.random() * 10, 2)
-    rv, out = getstatusoutput(f'{prg} --num {bad}')
+    rv, out = getstatusoutput(f"python {PRG} --num {bad}")
     assert rv != 0
     assert re.search(f"invalid int value: '{bad}'", out)
 
@@ -53,7 +53,7 @@ def test_str():
     """str value"""
 
     bad = random_string()
-    rv, out = getstatusoutput(f'{prg} -n {bad}')
+    rv, out = getstatusoutput(f"python {PRG} -n {bad}")
     assert rv != 0
     assert re.search(f"invalid int value: '{bad}'", out)
 
@@ -62,12 +62,14 @@ def test_str():
 def test_one():
     """One bottle of beer"""
 
-    expected = ('1 bottle of beer on the wall,\n'
-                '1 bottle of beer,\n'
-                'Take one down, pass it around,\n'
-                'No more bottles of beer on the wall!')
+    expected = (
+        "1 bottle of beer on the wall,\n"
+        "1 bottle of beer,\n"
+        "Take one down, pass it around,\n"
+        "No more bottles of beer on the wall!"
+    )
 
-    rv, out = getstatusoutput(f'{prg} --num 1')
+    rv, out = getstatusoutput(f"python {PRG} --num 1")
     assert rv == 0
     assert out == expected
 
@@ -76,16 +78,18 @@ def test_one():
 def test_two():
     """Two bottles of beer"""
 
-    expected = ('2 bottles of beer on the wall,\n'
-                '2 bottles of beer,\n'
-                'Take one down, pass it around,\n'
-                '1 bottle of beer on the wall!\n\n'
-                '1 bottle of beer on the wall,\n'
-                '1 bottle of beer,\n'
-                'Take one down, pass it around,\n'
-                'No more bottles of beer on the wall!')
-
-    rv, out = getstatusoutput(f'{prg} -n 2')
+    expected = (
+        "2 bottles of beer on the wall,\n"
+        "2 bottles of beer,\n"
+        "Take one down, pass it around,\n"
+        "1 bottle of beer on the wall!\n\n"
+        "1 bottle of beer on the wall,\n"
+        "1 bottle of beer,\n"
+        "Take one down, pass it around,\n"
+        "No more bottles of beer on the wall!"
+    )
+
+    rv, out = getstatusoutput(f"python {PRG} -n 2")
     assert rv == 0
     assert out == expected
 
@@ -94,21 +98,19 @@ def test_two():
 def test_random():
     """Random number"""
 
-    sums = dict(
-        map(lambda x: x.split('\t'),
-            open('sums.txt').read().splitlines()))
+    sums = dict(map(lambda x: x.split("\t"), open("sums.txt").read().splitlines()))
 
     for n in random.choices(list(sums.keys()), k=10):
-        flag = '-n' if random.choice([0, 1]) == 1 else '--num'
-        rv, out = getstatusoutput(f'{prg} {flag} {n}')
-        out += '\n'  # because the last newline is removed
+        flag = "-n" if random.choice([0, 1]) == 1 else "--num"
+        rv, out = getstatusoutput(f"python {PRG} {flag} {n}")
+        out += "\n"  # because the last newline is removed
         assert rv == 0
-        assert hashlib.md5(out.encode('utf-8')).hexdigest() == sums[n]
+        assert hashlib.md5(out.encode("utf-8")).hexdigest() == sums[n]
 
 
 # --------------------------------------------------
 def random_string():
     """generate a random string"""
 
     k = random.randint(5, 10)
-    return ''.join(random.choices(string.ascii_letters + string.digits, k=k))
+    return "".join(random.choices(string.ascii_letters + string.digits, k=k))
