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_bytes_not_enough_bits(self): with self.assertRaises(EOFError): byte_stream = pack('<BBB',1,2,3) byte_stream = bytearray(byte_stream) reader = LittleEndianByteArrayByteReader(byte_stream) value = reader.read_bytes(4)