def test_read_bytes(self):
byte_stream = pack('>BBB', 1, 2, 3)
byte_stream = bytearray(byte_stream)
reader = BigEndianByteArrayByteReader(byte_stream)
value = reader.read_bytes()
for i in range(3):
self.assertEqual(value[i], i+1)
def test_read_bytes_not_enough_bits(self):
with self.assertRaises(EOFError):
byte_stream = pack('>BBB', 1, 2, 3)
byte_stream = bytearray(byte_stream)
reader = BigEndianByteArrayByteReader(byte_stream)
value = reader.read_bytes(4)
