atrcopy/test/test_kboot.py

53 lines
1.5 KiB
Python
Raw Normal View History

2017-05-07 20:28:15 +00:00
from __future__ import print_function
2017-05-07 21:13:48 +00:00
from __future__ import division
from builtins import range
from builtins import object
2016-05-04 06:21:01 +00:00
import os
import numpy as np
2017-05-07 21:13:48 +00:00
from mock import *
from atrcopy import SegmentData, KBootImage, add_xexboot_header, add_atr_header
2016-05-04 06:21:01 +00:00
class TestKbootHeader:
2016-05-04 06:21:01 +00:00
def setup(self):
pass
def check_size(self, data):
xex_size = len(data)
bytes = add_xexboot_header(data)
# rawdata = SegmentData(bytes)
# size = len(rawdata)
# atr = KBootImage(rawdata)
# newatr = atr.as_new_format("ATR")
image_size = len(bytes)
bytes = add_atr_header(bytes)
rawdata = SegmentData(bytes)
newatr = KBootImage(rawdata)
image = newatr.bytes
2017-05-07 20:28:15 +00:00
print(image[0:16])
paragraphs = image_size // 16
2017-05-07 20:28:15 +00:00
print(newatr.header, paragraphs)
assert int(image[2:4].view(dtype='<u2')) == paragraphs
assert int(image[16 + 9:16 + 9 + 2].view('<u2')) == xex_size
2016-05-04 06:21:01 +00:00
return image
def test_simple(self):
for size in range(2000, 40000, 1000):
data = np.arange(size, dtype=np.uint8)
self.check_size(data)
def test_real(self):
data = np.fromfile("../test_data/air_defense_v18.xex", dtype=np.uint8)
2016-05-04 06:21:01 +00:00
image = self.check_size(data)
with open("../test_data/air_defense_v18.atr", "wb") as fh:
2016-05-04 06:21:01 +00:00
txt = image.tostring()
fh.write(txt)
if __name__ == "__main__":
t = TestKbootHeader()
t.setup()
t.test_simple()
t.test_real()