84 lines
		
	
	
		
			2.2 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			84 lines
		
	
	
		
			2.2 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
import textwrap
 | 
						|
 | 
						|
import pytest
 | 
						|
 | 
						|
from numpy.testing import IS_PYPY
 | 
						|
 | 
						|
from . import util
 | 
						|
 | 
						|
 | 
						|
@pytest.mark.slow
 | 
						|
class TestModuleFilterPublicEntities(util.F2PyTest):
 | 
						|
    sources = [
 | 
						|
        util.getpath(
 | 
						|
            "tests", "src", "modules", "gh26920",
 | 
						|
            "two_mods_with_one_public_routine.f90"
 | 
						|
        )
 | 
						|
    ]
 | 
						|
    # we filter the only public function mod2
 | 
						|
    only = ["mod1_func1", ]
 | 
						|
 | 
						|
    def test_gh26920(self):
 | 
						|
        # if it compiles and can be loaded, things are fine
 | 
						|
        pass
 | 
						|
 | 
						|
 | 
						|
@pytest.mark.slow
 | 
						|
class TestModuleWithoutPublicEntities(util.F2PyTest):
 | 
						|
    sources = [
 | 
						|
        util.getpath(
 | 
						|
            "tests", "src", "modules", "gh26920",
 | 
						|
            "two_mods_with_no_public_entities.f90"
 | 
						|
        )
 | 
						|
    ]
 | 
						|
    only = ["mod1_func1", ]
 | 
						|
 | 
						|
    def test_gh26920(self):
 | 
						|
        # if it compiles and can be loaded, things are fine
 | 
						|
        pass
 | 
						|
 | 
						|
 | 
						|
@pytest.mark.slow
 | 
						|
class TestModuleDocString(util.F2PyTest):
 | 
						|
    sources = [util.getpath("tests", "src", "modules", "module_data_docstring.f90")]
 | 
						|
 | 
						|
    @pytest.mark.xfail(IS_PYPY, reason="PyPy cannot modify tp_doc after PyType_Ready")
 | 
						|
    def test_module_docstring(self):
 | 
						|
        assert self.module.mod.__doc__ == textwrap.dedent(
 | 
						|
            """\
 | 
						|
                     i : 'i'-scalar
 | 
						|
                     x : 'i'-array(4)
 | 
						|
                     a : 'f'-array(2,3)
 | 
						|
                     b : 'f'-array(-1,-1), not allocated\x00
 | 
						|
                     foo()\n
 | 
						|
                     Wrapper for ``foo``.\n\n"""
 | 
						|
        )
 | 
						|
 | 
						|
 | 
						|
@pytest.mark.slow
 | 
						|
class TestModuleAndSubroutine(util.F2PyTest):
 | 
						|
    module_name = "example"
 | 
						|
    sources = [
 | 
						|
        util.getpath("tests", "src", "modules", "gh25337", "data.f90"),
 | 
						|
        util.getpath("tests", "src", "modules", "gh25337", "use_data.f90"),
 | 
						|
    ]
 | 
						|
 | 
						|
    def test_gh25337(self):
 | 
						|
        self.module.data.set_shift(3)
 | 
						|
        assert "data" in dir(self.module)
 | 
						|
 | 
						|
 | 
						|
@pytest.mark.slow
 | 
						|
class TestUsedModule(util.F2PyTest):
 | 
						|
    module_name = "fmath"
 | 
						|
    sources = [
 | 
						|
        util.getpath("tests", "src", "modules", "use_modules.f90"),
 | 
						|
    ]
 | 
						|
 | 
						|
    def test_gh25867(self):
 | 
						|
        compiled_mods = [x for x in dir(self.module) if "__" not in x]
 | 
						|
        assert "useops" in compiled_mods
 | 
						|
        assert self.module.useops.sum_and_double(3, 7) == 20
 | 
						|
        assert "mathops" in compiled_mods
 | 
						|
        assert self.module.mathops.add(3, 7) == 10
 |