def test_read_bytes_error(self): with self.assertRaises(EOFError): reader = LittleEndianDataReaderByteReader(MyDataReader(bytearray([1, 2, 3, 4, 5]))) reader.read_bytes(6)
def test_read_bytes(self): ba = bytearray([1, 2, 3, 4, 5]) reader = LittleEndianDataReaderByteReader(MyDataReader(ba)) stream = reader.read_bytes(5) for i in range(5): self.assertEqual(stream[i],ba[i])