def test_node_read():
file = io.BytesIO(struct.pack("h", 0x21))
node = nodes.Node(file, tools.Block("root", 0, file))
assert node
assert node.node_type_id == 0x21
assert [node_type.name
for node_type in node.node_types] == ['HEADER', 'MESH']
def test_node_serialize():
file = io.BytesIO(struct.pack("h", 1))
node = nodes.Node(file, tools.Block("root", 0, file))
serialized = node.__serialize__()
assert 'node_type_id' in serialized
assert 'node_types' in serialized
assert 'name' in serialized
assert 'headers' in serialized
def test_array_use_block():
file = SeekLoggingBytesIO(
bytearray([
0x01, 0x02, 0x03, 0x04, 0x04, 0x00, 0x00, 0x00, 0x05, 0x00, 0x00,
0x00, 0x01, 0x02, 0x03, 0x04, 0x05
]))
array = base.Array(file)
root_block = tools.Block("root", 0, file)
array.read_data(file, tools.readu8, root_block.start_block("foo", 0))
assert len(root_block.blocks) == 1
block = root_block.blocks[0]
assert block.name == "foo"
assert block.start == 12
assert block.end == 12 + 5
def test_node_header_serialize():
input = struct.pack("=hh hhh i fff ffff iii iii iii", 1, 2, 3, 4, 5,
0x06070809, 1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 1, 2, 3,
2, 3, 4, 3, 4, 5)
file = io.BytesIO(input)
node_header = nodes.NodeHeader(file, tools.Block("root", 0, file))
serialized = node_header.__serialize__()
assert 'parent_node' in serialized
assert 'node_id' in serialized
assert 'unknown' in serialized
assert 'parent_node_start' in serialized
assert 'position' in serialized
assert 'rotation' in serialized
assert 'child_offsets' in serialized
assert 'controllers' in serialized
assert 'controller_data' in serialized
def test_node_header_read():
input = struct.pack("=hh hhh i fff ffff iii iii iii", 1, 2, 3, 4, 5,
0x06070809, 1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 1, 2, 3,
2, 3, 4, 3, 4, 5)
print(input)
file = io.BytesIO(input)
node_header = nodes.NodeHeader(file, tools.Block("root", 0, file))
assert node_header
assert node_header.parent_node == 1
assert node_header.node_id == 2
assert node_header.unknown == bytearray([3, 0, 4, 0, 5, 0])
assert node_header.parent_node_start == 0x06070809
assert node_header.position.x == 1
assert assert_vertex(node_header.position) == [1, 2, 3]
assert assert_quaternion(node_header.rotation) == [4, 5, 6, 7]
assert assert_array(node_header.child_offsets) == [1, 2, 3]
assert assert_array(node_header.controllers) == [2, 3, 4]
assert assert_array(node_header.controller_data) == [3, 4, 5]
def test_node_get_childs():
file = io.BytesIO(struct.pack("h", 0x21))
node = nodes.Node(file, tools.Block("root", 0, file))
node.childs.append('foo')
assert node.get_childs() == ['foo']