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)