Exercism sync/c8f4d80207de6871

pig-latin/pig_latin.py

@@ -16,7 +16,7 @@ def translate(text: str) -> str:
     :param text: The English text to translate
     :return: The text translated to Pig Latin
     """
-    words: list[str] = text.lower().split(" ")
+    words: list[str] = text.split(" ")
     return " ".join(process_word(word) for word in words)
 
 
@@ -31,35 +31,37 @@ def process_word(text: str) -> str:
     if not text:
         return ""
 
+    text_lower: str = text.lower()
+
     # Rule 1
-    if is_vowel(text[0]) or text[:2] in ("xr", "yt"):
+    if is_vowel(text_lower[0]) or text_lower[:2] in ("xr", "yt"):
         # If a word begins with a vowel,
         # or starts with "xr" or "yt",
         # add an "ay" sound to the end of the word.
         return f"{text}ay"
 
     # Rule 3
-    if is_rule_3(text):
+    if is_rule_3(text_lower):
         # If a word starts with zero or more consonants followed by "qu",
         # first move those consonants (if any) and the "qu" part to the
         # end of the word, and then add an "ay" sound to the end of the word.
-        i = text.index("qu")
+        i = text_lower.index("qu")
         return f"{text[i + 2 :]}{text[: i + 2]}ay"
 
     # Rule 4
-    if is_rule_4(text):
+    if is_rule_4(text_lower):
         # If a word starts with one or more consonants followed by "y",
         # first move the consonants preceding the "y" to the end of the
         # word, and then add an "ay" sound to the end of the word.
-        i = text.index("y")
+        i = text_lower.index("y")
         return f"{text[i:]}{text[:i]}ay"
 
     # Rule 2
-    if not is_vowel(text[0]):
+    if not is_vowel(text_lower[0]):
         # If a word begins with one or more consonants, first move those
         # consonants to the end of the word and then add an "ay" sound
         # to the end of the word.
-        i = get_consonant_cluster_length(text)
+        i = get_consonant_cluster_index(text_lower)
         return f"{text[i + 1 :]}{text[: i + 1]}ay"
 
     raise ValueError(f"Unhandled word in Pig Latin translation: '{text}'")
@@ -69,32 +71,25 @@ def is_rule_3(text: str) -> bool:
     """
     Check if a word starts with zero or more consonants followed by "qu".
 
-    :param text:
-    :return:
+    :param text: The word to check
+    :return: True if the word matches the pattern, False otherwise
     """
     if "qu" in text:
-        if text[:2] == "qu":
-            return True
-
-        for char in text[: text.index("qu")]:
-            if is_vowel(char):
-                return False
-        return True
+        qu_indx: int = text.index("qu")
+        return all(not is_vowel(char) for char in text[: qu_indx])
     return False
 
 
 def is_rule_4(text: str) -> bool:
     """
     Check if a word starts with one or more consonants followed by "y".
 
-    :param text:
-    :return:
+    :param text: The word to check
+    :return: True if the word matches the pattern, False otherwise
     """
     if "y" in text and text[0] != "y":
-        for char in text[: text.index("y")]:
-            if is_vowel(char):
-                return False
-        return True
+        y_indx: int = text.index("y")
+        return all(not is_vowel(char) for char in text[: y_indx])
     return False
 
 
@@ -108,18 +103,18 @@ def is_vowel(char: str) -> bool:
     return char in "aeiou"
 
 
-def get_consonant_cluster_length(text: str) -> int:
+def get_consonant_cluster_index(text: str) -> int:
     """
-    Find the length of the consonant cluster at the beginning of a word.
+    Find the index of the last consonant in the initial consonant cluster.
 
     :param text: The word to analyze
     :return: The index of the last consonant in the initial consonant cluster
     """
-    i = 0
-    for n, char in enumerate(text):
+    index: int = 0
+    for i, char in enumerate(text):
         if not is_vowel(char):
-            i = n
+            index = i
         else:
             break
 
-    return i
+    return index

pig-latin/pig_latin_test.py

@@ -12,6 +12,10 @@
 
 
 class PigLatinTest(unittest.TestCase):
+
+    def test_word_with_capital_letter(self):
+        self.assertEqual(translate("Queen"), "eenQuay")
+
     def test_word_beginning_with_a(self):
         self.assertEqual(translate("apple"), "appleay")