test_twttr

Author: Sa3d-Ka

Problem Set 2/twttr/twttr.py

@@ -1,18 +1,7 @@
-def main():
-    word = input("Input: ")
-    print("Output: ", end="")
-    word_without_vowel = shorten(word)
-    print(word_without_vowel)
-    
+inp = input("Input: ")
+print("Output: ", end="")
 
-
-def shorten(word):
-    word_without_vowel = ''
-    for letter in word:
-        if not letter.lower() in ['a', 'e', 'i', 'o', 'u']:
-           word_without_vowel += letter
-    return word_without_vowel
-           
-
-if __name__ == "__main__":
-    main()
+for letter in inp:
+    if not letter.lower() in ['a', 'e', 'i', 'o', 'u']:
+        print(letter, end="")
+print()

Problem Set 5/test_twttr/test_twttr.py

@@ -0,0 +1,25 @@
+from twttr import shorten
+
+def main():
+    test_upper_lower_cases()
+    test_numbers()
+    test_punctuation()
+
+
+# Test upper and lower cases
+def test_upper_lower_cases():
+    assert shorten('twitter') == 'twttr'
+    assert shorten('TWITTER') == 'TWTTR'
+    assert shorten('TwiTter') == 'TwTtr'
+
+# Test numbers
+def test_numbers():
+    assert shorten('1234') == '1234'
+
+# Test punctuation
+def test_punctuation():
+    assert shorten('!?,.') == '!?,.'
+
+
+if __name__ == "__main__":
+    main()

Problem Set 5/test_twttr/twttr.py

@@ -0,0 +1,18 @@
+def main():
+    word = input("Input: ")
+    print("Output: ", end="")
+    word_without_vowel = shorten(word)
+    print(f'{word_without_vowel}')
+    
+
+
+def shorten(word):
+    word_without_vowel = ''
+    for letter in word:
+        if not letter.lower() in ['a', 'e', 'i', 'o', 'u']:
+           word_without_vowel += letter
+    return word_without_vowel
+           
+
+if __name__ == "__main__":
+    main()