Problem 15 solution

Author: lukemcguire

15_kentucky_friar/friar.py

@@ -0,0 +1,136 @@
+#!/usr/bin/env python3
+"""
+Author : Luke McGuire <[email protected]>
+Date   : 2024-06-29
+Purpose: Southern fry text
+"""
+
+import argparse
+import os
+import re
+
+
+# --------------------------------------------------
+def get_args():
+    """Get command-line arguments"""
+
+    parser = argparse.ArgumentParser(
+        description="Southern fry text",
+        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 drop_gs(text):
+    """Drop the g's from two-syllable texts ending in 'ing'"""
+    pattern = re.compile(r"\b(\w*[aeiou]+\w*in)g\b", flags=re.I)
+    return pattern.sub(r"\1'", text)
+
+
+# --------------------------------------------------
+def test_drop_gs():
+    """Test drop_gs function"""
+    assert drop_gs("") == ""
+    assert drop_gs("sing") == "sing"
+    assert drop_gs("cooking") == "cookin'"
+    assert drop_gs("SINGING") == "SINGIN'"
+
+
+def test_drop_gs_sentence():
+    """Test drop_gs function for full lines of text"""
+    text = "That's some good looking chicken."
+    expected_output = "That's some good lookin' chicken."
+    assert drop_gs(text) == expected_output
+
+
+def test_drop_gs_first_word():
+    """Test drop_gs function for first word of sentence"""
+    text = "Looking good, toots."
+    expected_output = "Lookin' good, toots."
+    assert drop_gs(text) == expected_output
+
+
+def test_drop_gs_punctuation():
+    """Test drop_gs function for words followed by punctuation"""
+    text = "Hey good looking, what's cooking?"
+    expected_output = "Hey good lookin', what's cookin'?"
+    assert drop_gs(text) == expected_output
+
+
+# --------------------------------------------------
+def hey_yall(text):
+    """Change 'you' to 'y'all'"""
+    pattern = re.compile(r"\b([yY])ou\b")
+    return pattern.sub(r"\1'all", text)
+
+
+# --------------------------------------------------
+def test_hey_yall_one_match():
+    """Test case 1: The text contains exactly one 'you'"""
+    text = "You are learning Python."
+    expected_output = "Y'all are learning Python."
+    assert hey_yall(text) == expected_output
+
+
+def test_hey_yall_no_match():
+    """Test case 2: The text does not contain 'you'"""
+    text = "Hello world!"
+    expected_output = "Hello world!"
+    assert hey_yall(text) == expected_output
+
+
+def test_hey_yall_multiple_matches():
+    """
+    Test case 3: The text contains multiple 'you' but
+    only the last one should be replaced
+    """
+    text = "You are so much fun. You're going to have a great time!"
+    expected_output = "Y'all are so much fun. " "Y'all're going to have a great time!"
+    assert hey_yall(text) == expected_output
+
+
+def test_hey_yall_match_case():
+    """
+    Test case 4: The text contains 'You', but it should be
+    replaced with 'Y'all' as the function is case-sensitive
+    """
+    text = "You are awesome."
+    expected_output = "Y'all are awesome."
+    assert hey_yall(text) == expected_output
+
+
+def test_hey_yall_your():
+    """Test case 5: The text contains 'your'"""
+    text = "You should watch your mouth."
+    expected_output = "Y'all should watch your mouth."
+    assert hey_yall(text) == expected_output
+
+
+def test_hey_yall_empty_string():
+    """Test case 6: The input text is an empty string"""
+    text = ""
+    expected_output = ""
+    assert hey_yall(text) == expected_output
+
+
+# --------------------------------------------------
+def main():
+    """Make a jazz noise here"""
+
+    args = get_args()
+    for line in args.text.splitlines():
+        print(hey_yall(drop_gs(line)))
+
+
+# --------------------------------------------------
+if __name__ == "__main__":
+    main()

15_kentucky_friar/friar_2.py

@@ -0,0 +1,68 @@
+#!/usr/bin/env python3
+"""
+Author : Luke McGuire <[email protected]>
+Date   : 2024-06-29
+Purpose: Southern fry text
+"""
+
+import argparse
+import os
+import re
+
+
+# --------------------------------------------------
+def get_args():
+    """Get command-line arguments"""
+
+    parser = argparse.ArgumentParser(
+        description="Southern fry text",
+        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 fry(word):
+    """Drog the g from -ing word, you to y'all"""
+    you_match = re.match(r"([Yy])ou", word)
+    ing_match = re.match(r"(^\w*[aeiou]+\w*in)g$", word, flags=re.I)
+    if you_match:
+        word = you_match.group(1) + "'all"
+    elif ing_match:
+        word = ing_match.group(1) + "'"
+
+    return word
+
+
+# --------------------------------------------------
+def test_fry():
+    """Test fry"""
+
+    assert fry("") == ""
+    assert fry("you") == "y'all"
+    assert fry("You") == "Y'all"
+    assert fry("fishing") == "fishin'"
+    assert fry("Aching") == "Achin'"
+    assert fry("swing") == "swing"
+
+
+# --------------------------------------------------
+def main():
+    """Make a jazz noise here"""
+
+    args = get_args()
+    for line in args.text.splitlines():
+        print("".join(map(fry, re.split(r"(\W+)", line.rstrip()))))
+
+
+# --------------------------------------------------
+if __name__ == "__main__":
+    main()

15_kentucky_friar/test.py

@@ -5,22 +5,22 @@
 import re
 from subprocess import getstatusoutput, getoutput
 
-prg = './friar.py'
+PRG = "./friar.py"
 
 
 # --------------------------------------------------
 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)
 
@@ -31,7 +31,7 @@ def test_two_syllable_ing_words():
 
     tests = [("cooking", "cookin'"), ("Fishing", "Fishin'")]
     for given, expected in tests:
-        out = getoutput(f'{prg} {given}')
+        out = getoutput(f"python {PRG} {given}")
         assert out.strip() == expected.strip()
 
 
@@ -41,7 +41,7 @@ def test_one_syllable_ing_words():
 
     tests = [("sing", "sing"), ("Fling", "Fling")]
     for given, expected in tests:
-        out = getoutput(f'{prg} {given}')
+        out = getoutput(f"python {PRG} {given}")
         assert out.strip() == expected.strip()
 
 
@@ -51,7 +51,7 @@ def test_you_yall():
 
     tests = [("you", "y'all"), ("You", "Y'all")]
     for given, expected in tests:
-        out = getoutput(f'{prg} {given}')
+        out = getoutput(f"python {PRG} {given}")
         assert out.strip() == expected.strip()
 
 
@@ -60,45 +60,45 @@ def run_file(file):
     """run with file"""
 
     assert os.path.isfile(file)
-    expected_file = file + '.out'
+    expected_file = file + ".out"
 
     assert os.path.isfile(expected_file)
     expected = open(expected_file).read()
 
-    out = getoutput(f'{prg} {file}')
+    out = getoutput(f"python {PRG} {file}")
     assert out.strip() == expected.strip()
 
 
 # --------------------------------------------------
 def test_blake():
     """blake"""
 
-    run_file('inputs/blake.txt')
+    run_file("inputs/blake.txt")
 
 
 # --------------------------------------------------
 def test_banner():
     """banner"""
 
-    run_file('inputs/banner.txt')
+    run_file("inputs/banner.txt")
 
 
 # --------------------------------------------------
 def test_raven():
     """raven"""
 
-    run_file('inputs/raven.txt')
+    run_file("inputs/raven.txt")
 
 
 # --------------------------------------------------
 def test_dickinson():
     """dickinson"""
 
-    run_file('inputs/dickinson.txt')
+    run_file("inputs/dickinson.txt")
 
 
 # --------------------------------------------------
 def test_shakespeare():
     """shakespeare"""
 
-    run_file('inputs/shakespeare.txt')
+    run_file("inputs/shakespeare.txt")