Reset return_var's when making a new call

This is needed to make a new ctype so we don't overwrite the old one
rjfarmer · Feb 29, 2024 · 0776fd3 · 0776fd3
1 parent b8497f9
commit 0776fd3
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 4 deletions.
diff --git a/gfort2py/fProc.py b/gfort2py/fProc.py
@@ -52,6 +52,9 @@ def name(self):
         return self.obj.name
 
     def __call__(self, *args, **kwargs):
+        # Reset return value
+        self._return_value = None
+
         self._func_args = self._convert_args(*args, **kwargs)
 
         # print(self._func_args)

diff --git a/tests/gh56_test.py b/tests/gh56_test.py
@@ -17,7 +17,6 @@
 x = gf.fFort(SO, MOD)
 
 
-@pytest.mark.skip("Not fixed yet")
 class Testgh56Methods:
     def test_gh56(self, capfd):
         y1 = x.get_array(10).result
@@ -27,8 +26,13 @@ def test_gh56(self, capfd):
         assert len(y1["d"]) == 10
         assert len(y2["d"]) == 5
 
-        assert y1["d"] == np.array(
-            [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0], dtype=np.float32
+        assert np.array_equal(
+            y1["d"],
+            np.array(
+                [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0], dtype=np.float32
+            ),
         )
 
-        assert y2["d"] == np.array([1.0, 2.0, 3.0, 4.0, 5.0], dtype=np.float32)
+        assert np.array_equal(
+            y2["d"], np.array([1.0, 2.0, 3.0, 4.0, 5.0], dtype=np.float32)
+        )