mirror of https://github.com/Nonannet/copapy.git
test for stencil math function sqrt added
This commit is contained in:
parent
8214689714
commit
4cb91be02b
|
|
@ -0,0 +1,53 @@
|
|||
from copapy import variable
|
||||
from copapy.backend import Write, compile_to_dag, add_read_command
|
||||
import copapy as cp
|
||||
import subprocess
|
||||
from copapy import _binwrite
|
||||
import copapy.backend
|
||||
|
||||
|
||||
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():
|
||||
test_vals = [0.0, 0.0001, 0.1, 0.5, 1.0, 1.5, 2.0, 2.5, 3.0, 3.5, 4.0, 4.5, 5.0, 5.5, 6.0, 6.28318530718, 100.0, 1000.0, 100000.0]
|
||||
|
||||
test_vals = [100000.0]
|
||||
|
||||
ret = [r for v in test_vals for r in (cp.sqrt(variable(v)),)]
|
||||
|
||||
|
||||
out = [Write(r) for r in ret]
|
||||
|
||||
il, variables = compile_to_dag(out, copapy.generic_sdb)
|
||||
|
||||
# run program command
|
||||
il.write_com(_binwrite.Command.RUN_PROG)
|
||||
|
||||
for net in ret:
|
||||
assert isinstance(net, copapy.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(['bin/coparun', 'bin/test.copapy'])
|
||||
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()
|
||||
Loading…
Reference in New Issue