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))
