Problem 04 solution

Author: lukemcguire

04_jump_the_five/jump.py

@@ -0,0 +1,39 @@
+#!/usr/bin/env python3
+"""
+Author : Luke McGuire <[email protected]>
+Date   : 2024-06-18
+Purpose: Jump the Five.
+"""
+
+import argparse
+
+
+# --------------------------------------------------
+def get_args():
+    """Get command-line arguments"""
+
+    parser = argparse.ArgumentParser(
+        description="Jump the Five.",
+        formatter_class=argparse.ArgumentDefaultsHelpFormatter,
+    )
+
+    parser.add_argument("message", metavar="str", help="Message to encode.")
+
+    return parser.parse_args()
+
+
+# --------------------------------------------------
+def main():
+    """Make a jazz noise here"""
+
+    args = get_args()
+    # print(args.message.translate(str.maketrans("1234567890", "9876043215")))
+    cipher = dict(zip("1234567890", "9876043215"))
+
+    # print("".join(cipher.get(c, c) for c in message))
+    print(args.message.translate(str.maketrans(cipher)))
+
+
+# --------------------------------------------------
+if __name__ == "__main__":
+    main()

04_jump_the_five/test.py

@@ -4,7 +4,7 @@
 import os
 from subprocess import getstatusoutput
 
-prg = './jump.py'
+prg = "./jump.py"
 
 
 # --------------------------------------------------
@@ -18,25 +18,25 @@ def test_exists():
 def test_usage():
     """usage"""
 
-    for flag in ['-h', '--help']:
-        rv, out = getstatusoutput(f'{prg} {flag}')
+    for flag in ["-h", "--help"]:
+        rv, out = getstatusoutput(f"python3 {prg} {flag}")
         assert rv == 0
-        assert out.lower().startswith('usage')
+        assert out.lower().startswith("usage")
 
 
 # --------------------------------------------------
 def test_01():
     """test"""
 
-    rv, out = getstatusoutput(f'{prg} 123-456-7890')
+    rv, out = getstatusoutput(f"python3 {prg} 123-456-7890")
     assert rv == 0
-    assert out == '987-604-3215'
+    assert out == "987-604-3215"
 
 
 # --------------------------------------------------
 def test_02():
     """test"""
 
-    rv, out = getstatusoutput(f'{prg} "That number to call is 098-765-4321."')
+    rv, out = getstatusoutput(f'python3 {prg} "That number to call is 098-765-4321."')
     assert rv == 0
-    assert out.rstrip() == 'That number to call is 512-340-6789.'
+    assert out.rstrip() == "That number to call is 512-340-6789."