38 lines
781 B
Python
38 lines
781 B
Python
from um import count
|
|
|
|
def main():
|
|
test_count_ums()
|
|
test_words_containing_ums()
|
|
test_punctuation()
|
|
test_from_cs50p()
|
|
|
|
|
|
def test_count_ums():
|
|
assert count("Hello, um, world") == 1
|
|
assert count("Hello um, um, world um") == 3
|
|
assert count("Um Hello, um, world um") == 3
|
|
|
|
|
|
def test_words_containing_ums():
|
|
assert count("instrumentation") == 0
|
|
assert count("umbrella") == 0
|
|
assert count("momentum") == 0
|
|
|
|
|
|
def test_punctuation():
|
|
assert count("Hello um?") == 1
|
|
assert count("Hello um!") == 1
|
|
assert count("Hello um") == 1
|
|
assert count("Hello um.") == 1
|
|
|
|
|
|
def test_from_cs50p():
|
|
assert count("um") == 1
|
|
assert count("um?") == 1
|
|
assert count("Um, thanks for the album.") == 1
|
|
assert count("Um, thanks, um...") == 2
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|