Initialisation du repository de Beta

venv/lib/python3.12/site-packages/sympy/codegen/tests/test_applications.py

@@ -0,0 +1,58 @@
+# 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, skip_under_pyodide
+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
+@skip_under_pyodide("Emscripten does not support process spawning")
+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)