Zy Filemanager

📁 File Browser

Name	Size	Permissions	Actions
📁 __pycache__	DIR	0755	🗑️
📁 src	DIR	0755	🗑️
📄 __init__.py (0 bytes)	0 B	0644	✍️ ⬇️ 🗑️
📄 test_abstract_interface.py (832 bytes)	832 B	0644	✍️ ⬇️ 🗑️
📄 test_array_from_pyobj.py (24,049 bytes)	23.49 KB	0644	✍️ ⬇️ 🗑️
📄 test_assumed_shape.py (1,466 bytes)	1.43 KB	0644	✍️ ⬇️ 🗑️
📄 test_block_docstring.py (564 bytes)	564 B	0644	✍️ ⬇️ 🗑️
📄 test_callback.py (6,152 bytes)	6.01 KB	0644	✍️ ⬇️ 🗑️
📄 test_character.py (20,513 bytes)	20.03 KB	0644	✍️ ⬇️ 🗑️
📄 test_common.py (584 bytes)	584 B	0644	✍️ ⬇️ 🗑️
📄 test_compile_function.py (4,186 bytes)	4.09 KB	0644	✍️ ⬇️ 🗑️
📄 test_crackfortran.py (13,078 bytes)	12.77 KB	0644	✍️ ⬇️ 🗑️
📄 test_docs.py (1,662 bytes)	1.62 KB	0644	✍️ ⬇️ 🗑️
📄 test_f2cmap.py (391 bytes)	391 B	0644	✍️ ⬇️ 🗑️
📄 test_f2py2e.py (21,570 bytes)	21.06 KB	0644	✍️ ⬇️ 🗑️
📄 test_kind.py (1,671 bytes)	1.63 KB	0644	✍️ ⬇️ 🗑️
📄 test_mixed.py (848 bytes)	848 B	0644	✍️ ⬇️ 🗑️
📄 test_module_doc.py (863 bytes)	863 B	0644	✍️ ⬇️ 🗑️
📄 test_parameter.py (3,941 bytes)	3.85 KB	0644	✍️ ⬇️ 🗑️
📄 test_quoted_character.py (454 bytes)	454 B	0644	✍️ ⬇️ 🗑️
📄 test_regression.py (2,157 bytes)	2.11 KB	0644	✍️ ⬇️ 🗑️
📄 test_return_character.py (1,493 bytes)	1.46 KB	0644	✍️ ⬇️ 🗑️
📄 test_return_complex.py (2,397 bytes)	2.34 KB	0644	✍️ ⬇️ 🗑️
📄 test_return_integer.py (1,850 bytes)	1.81 KB	0644	✍️ ⬇️ 🗑️
📄 test_return_logical.py (2,017 bytes)	1.97 KB	0644	✍️ ⬇️ 🗑️
📄 test_return_real.py (3,353 bytes)	3.27 KB	0644	✍️ ⬇️ 🗑️
📄 test_semicolon_split.py (1,635 bytes)	1.6 KB	0644	✍️ ⬇️ 🗑️
📄 test_size.py (1,164 bytes)	1.14 KB	0644	✍️ ⬇️ 🗑️
📄 test_string.py (2,962 bytes)	2.89 KB	0644	✍️ ⬇️ 🗑️
📄 test_symbolic.py (18,341 bytes)	17.91 KB	0644	✍️ ⬇️ 🗑️
📄 test_value_attrspec.py (323 bytes)	323 B	0644	✍️ ⬇️ 🗑️
📄 util.py (11,129 bytes)	10.87 KB	0644	✍️ ⬇️ 🗑️

✍️ Editing: test_callback.py

            File: /opt/cloudlinux/venv/lib/python3.11/site-packages/numpy/f2py/tests/test_callback.py

            Size: 6.01 KB | 
            Last Modified: 2026-01-20 20:01:48        

import math
import textwrap
import sys
import pytest
import threading
import traceback
import time

import numpy as np
from numpy.testing import IS_PYPY
from . import util

class TestF77Callback(util.F2PyTest):
    sources = [util.getpath("tests", "src", "callback", "foo.f")]

@pytest.mark.parametrize("name", "t,t2".split(","))
    def test_all(self, name):
        self.check_function(name)

@pytest.mark.xfail(IS_PYPY,
                       reason="PyPy cannot modify tp_doc after PyType_Ready")
    def test_docstring(self):
        expected = textwrap.dedent("""\
        a = t(fun,[fun_extra_args])

Wrapper for ``t``.

Parameters
        ----------
        fun : call-back function

Other Parameters
        ----------------
        fun_extra_args : input tuple, optional
            Default: ()

Returns
        -------
        a : int

Notes
        -----
        Call-back functions::

def fun(): return a
            Return objects:
                a : int
        """)
        assert self.module.t.__doc__ == expected

