NewContext: ensure NumPy preference is propagated to new contexts.

PMeira · PMeira · commit d7ec602b8eb8 · 2024-02-12T09:47:52.000-03:00
diff --git a/opendssdirect/OpenDSSDirect.py b/opendssdirect/OpenDSSDirect.py
@@ -352,7 +352,8 @@ def NewContext(self):
         new_ctx = ffi.gc(lib.ctx_New(), lib.ctx_Dispose)
         new_api_util = dss_py.CffiApiUtil(ffi, lib, new_ctx)
         new_api_util._allow_complex = self._api_util._allow_complex
-        return OpenDSSDirect(new_api_util)
+        prefer_lists = (self._get_complex128_array == self._api_util.get_complex128_array2)
+        return OpenDSSDirect(new_api_util, prefer_lists=prefer_lists)
 
     def to_dss_python(self) -> DSSPython:
         """
diff --git a/tests/test_opendssdirect.py b/tests/test_opendssdirect.py
@@ -5564,6 +5564,23 @@ def test_numpy():
     assert isinstance(odd_lst.Circuit.AllBusMagPu(), list)
 
 
+def test_numpy_option_propagation():
+    import_odd()
+    from opendssdirect.OpenDSSDirect import OpenDSSDirect
+    from numpy import ndarray
+
+    # NOTE: this constructors ALWAYS binds to the default DSS engine.
+    odd_np = OpenDSSDirect(prefer_lists=False)
+    # Use it normally
+    odd_np(f"Redirect '{PATH_TO_DSS}'")
+    assert isinstance(odd_np.Circuit.AllBusMagPu(), ndarray)
+
+    odd_np2 = odd_np.NewContext()
+    odd_np2(f"Redirect '{PATH_TO_DSS}'")
+    assert isinstance(odd_np2.Circuit.AllBusMagPu(), ndarray)
+
+
+
 def test_threading2(dss):
     # Ported directly from DSS-Python, but using only the 13Bus circuit
     from opendssdirect.OpenDSSDirect import OpenDSSDirect
