class TestParserRead(ut.TestCase): """Tests the Parser.read() method""" def setUp(self): self.data = b'\x01\x02\x03\x04\x05' self.parser = Parser(io.BytesIO(self.data)) def test_read_full(self): self.assertEqual(self.parser.read(5), self.data) def test_read_parial(self): self.assertEqual(self.parser.read(4), self.data[:4]) self.assertEqual(self.parser.read(0), b'') self.assertEqual(self.parser.read(1), self.data[4:]) def test_read_beyond_EOF(self): with self.assertRaises(IOError): self.parser.read(6)
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)
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)
def setUp(self): fname = os.path.join(os.path.dirname(__file__), 'Test1.bin') self.file = open(fname, 'rb') self.parser = Parser(self.file)
def setUp(self): self.data = b'\x01\x02\x03\x04\x05' self.parser = Parser(io.BytesIO(self.data))