38 lines
1.2 KiB
Python
38 lines
1.2 KiB
Python
|
import unittest
|
||
|
|
|
||
|
|
|
||
|
|
from count import count_words
|
||
|
|
|
||
|
|
|
||
|
|
class CountWordsTests(unittest.TestCase):
|
||
|
|
|
||
|
|
"""Tests for count_words."""
|
||
|
|
|
||
|
|
def test_simple_sentence(self):
|
||
|
|
actual = count_words("oh what a day what a lovely day")
|
||
|
|
expected = {"oh": 1, "what": 2, "a": 2, "day": 2, "lovely": 1}
|
||
|
|
self.assertEqual(actual, expected)
|
||
|
|
|
||
|
|
def test_apostrophe(self):
|
||
|
|
actual = count_words("don't stop believing")
|
||
|
|
expected = {"don't": 1, "stop": 1, "believing": 1}
|
||
|
|
self.assertEqual(actual, expected)
|
||
|
|
|
||
|
|
# To test the Bonus part of this exercise, comment out the following line
|
||
|
|
# @unittest.expectedFailure
|
||
|
|
def test_capitalization(self):
|
||
|
|
actual = count_words("Oh what a day what a lovely day")
|
||
|
|
expected = {"oh": 1, "what": 2, "a": 2, "day": 2, "lovely": 1}
|
||
|
|
self.assertEqual(actual, expected)
|
||
|
|
|
||
|
|
# To test the Bonus part of this exercise, comment out the following line
|
||
|
|
# @unittest.expectedFailure
|
||
|
|
def test_symbols(self):
|
||
|
|
actual = count_words("¿Te gusta Python?")
|
||
|
|
expected = {"te": 1, "gusta": 1, "python": 1}
|
||
|
|
self.assertEqual(actual, expected)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
unittest.main(verbosity=2)
|