Problem10 solution

Author: lukemcguire

10_telephone/telephone.py

@@ -0,0 +1,76 @@
+#!/usr/bin/env python3
+"""
+Author : Luke McGuire <[email protected]>
+Date   : 2024-06-27
+Purpose: Telephone
+"""
+
+import argparse
+import os
+import random
+import string
+
+
+# --------------------------------------------------
+def float_percentage(s):
+    """check for float between 0 and 1"""
+    try:
+        value = float(s)
+    except ValueError as e:
+        raise argparse.ArgumentTypeError(f"invalid float value: '{s}'") from e
+    if value < 0 or value >= 1:
+        raise argparse.ArgumentTypeError(f'"{value}" must be between 0 and 1')
+    return value
+
+
+# --------------------------------------------------
+def get_args():
+    """Get command-line arguments"""
+
+    parser = argparse.ArgumentParser(
+        description="Telephone", 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)
+
+    parser.add_argument(
+        "-m",
+        "--mutations",
+        help="Percent mutations",
+        metavar="float",
+        type=float_percentage,
+        default=0.1,
+    )
+
+    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 main():
+    """Make a jazz noise here"""
+
+    args = get_args()
+    text = args.text
+    random.seed(args.seed)
+    num_mutations = round(len(text) * args.mutations)
+    alpha = "".join(sorted(string.ascii_letters + string.punctuation))
+    scrambled = list(text)
+    for i in random.sample(range(len(text)), k=num_mutations):
+        scrambled[i] = random.choice(alpha.replace(text[i], ""))
+
+    scrambled = "".join(scrambled)
+
+    print(f'You said: "{text}"')
+    print(f'I heard : "{scrambled}"')
+
+
+# --------------------------------------------------
+if __name__ == "__main__":
+    main()

10_telephone/test.py

@@ -7,33 +7,33 @@
 import re
 import string
 
-prg = "./telephone.py"
-fox = '../inputs/fox.txt'
-now = '../inputs/now.txt'
+PRG = "./telephone.py"
+FOX = "../inputs/fox.txt"
+NOW = "../inputs/now.txt"
 
 
 # --------------------------------------------------
 def test_exists():
     """exists"""
 
-    assert os.path.isfile(prg)
+    assert os.path.isfile(PRG)
 
 
 # --------------------------------------------------
 def test_usage():
     """usage"""
 
-    for flag in ['', '-h', '--help']:
-        out = getoutput(f'{prg} {flag}')
-        assert re.match('usage', out, re.IGNORECASE)
+    for flag in ["", "-h", "--help"]:
+        out = getoutput(f"python3 {PRG} {flag}")
+        assert re.match("usage", out, re.IGNORECASE)
 
 
 # --------------------------------------------------
 def test_bad_seed_str():
     """bad seed str value"""
 
     bad = random_string()
-    rv, out = getstatusoutput(f'{prg} -s {bad} {fox}')
+    rv, out = getstatusoutput(f"python3 {PRG} -s {bad} {FOX}")
     assert rv > 0
     assert re.search(f"invalid int value: '{bad}'", out)
 
@@ -43,7 +43,7 @@ def test_bad_mutation_str():
     """bad mutation str value"""
 
     bad = random_string()
-    rv, out = getstatusoutput(f'{prg} -m {bad} {fox}')
+    rv, out = getstatusoutput(f"python3 {PRG} -m {bad} {FOX}")
     assert rv > 0
     assert re.search(f"invalid float value: '{bad}'", out)
 
@@ -52,18 +52,18 @@ def test_bad_mutation_str():
 def test_bad_mutation():
     """bad mutation values"""
 
-    for val in ['-1.0', '10.0']:
-        rv, out = getstatusoutput(f'{prg} -m {val} {fox}')
+    for val in ["-1.0", "10.0"]:
+        rv, out = getstatusoutput(f"python3 {PRG} -m {val} {FOX}")
         assert rv > 0
-        assert re.search(f'--mutations "{val}" must be between 0 and 1', out)
+        assert re.search(f'--mutations: "{val}" must be between 0 and 1', out)
 
 
 # --------------------------------------------------
 def test_for_echo():
     """test"""
 
-    txt = open(now).read().rstrip()
-    rv, out = getstatusoutput(f'{prg} -m 0 "{txt}"')
+    txt = open(NOW).read().rstrip()
+    rv, out = getstatusoutput(f'python3 {PRG} -m 0 "{txt}"')
     assert rv == 0
     assert out.rstrip() == f'You said: "{txt}"\nI heard : "{txt}"'
 
@@ -72,8 +72,8 @@ def test_for_echo():
 def test_now_cmd_s1():
     """test"""
 
-    txt = open(now).read().rstrip()
-    rv, out = getstatusoutput(f'{prg} -s 1 "{txt}"')
+    txt = open(NOW).read().rstrip()
+    rv, out = getstatusoutput(f'python3 {PRG} -s 1 "{txt}"')
     assert rv == 0
     expected = """
     Now is Ege time [dr all good me- to come to the jid of the party.
@@ -85,8 +85,8 @@ def test_now_cmd_s1():
 def test_now_cmd_s2_m4():
     """test"""
 
-    txt = open(now).read().rstrip()
-    rv, out = getstatusoutput(f'{prg} -s 2 -m .4 "{txt}"')
+    txt = open(NOW).read().rstrip()
+    rv, out = getstatusoutput(f'python3 {PRG} -s 2 -m .4 "{txt}"')
     assert rv == 0
     expected = """
     No$ i% khefMiIe sor@all$glo<BmenYts cAAeltaTtheSaid[HYnthe Aalty.
@@ -98,9 +98,9 @@ def test_now_cmd_s2_m4():
 def test_fox_file_s1():
     """test"""
 
-    rv, out = getstatusoutput(f'{prg} --seed 1 {fox}')
+    rv, out = getstatusoutput(f"python3 {PRG} --seed 1 {FOX}")
     assert rv == 0
-    txt = open(fox).read().rstrip()
+    txt = open(FOX).read().rstrip()
     expected = "The duic: brown hox jumps over the lkzy dog."
     assert out.rstrip() == f'You said: "{txt}"\nI heard : "{expected}"'
 
@@ -109,9 +109,9 @@ def test_fox_file_s1():
 def test_fox_file_s2_m6():
     """test"""
 
-    rv, out = getstatusoutput(f'{prg} --seed 2 --mutations .6 {fox}')
+    rv, out = getstatusoutput(f"python3 {PRG} --seed 2 --mutations .6 {FOX}")
     assert rv == 0
-    txt = open(fox).read().rstrip()
+    txt = open(FOX).read().rstrip()
     expected = "ZoA@qric` HwdTB Alx$jumIslolXs th^Yl?dy<YoA."
     assert out.rstrip() == f'You said: "{txt}"\nI heard : "{expected}"'
 
@@ -120,4 +120,4 @@ def test_fox_file_s2_m6():
 def random_string():
     """generate a random filename"""
 
-    return ''.join(random.choices(string.ascii_lowercase + string.digits, k=5))
+    return "".join(random.choices(string.ascii_lowercase + string.digits, k=5))