Updated test and code for matching case for article.

RyanAlcoran · RyanAlcoran · commit ed69f2d0a271 · 2020-11-16T14:20:26.000-05:00
diff --git a/02_crowsnest/crowsnest.py b/02_crowsnest/crowsnest.py
@@ -27,7 +27,15 @@ def main():
 
     args = get_args()
     word = args.word
-    article = 'an' if word[0].lower() in 'aeiou' else 'a'
+
+    """Original solution"""
+    #article = 'an' if word[0].lower() in 'aeiou' else 'a'
+
+    if word[0].islower():
+        article = 'an' if word[0] in 'aeiou' else 'a'
+    else:
+        article = 'An' if word[0] in 'AEIOU' else 'A'
+
     print(f'Ahoy, Captain, {article} {word} off the larboard bow!')
 
 # --------------------------------------------------
diff --git a/02_crowsnest/test.py b/02_crowsnest/test.py
@@ -1,4 +1,4 @@
-#!/usr/bin/env python3
+ #!/usr/bin/env python3
 """tests for crowsnest.py"""
 
 import os
@@ -47,7 +47,7 @@ def test_consonant_upper():
 
     for word in consonant_words:
         out = getoutput(f'{prg} {word.title()}')
-        assert out.strip() == template.format('a', word.title())
+        assert out.strip() == template.format('A', word.title())
 
 
 # --------------------------------------------------
@@ -64,5 +64,5 @@ def test_vowel_upper():
     """octopus -> an Octopus"""
 
     for word in vowel_words:
-        out = getoutput(f'{prg} {word.upper()}')
-        assert out.strip() == template.format('an', word.upper())
+        out = getoutput(f'{prg} {word.title()}')
+        assert out.strip() == template.format('An', word.title())
