Spaces:
Sleeping
Sleeping
| # This file contains tests that exercise multiple AST nodes | |
| import tempfile | |
| from sympy.external import import_module | |
| from sympy.printing.codeprinter import ccode | |
| from sympy.utilities._compilation import compile_link_import_strings, has_c | |
| from sympy.utilities._compilation.util import may_xfail | |
| from sympy.testing.pytest import skip | |
| from sympy.codegen.ast import ( | |
| FunctionDefinition, FunctionPrototype, Variable, Pointer, real, Assignment, | |
| integer, CodeBlock, While | |
| ) | |
| from sympy.codegen.cnodes import void, PreIncrement | |
| from sympy.codegen.cutils import render_as_source_file | |
| cython = import_module('cython') | |
| np = import_module('numpy') | |
| def _mk_func1(): | |
| declars = n, inp, out = Variable('n', integer), Pointer('inp', real), Pointer('out', real) | |
| i = Variable('i', integer) | |
| whl = While(i<n, [Assignment(out[i], inp[i]), PreIncrement(i)]) | |
| body = CodeBlock(i.as_Declaration(value=0), whl) | |
| return FunctionDefinition(void, 'our_test_function', declars, body) | |
| def _render_compile_import(funcdef, build_dir): | |
| code_str = render_as_source_file(funcdef, settings={"contract": False}) | |
| declar = ccode(FunctionPrototype.from_FunctionDefinition(funcdef)) | |
| return compile_link_import_strings([ | |
| ('our_test_func.c', code_str), | |
| ('_our_test_func.pyx', ("#cython: language_level={}\n".format("3") + | |
| "cdef extern {declar}\n" | |
| "def _{fname}({typ}[:] inp, {typ}[:] out):\n" | |
| " {fname}(inp.size, &inp[0], &out[0])").format( | |
| declar=declar, fname=funcdef.name, typ='double' | |
| )) | |
| ], build_dir=build_dir) | |
| def test_copying_function(): | |
| if not np: | |
| skip("numpy not installed.") | |
| if not has_c(): | |
| skip("No C compiler found.") | |
| if not cython: | |
| skip("Cython not found.") | |
| info = None | |
| with tempfile.TemporaryDirectory() as folder: | |
| mod, info = _render_compile_import(_mk_func1(), build_dir=folder) | |
| inp = np.arange(10.0) | |
| out = np.empty_like(inp) | |
| mod._our_test_function(inp, out) | |
| assert np.allclose(inp, out) | |