Fix slow tests

Spin-Chemistry-Labs · Aug 16, 2024 · 9d182c4 · 9d182c4
1 parent 526f5e6
commit 9d182c4
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/tests/test_simulation.py b/tests/test_simulation.py
@@ -401,8 +401,8 @@ def test_ST_vs_Zeeman_basis(self):
                     J=PARAMS["J"],
                     kinetics=[kinetics.Exponential(k)],
                 )
-                rslt = self.sim.MARY(**kwargs)
-                strs = st_sim.MARY(**kwargs)
+                rslt = mary(self.sim, **kwargs)
+                strs = mary(st_sim, **kwargs)
 
                 key = "time_evolutions"
                 Bi = 1
@@ -535,15 +535,17 @@ def test_kinetics(self):
             J=0,
         )
         k = 1e6
-        results_haberkorn = self.sim.MARY(
+        results_haberkorn = mary(
+            self.sim,
             kinetics=[
                 kinetics.Haberkorn(k, rp.simulation.State.TRIPLET),
                 kinetics.Haberkorn(k, rp.simulation.State.SINGLET),
                 kinetics.Exponential(k),
             ],
             **kwargs,
         )
-        results_jones_hore = self.sim.MARY(
+        results_jones_hore = mary(
+            self.sim,
             kinetics=[
                 kinetics.JonesHore(k, k),
                 kinetics.Exponential(k),