From 9d182c4ed0b049aa1d7295ef66f7925f63415781 Mon Sep 17 00:00:00 2001
From: Emil VATAI <emil.vatai@gmail.com>
Date: Fri, 16 Aug 2024 18:23:08 +0200
Subject: [PATCH] Fix slow tests

---
 tests/test_simulation.py | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/tests/test_simulation.py b/tests/test_simulation.py
index 5b2ffaa..9265ab7 100644
--- 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,7 +535,8 @@ 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),
@@ -543,7 +544,8 @@ def test_kinetics(self):
             ],
             **kwargs,
         )
-        results_jones_hore = self.sim.MARY(
+        results_jones_hore = mary(
+            self.sim,
             kinetics=[
                 kinetics.JonesHore(k, k),
                 kinetics.Exponential(k),