def test_read_bytes(self):
byte_stream = pack('<BBB',1,2,3)
byte_stream = bytearray(byte_stream)
reader = LittleEndianByteArrayByteReader(byte_stream)
value = reader.read_bytes()
for i in range(3):
self.assertEqual(value[i],i+1)
def test_read_byte(self):
byte_stream = pack('<B',1)
byte_stream = bytearray(byte_stream)
reader = LittleEndianByteArrayByteReader(byte_stream)
self.assertEqual(reader.read_byte(),1)
def test_read_long_not_enough_bits(self):
with self.assertRaises(EOFError):
byte_stream = pack('<I',10)
byte_stream = bytearray(byte_stream)
reader = LittleEndianByteArrayByteReader(byte_stream)
value = reader.read_long()
def test_read_long(self):
byte_stream = pack('<Q',0xFF1200A3)
byte_stream = bytearray(byte_stream)
reader = LittleEndianByteArrayByteReader(byte_stream)
value = reader.read_long()
self.assertEqual(value,0xFF1200A3)
def test_read_short(self):
byte_stream = pack('<H',0x87)
byte_stream = bytearray(byte_stream)
reader = LittleEndianByteArrayByteReader(byte_stream)
value = reader.read_short()
self.assertEqual(value,0x87)
