class WordCounterTest(unittest.TestCase): def set_up(self, filename): self.f = open(filename) self.fr = self.f.read() self.wc = WordCounter(self.fr) def test_empty(self): self.set_up("samples/empty.txt") self.assertEqual(self.wc.get_count(), 0) self.assertEqual(self.wc.get_most_frequent(), {}) def test_base(self): self.set_up("samples/test.txt") self.assertEqual(self.wc.get_count(), 9) self.assertEqual(len(self.wc.get_most_frequent().keys()), 8) def test_long(self): self.set_up("samples/long.txt") self.assertEqual(self.wc.get_count(), 188) self.assertEqual(len(self.wc.get_most_frequent().keys()), 10) def tearDown(self): self.wc = None self.f.close()