def test_word_frequency(self):
with open('romeojuliet.txt', 'r') as reader:
freq = word_frequency(
sanitize(word) for line in reader for word in line.split())
self.assertEqual(3741, len(freq))
self.assertEqual(293, freq['romeo'])
self.assertEqual(176, freq['juliet'])
def main(filename, words):
with open(filename, 'r') as reader:
freq = word_frequency(sanitize(word) for line in reader for word in line.split())
print('unique words:', len(freq))
for word in words:
print(word, freq[word])