| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657 |
- # 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)
- @may_xfail
- 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)
|
