class TestParserReadFile(ut.TestCase): """Tests the Parser reading a sample valid file (Test1.bin)""" def setUp(self): fname = os.path.join(os.path.dirname(__file__), 'Test1.bin') self.file = open(fname, 'rb') self.parser = Parser(self.file) def tearDown(self): self.file.close() def test_read_header(self): self.assertEqual(self.parser.read_header(), 1)
class TestParserReadEmpty(ut.TestCase): """Tests the Parser reading an empty character pool (Empty.bin)""" def setUp(self): fname = os.path.join(os.path.dirname(__file__), 'Empty.bin') self.file = open(fname, 'rb') self.parser = Parser(self.file) def tearDown(self): self.file.close() def test_read_header(self): self.assertEqual(self.parser.read_header(), 0)