Problem 12 solution

Author: lukemcguire

12_ransom/ransom.py

@@ -0,0 +1,66 @@
+#!/usr/bin/env python3
+"""
+Author : Luke McGuire <[email protected]>
+Date   : 2024-06-28
+Purpose: Ransom Note
+"""
+
+import argparse
+import os
+import random
+
+# import string
+
+
+# --------------------------------------------------
+def get_args():
+    """Get command-line arguments"""
+
+    parser = argparse.ArgumentParser(
+        description="Ransom Note",
+        formatter_class=argparse.ArgumentDefaultsHelpFormatter,
+    )
+
+    parser.add_argument("text", metavar="text", help="Input text or file")
+    parser.add_argument("-s", "--seed", help="Random seed", metavar="int", type=int)
+
+    args = parser.parse_args()
+    if os.path.isfile(args.text):
+        with open(args.text, "rt", encoding="utf-8") as fh:
+            args.text = fh.read().rstrip()
+
+    return args
+
+
+# --------------------------------------------------
+def choose(char: str) -> str:
+    """Choose a random case for a character"""
+
+    return random.choice([char.lower(), char.upper()])
+
+
+# --------------------------------------------------
+def test_choose():
+    """Test choose"""
+
+    state = random.getstate()
+    random.seed(1)
+    assert choose("a") == "a"
+    assert choose("b") == "b"
+    assert choose("c") == "C"
+    assert choose("d") == "d"
+    random.setstate(state)
+
+
+# --------------------------------------------------
+def main():
+    """Make a jazz noise here"""
+
+    args = get_args()
+    random.seed(args.seed)
+    print("".join(map(choose, args.text)))
+
+
+# --------------------------------------------------
+if __name__ == "__main__":
+    main()

12_ransom/test.py

@@ -6,29 +6,30 @@
 import random
 from subprocess import getstatusoutput
 
-prg = './ransom.py'
-fox = '../inputs/fox.txt'
-now = '../inputs/now.txt'
+PRG = "./ransom.py"
+FOX = "../inputs/FOX.txt"
+NOW = "../inputs/NOW.txt"
 
 
 # --------------------------------------------------
 def seed_flag():
-    return '-s' if random.randint(0, 1) else '--seed'
+    """seed"""
+    return "-s" if random.randint(0, 1) else "--seed"
 
 
 # --------------------------------------------------
 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)
 
@@ -37,12 +38,14 @@ def test_usage():
 def test_text1():
     """Test"""
 
-    in_text = 'The quick brown fox jumps over the lazy dog.'
-    tests = [('1', 'thE QUICk BrOWn Fox jumpS OveR tHe LAzY dOg.'),
-             ('3', 'thE quICk BROwn Fox jUmPS OVEr the lAZY DOG.')]
+    in_text = "The quick brown FOX jumps over the lazy dog."
+    tests = [
+        ("1", "thE QUICk BrOWn Fox jumpS OveR tHe LAzY dOg."),
+        ("3", "thE quICk BROwn Fox jUmPS OVEr the lAZY DOG."),
+    ]
 
     for seed, expected in tests:
-        rv, out = getstatusoutput(f'{prg} {seed_flag()} {seed} "{in_text}"')
+        rv, out = getstatusoutput(f'python {PRG} {seed_flag()} {seed} "{in_text}"')
         assert rv == 0
         assert out.strip() == expected
 
@@ -51,16 +54,14 @@ def test_text1():
 def test_text2():
     """Test"""
 
-    in_text = 'Now is the time for all good men to come to the aid of the party.'
+    in_text = "Now is the time for all good men to come to the aid of the party."
     tests = [
-        ('2',
-         'now iS the TIME fOR ALl good meN TO COMe To THE AID oF THE PArTY.'),
-        ('5',
-         'NOw is tHE Time FOr all good men To coME TO tHe AiD OF THe ParTy.')
+        ("2", "now iS the TIME fOR ALl good meN TO COMe To THE AID oF THE PArTY."),
+        ("5", "NOw is tHE Time FOr all good men To coME TO tHe AiD OF THe ParTy."),
     ]
 
     for seed, expected in tests:
-        rv, out = getstatusoutput(f'{prg} {seed_flag()} {seed} "{in_text}"')
+        rv, out = getstatusoutput(f'python {PRG} {seed_flag()} {seed} "{in_text}"')
         assert rv == 0
         assert out.strip() == expected
 
@@ -69,11 +70,13 @@ def test_text2():
 def test_file1():
     """Test"""
 
-    tests = [('1', 'thE QUICk BrOWn Fox jumpS OveR tHe LAzY dOg.'),
-             ('3', 'thE quICk BROwn Fox jUmPS OVEr the lAZY DOG.')]
+    tests = [
+        ("1", "thE QUICk BrOWn Fox jumpS OveR tHe LAzY dOg."),
+        ("3", "thE quICk BROwn Fox jUmPS OVEr the lAZY DOG."),
+    ]
 
     for seed, expected in tests:
-        rv, out = getstatusoutput(f'{prg} {seed_flag()} {seed} {fox}')
+        rv, out = getstatusoutput(f"python {PRG} {seed_flag()} {seed} {FOX}")
         assert rv == 0
         assert out.strip() == expected
 
@@ -83,13 +86,11 @@ def test_file2():
     """Test"""
 
     tests = [
-        ('2',
-         'now iS the TIME fOR ALl good meN TO COMe To THE AID oF THE PArTY.'),
-        ('5',
-         'NOw is tHE Time FOr all good men To coME TO tHe AiD OF THe ParTy.')
+        ("2", "now iS the TIME fOR ALl good meN TO COMe To THE AID oF THE PArTY."),
+        ("5", "NOw is tHE Time FOr all good men To coME TO tHe AiD OF THe ParTy."),
     ]
 
     for seed, expected in tests:
-        rv, out = getstatusoutput(f'{prg} {seed_flag()} {seed} {now}')
+        rv, out = getstatusoutput(f"python {PRG} {seed_flag()} {seed} {NOW}")
         assert rv == 0
         assert out.strip() == expected