import os import pytest import tempfile from . import util class TestAssumedShapeSumExample(util.F2PyTest): sources = [ util.getpath("tests", "src", "assumed_shape", "foo_free.f90"), util.getpath("tests", "src", "assumed_shape", "foo_use.f90"), util.getpath("tests", "src", "assumed_shape", "precision.f90"), util.getpath("tests", "src", "assumed_shape", "foo_mod.f90"), util.getpath("tests", "src", "assumed_shape", ".f2py_f2cmap"), ] @pytest.mark.slow def test_all(self): r = self.module.fsum([1, 2]) assert r == 3 r = self.module.sum([1, 2]) assert r == 3 r = self.module.sum_with_use([1, 2]) assert r == 3 r = self.module.mod.sum([1, 2]) assert r == 3 r = self.module.mod.fsum([1, 2]) assert r == 3 class TestF2cmapOption(TestAssumedShapeSumExample): def setup_method(self): # Use a custom file name for .f2py_f2cmap self.sources = list(self.sources) f2cmap_src = self.sources.pop(-1) self.f2cmap_file = tempfile.NamedTemporaryFile(delete=False) with open(f2cmap_src, "rb") as f: self.f2cmap_file.write(f.read()) self.f2cmap_file.close() self.sources.append(self.f2cmap_file.name) self.options = ["--f2cmap", self.f2cmap_file.name] super().setup_method() def teardown_method(self): os.unlink(self.f2cmap_file.name)