mirror of https://github.com/Nonannet/copapy.git
89 lines
2.3 KiB
Python
89 lines
2.3 KiB
Python
|
from copapy import variable, NumLike
|
||
|
|
from copapy.backend import Write, compile_to_dag, add_read_command
|
||
|
|
import subprocess
|
||
|
|
import struct
|
||
|
|
from copapy import _binwrite
|
||
|
|
import copapy.backend as backend
|
||
|
|
import os
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
if os.name == 'nt':
|
||
|
|
# On Windows wsl and qemu-user is required:
|
||
|
|
# sudo apt install qemu-user
|
||
|
|
qemu_command = ['wsl', 'qemu-aarch64', 'bin/coparun-aarch64', 'bin/test.copapy']
|
||
|
|
else:
|
||
|
|
qemu_command = ['qemu-aarch64', 'bin/coparun-aarch64', 'bin/test.copapy']
|
||
|
|
|
||
|
|
|
||
|
|
def run_command(command: list[str]) -> str:
|
||
|
|
result = subprocess.run(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE, encoding='utf8', check=False)
|
||
|
|
assert result.returncode != 11, f"SIGSEGV (segmentation fault)\n -Error occurred: {result.stderr}\n -Output: {result.stdout}"
|
||
|
|
assert result.returncode == 0, f"\n -Error occurred: {result.stderr}\n -Output: {result.stdout}"
|
||
|
|
return result.stdout
|
||
|
|
|
||
|
|
|
||
|
|
def test_example():
|
||
|
|
c1 = 4
|
||
|
|
c2 = 2
|
||
|
|
|
||
|
|
i1 = c1 * 2
|
||
|
|
r1 = i1 + 7 + (c2 + 7 * 9)
|
||
|
|
r2 = i1 + 9
|
||
|
|
|
||
|
|
en = {'little': '<', 'big': '>'}['little']
|
||
|
|
data = struct.pack(en + 'i', r1)
|
||
|
|
print("example r1 " + ' '.join(f'{b:02X}' for b in data))
|
||
|
|
|
||
|
|
data = struct.pack(en + 'i', r2)
|
||
|
|
print("example r2 " + ' '.join(f'{b:02X}' for b in data))
|
||
|
|
|
||
|
|
|
||
|
|
def function(c1: NumLike, c2: NumLike) -> tuple[NumLike, ...]:
|
||
|
|
i1 = c1 // 3.3 + 5
|
||
|
|
i2 = c2 * 5 + c1
|
||
|
|
r1 = i1 + i2 * 55 / 4
|
||
|
|
r2 = 4 * i2 + 5
|
||
|
|
|
||
|
|
return i1, i2, r1, r2
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.skip(reason="no way of currently testing this")
|
||
|
|
def test_compile():
|
||
|
|
c1 = variable(4)
|
||
|
|
c2 = variable(2)
|
||
|
|
|
||
|
|
ret = function(c1, c2)
|
||
|
|
#ret = [c1 // 3.3 + 5]
|
||
|
|
|
||
|
|
out = [Write(r) for r in ret]
|
||
|
|
|
||
|
|
sdb = backend.stencil_db_from_package('aarch64', 'O3')
|
||
|
|
il, variables = compile_to_dag(out, sdb)
|
||
|
|
|
||
|
|
# run program command
|
||
|
|
il.write_com(_binwrite.Command.RUN_PROG)
|
||
|
|
|
||
|
|
for net in ret:
|
||
|
|
assert isinstance(net, backend.Net)
|
||
|
|
add_read_command(il, variables, net)
|
||
|
|
|
||
|
|
il.write_com(_binwrite.Command.END_COM)
|
||
|
|
|
||
|
|
print('* Data to runner:')
|
||
|
|
il.print()
|
||
|
|
|
||
|
|
il.to_file('bin/test.copapy')
|
||
|
|
|
||
|
|
result = run_command(qemu_command)
|
||
|
|
print('* Output from runner:\n--')
|
||
|
|
print(result)
|
||
|
|
print('--')
|
||
|
|
|
||
|
|
assert 'Return value: 1' in result
|
||
|
|
#assert 'END_COM' in result
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
#test_example()
|
||
|
|
test_compile()
|