TST: Multiple modules in single pyf for gh-27622

numpy · charris · Nov 17, 2024 · Nov 4, 2024 · Nov 4, 2024 · Nov 17, 2024
commit 9ab12dcbce38a06f16e54c82f00baef4e5e9f7dc
diff --git a/doc/release/upcoming_changes/27695.improvement.rst b/doc/release/upcoming_changes/27695.improvement.rst
@@ -0,0 +1,5 @@
+``f2py`` handles multiple modules and exposes variables again
+-------------------------------------------------------------
+A regression has been fixed which allows F2PY users to expose variables to
+Python in modules with only assignments, and also fixes situations where
+multiple modules are present within a single source file.
diff --git a/numpy/f2py/tests/src/regression/datonly.f90 b/numpy/f2py/tests/src/regression/datonly.f90
@@ -0,0 +1,17 @@
+module datonly
+  implicit none
+  integer, parameter :: max_value = 100
+  real, dimension(:), allocatable :: data_array
+end module datonly
+
+module dat
+  implicit none
+  integer, parameter :: max_= 1009
+end module dat
+
+subroutine simple_subroutine(ain, aout)
+  use dat, only: max_
+  integer, intent(in) :: ain
+  integer, intent(out) :: aout
+  aout = ain + max_
+end subroutine simple_subroutine
diff --git a/numpy/f2py/tests/test_regression.py b/numpy/f2py/tests/test_regression.py
@@ -24,6 +24,18 @@ def test_inout(self):
        assert np.allclose(x, [3, 1, 2])


+class TestDataOnlyMultiModule(util.F2PyTest):
+    # Check that modules without subroutines work
+    sources = [util.getpath("tests", "src", "regression", "datonly.f90")]
+
+    @pytest.mark.slow
+    def test_mdat(self):
+        assert self.module.datonly.max_value == 100
+        assert self.module.dat.max_ == 1009
+        int_in = 5
+        assert self.module.simple_subroutine(5) == 1014
+
+
 class TestNegativeBounds(util.F2PyTest):
    # Check that negative bounds work correctly
    sources = [util.getpath("tests", "src", "negative_bounds", "issue_20853.f90")]