mirror of https://github.com/Nonannet/copapy.git
114 lines
2.9 KiB
Python
114 lines
2.9 KiB
Python
from copapy import variable
|
|
from copapy.backend import Write, compile_to_dag, stencil_db_from_package
|
|
import subprocess
|
|
import copapy as cp
|
|
from copapy._binwrite import Command
|
|
import os
|
|
|
|
|
|
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-aarch64.copapy',
|
|
'bin/test-aarch64.copapy.bin']
|
|
else:
|
|
qemu_command = ['qemu-aarch64',
|
|
'bin/coparun-aarch64',
|
|
'bin/test-aarch64.copapy',
|
|
'bin/test-aarch64.copapy.bin']
|
|
|
|
|
|
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_compile() -> None:
|
|
"""Test compilation of a simple program."""
|
|
c1 = variable(9.0)
|
|
|
|
#ret = [c1 / 4, c1 / -4, c1 // 4, c1 // -4, (c1 * -1) // 4]
|
|
ret = [c1 // 3.3 + 5]
|
|
#ret = [cp.sqrt(c1)]
|
|
#c2 = cp._math.get_42()
|
|
#ret = [c2]
|
|
|
|
out = [Write(r) for r in ret]
|
|
|
|
dw, vars = compile_to_dag(out, cp.generic_sdb)
|
|
|
|
# run program command
|
|
dw.write_com(Command.RUN_PROG)
|
|
|
|
# read first 32 byte
|
|
dw.write_com(Command.READ_DATA)
|
|
dw.write_int(0)
|
|
dw.write_int(32)
|
|
|
|
# read variables
|
|
for addr, lengths, _ in vars.values():
|
|
dw.write_com(Command.READ_DATA)
|
|
dw.write_int(addr)
|
|
dw.write_int(lengths)
|
|
|
|
dw.write_com(Command.END_COM)
|
|
|
|
print('* Data to runner:')
|
|
dw.print()
|
|
|
|
dw.to_file('bin/test.copapy')
|
|
|
|
|
|
def test_compile_aarch64() -> None:
|
|
"""Test compilation of a simple program."""
|
|
c1 = variable(9.0)
|
|
|
|
#ret = [c1 / 4, c1 / -4, c1 // 4, c1 // -4, (c1 * -1) // 4]
|
|
#ret = [cp.sin(c1), cp.sqrt(c1) + 5]
|
|
ret = [cp.sqrt(c1)]
|
|
#c2 = cp._math.get_42()
|
|
#ret = [c2]
|
|
|
|
out = [Write(r) for r in ret]
|
|
|
|
sdb = stencil_db_from_package('aarch64')
|
|
dw, vars = compile_to_dag(out, sdb)
|
|
|
|
# run program command
|
|
#dw.write_com(Command.RUN_PROG)
|
|
|
|
# read first 32 byte
|
|
#dw.write_com(Command.READ_DATA)
|
|
#dw.write_int(0)
|
|
#dw.write_int(32)
|
|
|
|
# read variables
|
|
#for addr, lengths, _ in vars.values():
|
|
# dw.write_com(Command.READ_DATA)
|
|
# dw.write_int(addr)
|
|
# dw.write_int(lengths)
|
|
|
|
#dw.write_com(Command.END_COM)
|
|
dw.write_com(Command.DUMP_CODE)
|
|
|
|
print('* Data to runner:')
|
|
dw.print()
|
|
|
|
dw.to_file('bin/test-aarch64.copapy')
|
|
|
|
result = run_command(qemu_command)
|
|
print('* Output from runner:\n--')
|
|
print(result)
|
|
print('--')
|
|
|
|
assert 'DUMP_CODE' in result
|
|
|
|
|
|
if __name__ == "__main__":
|
|
#test_compile()
|
|
test_compile_aarch64()
|