class TestFileTypes(unittest.TestCase):
def setUp(self):
self.endnumber = TextScanner("numberatend.txt")
self.galaxysong = TextScanner("galaxysong.txt")
self.empty = TextScanner("emptylist.txt")
def test_numberword_at_list_end(self):
self.assertEqual(self.endnumber.wordsum(), 900000002)
def test_galaxysong(self):
self.assertEqual(self.galaxysong.wordsum(), 100213189920)
def test_empty_file(self):
self.assertEqual(self.empty.wordsum(), 0)
def setUp(self):
self.endnumber = TextScanner("numberatend.txt")
self.galaxysong = TextScanner("galaxysong.txt")
self.empty = TextScanner("emptylist.txt")
