Problem 14 solution

Author: lukemcguire

14_rhymer/rhymer.py

@@ -0,0 +1,66 @@
+#!/usr/bin/env python3
+"""
+Author : Luke McGuire <[email protected]>
+Date   : 2024-06-29
+Purpose: Make rhyming "words"
+"""
+
+import argparse
+import re
+
+
+# --------------------------------------------------
+def get_args():
+    """Get command-line arguments"""
+
+    parser = argparse.ArgumentParser(
+        description='Make rhyming "words"',
+        formatter_class=argparse.ArgumentDefaultsHelpFormatter,
+    )
+
+    parser.add_argument("str", metavar="str", help="A word to rhyme")
+
+    return parser.parse_args()
+
+
+# --------------------------------------------------
+def stemmer(word: str):
+    """Break a word into its leading consonants and the remainder"""
+    pattern = re.compile(r"^([^aeiou]*)(.*)$")
+    return pattern.search(word.lower()).groups()
+
+
+# --------------------------------------------------
+def test_stemmer():
+    """Test stemmer"""
+    assert stemmer("") == ("", "")
+    assert stemmer("cake") == ("c", "ake")
+    assert stemmer("chair") == ("ch", "air")
+    assert stemmer("APPLE") == ("", "apple")
+    assert stemmer("RDNZL") == ("rdnzl", "")
+
+
+# --------------------------------------------------
+def main():
+    """Make a jazz noise here"""
+
+    consonants = list("bcdfghjklmnpqrstvwxyz")
+    clusters = """
+bl br ch cl cr dr fl fr gl gr pl pr sc 
+sh sk sl sm sn sp st sw th tr tw thw wh wr 
+sch scr shr sph spl spr squ str thr""".split()
+    prefixes = sorted(consonants + clusters)
+
+    args = get_args()
+    word = args.str
+    start, rest = stemmer(word)
+
+    if rest:
+        print("\n".join([p + rest for p in prefixes if p != start]))
+    else:
+        print(f'Cannot rhyme "{word}"')
+
+
+# --------------------------------------------------
+if __name__ == "__main__":
+    main()

14_rhymer/test.py

@@ -5,70 +5,70 @@
 import random
 from subprocess import getoutput
 
-prg = './rhymer.py'
+PRG = "./rhymer.py"
 
 
 # --------------------------------------------------
 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 out.lower().startswith('usage')
+    for flag in ["", "-h", "--help"]:
+        out = getoutput(f"python {PRG} {flag}")
+        assert out.lower().startswith("usage")
 
 
 # --------------------------------------------------
 def test_take():
     """leading consonant"""
 
-    out = getoutput(f'{prg} take').splitlines()
+    out = getoutput(f"python {PRG} take").splitlines()
     assert len(out) == 56
-    assert out[0] == 'bake'
-    assert out[-1] == 'zake'
+    assert out[0] == "bake"
+    assert out[-1] == "zake"
 
 
 # --------------------------------------------------
 def test_chair():
     """consonant cluster"""
 
-    out = getoutput(f'{prg} chair').splitlines()
+    out = getoutput(f"python {PRG} chair").splitlines()
     assert len(out) == 56
-    assert out[1] == 'blair'
-    assert out[-2] == 'yair'
+    assert out[1] == "blair"
+    assert out[-2] == "yair"
 
 
 # --------------------------------------------------
 def test_chair_uppercase():
     """consonant cluster"""
 
-    out = getoutput(f'{prg} CHAIR').splitlines()
+    out = getoutput(f"python {PRG} CHAIR").splitlines()
     assert len(out) == 56
-    assert out[1] == 'blair'
-    assert out[-2] == 'yair'
+    assert out[1] == "blair"
+    assert out[-2] == "yair"
 
 
 # --------------------------------------------------
 def test_apple():
     """leading vowel"""
 
-    out = getoutput(f'{prg} apple').splitlines()
+    out = getoutput(f"python {PRG} apple").splitlines()
     assert len(out) == 57
-    assert out[10] == 'flapple'
-    assert out[-10] == 'thwapple'
+    assert out[10] == "flapple"
+    assert out[-10] == "thwapple"
 
 
 # --------------------------------------------------
 def test_no_vowels():
     """no vowels"""
 
-    consonants = 'bcdfghjklmnpqrstvwxyzBCDFGHJKLMNPQRSTVWXYZ'
-    bad = ''.join(random.sample(consonants, k=random.randint(4, 10)))
-    out = getoutput(f'{prg} {bad}')
+    consonants = "bcdfghjklmnpqrstvwxyzBCDFGHJKLMNPQRSTVWXYZ"
+    bad = "".join(random.sample(consonants, k=random.randint(4, 10)))
+    out = getoutput(f"python {PRG} {bad}")
     assert out == f'Cannot rhyme "{bad}"'