/ usr / lib / python3 / dist-packages / numpy / f2py / tests / [ HOME SHELL ]

Name	Size	Permission	Action
__pycache__	[ DIR ]	drwxr-xr-x
src	[ DIR ]	drwxr-xr-x
__init__.py	0 B	-rw-r--r--
test_abstract_interface.py	1.77 KB	-rw-r--r--
test_array_from_pyobj.py	22.28 KB	-rw-r--r--
test_assumed_shape.py	1.53 KB	-rw-r--r--
test_block_docstring.py	627 B	-rw-r--r--
test_callback.py	8 KB	-rw-r--r--
test_common.py	802 B	-rw-r--r--
test_compile_function.py	4.21 KB	-rw-r--r--
test_crackfortran.py	3.96 KB	-rw-r--r--
test_kind.py	1012 B	-rw-r--r--
test_mixed.py	911 B	-rw-r--r--
test_module_doc.py	950 B	-rw-r--r--
test_parameter.py	3.82 KB	-rw-r--r--
test_quoted_character.py	927 B	-rw-r--r--
test_regression.py	1.77 KB	-rw-r--r--
test_return_character.py	3.83 KB	-rw-r--r--
test_return_complex.py	4.51 KB	-rw-r--r--
test_return_integer.py	4.47 KB	-rw-r--r--
test_return_logical.py	4.73 KB	-rw-r--r--
test_return_real.py	5.28 KB	-rw-r--r--
test_semicolon_split.py	1.48 KB	-rw-r--r--
test_size.py	1.26 KB	-rw-r--r--
test_string.py	610 B	-rw-r--r--
util.py	9.36 KB	-rw-r--r--

Code Editor : test_compile_function.py

"""See https://github.com/numpy/numpy/pull/11937.

"""
import sys
import os
import uuid
from importlib import import_module
import pytest

import numpy.f2py

from numpy.testing import assert_equal
from . import util

def setup_module():
    if not util.has_c_compiler():
        pytest.skip("Needs C compiler")
    if not util.has_f77_compiler():
        pytest.skip('Needs FORTRAN 77 compiler')

# extra_args can be a list (since gh-11937) or string.
# also test absence of extra_args
@pytest.mark.parametrize(
    "extra_args", [['--noopt', '--debug'], '--noopt --debug', '']
    )
@pytest.mark.leaks_references(reason="Imported module seems never deleted.")
def test_f2py_init_compile(extra_args):
    # flush through the f2py __init__ compile() function code path as a
    # crude test for input handling following migration from
    # exec_command() to subprocess.check_output() in gh-11937

# the Fortran 77 syntax requires 6 spaces before any commands, but
    # more space may be added/
    fsource =  """
        integer function foo()
        foo = 10 + 5
        return
        end
    """
    # use various helper functions in util.py to enable robust build /
    # compile and reimport cycle in test suite
    moddir = util.get_module_dir()
    modname = util.get_temp_module_name()

cwd = os.getcwd()
    target = os.path.join(moddir, str(uuid.uuid4()) + '.f')
    # try running compile() with and without a source_fn provided so
    # that the code path where a temporary file for writing Fortran
    # source is created is also explored
    for source_fn in [target, None]:
        # mimic the path changing behavior used by build_module() in
        # util.py, but don't actually use build_module() because it has
        # its own invocation of subprocess that circumvents the
        # f2py.compile code block under test
        try:
            os.chdir(moddir)
            ret_val = numpy.f2py.compile(
                fsource,
                modulename=modname,
                extra_args=extra_args,
                source_fn=source_fn
                )
        finally:
            os.chdir(cwd)

# check for compile success return value
        assert_equal(ret_val, 0)

# we are not currently able to import the Python-Fortran
        # interface module on Windows / Appveyor, even though we do get
        # successful compilation on that platform with Python 3.x
        if sys.platform != 'win32':
            # check for sensible result of Fortran function; that means
            # we can import the module name in Python and retrieve the
            # result of the sum operation
            return_check = import_module(modname)
            calc_result = return_check.foo()
            assert_equal(calc_result, 15)
            # Removal from sys.modules, is not as such necessary. Even with
            # removal, the module (dict) stays alive.
            del sys.modules[modname]

def test_f2py_init_compile_failure():
    # verify an appropriate integer status value returned by
    # f2py.compile() when invalid Fortran is provided
    ret_val = numpy.f2py.compile(b"invalid")
    assert_equal(ret_val, 1)

def test_f2py_init_compile_bad_cmd():
    # verify that usage of invalid command in f2py.compile() returns
    # status value of 127 for historic consistency with exec_command()
    # error handling

# patch the sys Python exe path temporarily to induce an OSError
    # downstream NOTE: how bad of an idea is this patching?
    try:
        temp = sys.executable
        sys.executable = 'does not exist'

# the OSError should take precedence over invalid Fortran
        ret_val = numpy.f2py.compile(b"invalid")
        assert_equal(ret_val, 127)
    finally:
        sys.executable = temp

@pytest.mark.parametrize('fsource',
        ['program test_f2py\nend program test_f2py',
         b'program test_f2py\nend program test_f2py',])
def test_compile_from_strings(tmpdir, fsource):
    # Make sure we can compile str and bytes gh-12796
    cwd = os.getcwd()
    try:
        os.chdir(str(tmpdir))
        ret_val = numpy.f2py.compile(
                fsource,
                modulename='test_compile_from_strings',
                extension='.f90')
        assert_equal(ret_val, 0)
    finally:
        os.chdir(cwd)

Urban Research Lab

Urban Research Lab

The Urban Research Lab (URL) at IIIT-Delhi was established in 2020, during a time when the pandemic brought the complexities of urban life more visible than ever. What started as an initiative by faculty and students has grown into a collaborative space for critically exploring and reimagining the urban questions through research, dialogue, and interdisciplinary engagement.