Problem 18 solution

Author: lukemcguire

18_gematria/gematria.py

@@ -0,0 +1,60 @@
+#!/usr/bin/env python3
+"""
+Author : Luke McGuire <[email protected]>
+Date   : 2024-07-01
+Purpose: Gematria
+"""
+
+import argparse
+import os
+import re
+
+
+# --------------------------------------------------
+def get_args():
+    """Get command-line arguments"""
+
+    parser = argparse.ArgumentParser(
+        description="Gematria", formatter_class=argparse.ArgumentDefaultsHelpFormatter
+    )
+
+    parser.add_argument("text", metavar="text", help="Input text or file")
+
+    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 word_to_num(word: str) -> str:
+    """Return the sum of ascii character values for a word"""
+
+    return str(sum(map(ord, re.sub(r"[^a-zA-Z0-9]", "", word))))
+
+
+# --------------------------------------------------
+def test_word_to_num():
+    """Test word_to_num function"""
+
+    assert word_to_num("a") == "97"
+    assert word_to_num("abc") == "294"
+    assert word_to_num("ab'c") == "294"
+    assert word_to_num("4a-b'c,") == "346"
+
+
+# --------------------------------------------------
+def main():
+    """Make a jazz noise here"""
+
+    args = get_args()
+
+    for line in args.text.splitlines():
+        print(" ".join(map(word_to_num, line.split())))
+
+
+# --------------------------------------------------
+if __name__ == "__main__":
+    main()

18_gematria/test.py

@@ -5,25 +5,25 @@
 import re
 from subprocess import getstatusoutput, getoutput
 
-prg = './gematria.py'
-spiders = '../inputs/spiders.txt'
-fox = '../inputs/fox.txt'
-sonnet = '../inputs/sonnet-29.txt'
+PRG = "./gematria.py"
+SPIDERS = "../inputs/spiders.txt"
+FOX = "../inputs/fox.txt"
+SONNET = "../inputs/sonnet-29.txt"
 
 
 # --------------------------------------------------
 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)
 
@@ -32,31 +32,31 @@ def test_usage():
 def test_text():
     """Text"""
 
-    out = getoutput(f'{prg} "foo bar baz"')
-    assert out.strip() == '324 309 317'
+    out = getoutput(f'python {PRG} "foo bar baz"')
+    assert out.strip() == "324 309 317"
 
 
 # --------------------------------------------------
 def test_fox():
     """File"""
 
-    out = getoutput(f'{prg} {fox}')
-    assert out.strip() == '289 541 552 333 559 444 321 448 314'
+    out = getoutput(f"python {PRG} {FOX}")
+    assert out.strip() == "289 541 552 333 559 444 321 448 314"
 
 
 # --------------------------------------------------
 def test_spiders():
     """File"""
 
-    out = getoutput(f'{prg} {spiders}')
-    assert out.strip() == '405 579 762\n73 421 548\n862'
+    out = getoutput(f"python {PRG} {SPIDERS}")
+    assert out.strip() == "405 579 762\n73 421 548\n862"
 
 
 # --------------------------------------------------
 def test_sonnet():
     """File"""
 
-    out = getoutput(f'{prg} {sonnet}')
+    out = getoutput(f"python {PRG} {SONNET}")
     expected = """
 631 107
 719 1132