def check_function(self, name):
        t = getattr(self.module, name)
        r = t(lambda: 4)
        assert r == 4
        r = t(lambda a: 5, fun_extra_args=(6, ))
        assert r == 5
        r = t(lambda a: a, fun_extra_args=(6, ))
        assert r == 6
        r = t(lambda a: 5 + a, fun_extra_args=(7, ))
        assert r == 12
        r = t(lambda a: math.degrees(a), fun_extra_args=(math.pi, ))
        assert r == 180
        r = t(math.degrees, fun_extra_args=(math.pi, ))
        assert r == 180

r = t(self.module.func, fun_extra_args=(6, ))
        assert r == 17
        r = t(self.module.func0)
        assert r == 11
        r = t(self.module.func0._cpointer)
        assert r == 11

class A:
            def __call__(self):
                return 7

def mth(self):
                return 9

a = A()
        r = t(a)
        assert r == 7
        r = t(a.mth)
        assert r == 9

@pytest.mark.skipif(sys.platform == 'win32',
                        reason='Fails with MinGW64 Gfortran (Issue #9673)')
    def test_string_callback(self):
        def callback(code):
            if code == "r":
                return 0
            else:
                return 1

f = getattr(self.module, "string_callback")
        r = f(callback)
        assert r == 0

@pytest.mark.skipif(sys.platform == 'win32',
                        reason='Fails with MinGW64 Gfortran (Issue #9673)')
    def test_string_callback_array(self):
        # See gh-10027
        cu1 = np.zeros((1, ), "S8")
        cu2 = np.zeros((1, 8), "c")
        cu3 = np.array([""], "S8")

def callback(cu, lencu):
            if cu.shape != (lencu,):
                return 1
            if cu.dtype != "S8":
                return 2
            if not np.all(cu == b""):
                return 3
            return 0

f = getattr(self.module, "string_callback_array")
        for cu in [cu1, cu2, cu3]:
            res = f(callback, cu, cu.size)
            assert res == 0

def test_threadsafety(self):
        # Segfaults if the callback handling is not threadsafe

errors = []

def cb():
            # Sleep here to make it more likely for another thread
            # to call their callback at the same time.
            time.sleep(1e-3)

# Check reentrancy
            r = self.module.t(lambda: 123)
            assert r == 123

return 42

def runner(name):
            try:
                for j in range(50):
                    r = self.module.t(cb)
                    assert r == 42
                    self.check_function(name)
            except Exception:
                errors.append(traceback.format_exc())

threads = [
            threading.Thread(target=runner, args=(arg, ))
            for arg in ("t", "t2") for n in range(20)
        ]

for t in threads:
            t.start()

for t in threads:
            t.join()

errors = "\n\n".join(errors)
        if errors:
            raise AssertionError(errors)

def test_hidden_callback(self):
        try:
            self.module.hidden_callback(2)
        except Exception as msg:
            assert str(msg).startswith("Callback global_f not defined")

try:
            self.module.hidden_callback2(2)
        except Exception as msg:
            assert str(msg).startswith("cb: Callback global_f not defined")

self.module.global_f = lambda x: x + 1
        r = self.module.hidden_callback(2)
        assert r == 3

self.module.global_f = lambda x: x + 2
        r = self.module.hidden_callback(2)
        assert r == 4

del self.module.global_f
        try:
            self.module.hidden_callback(2)
        except Exception as msg:
            assert str(msg).startswith("Callback global_f not defined")

self.module.global_f = lambda x=0: x + 3
        r = self.module.hidden_callback(2)
        assert r == 5

# reproducer of gh18341
        r = self.module.hidden_callback2(2)
        assert r == 3

class TestF77CallbackPythonTLS(TestF77Callback):
    """
    Callback tests using Python thread-local storage instead of
    compiler-provided
    """

options = ["-DF2PY_USE_PYTHON_TLS"]

class TestF90Callback(util.F2PyTest):
    sources = [util.getpath("tests", "src", "callback", "gh17797.f90")]

def test_gh17797(self):
        def incr(x):
            return x + 123

y = np.array([1, 2, 3], dtype=np.int64)
        r = self.module.gh17797(incr, y)
        assert r == 123 + 1 + 2 + 3

class TestGH18335(util.F2PyTest):
    """The reproduction of the reported issue requires specific input that
    extensions may break the issue conditions, so the reproducer is
    implemented as a separate test class. Do not extend this test with
    other tests!
    """
    sources = [util.getpath("tests", "src", "callback", "gh18335.f90")]

def test_gh18335(self):
        def foo(x):
            x[0] += 1

r = self.module.gh18335(foo)
        assert r == 123 + 1

↩️ Back

Zy Filemanager

📁 File Browser

📤 Upload Files

Single File

Multiple Files

ZIP Extract

➕ Create New

New Folder

New File

✍️ Editing: test_callback.py

Backup Management