Make solvers deepcopyable

dweindl · Dec 17, 2023 · caeb9f0 · caeb9f0
1 parent 594b07e
commit caeb9f0
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/python/tests/test_swig_interface.py b/python/tests/test_swig_interface.py
@@ -471,3 +471,13 @@ def test_expdata_and_expdataview_are_deepcopyable():
     ev2 = copy.deepcopy(ev1)
     assert ev2._swigptr.this != ev1._swigptr.this
     assert ev1 == ev2
+
+
+def test_solvers_are_deepcopyable():
+    for solver_type in (amici.CVodeSolver, amici.IDASolver):
+        solver1 = solver_type()
+        solver2 = copy.deepcopy(solver1)
+        assert solver1.this != solver2.this
+        assert solver1.getRelativeTolerance() == solver2.getRelativeTolerance()
+        solver2.setRelativeTolerance(100 * solver2.getRelativeTolerance())
+        assert solver1.getRelativeTolerance() != solver2.getRelativeTolerance()
diff --git a/swig/solver.i b/swig/solver.i
@@ -116,6 +116,12 @@ def __repr__(self):
 %}
 };
 
+%extend amici::Solver {
+%pythoncode %{
+def __deepcopy__(self, memo):
+    return self.clone()
+%}
+};
 
 %newobject amici::Solver::clone;
 // Process symbols in header