I had forgotten pytest fixtures on some tests

gsuarezr · Dec 11, 2023 · 67a7178 · 67a7178
1 parent 7cc95ec
commit 67a7178
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/qutip/solver/heom/bofin_baths.py b/qutip/solver/heom/bofin_baths.py
@@ -347,7 +347,7 @@ def power_spectrum(self, w, beta=None):
         return S
 
     def correlation_function(self, t):
-        """Computes the correlation function from 
+        """Computes the correlation function from
          the exponents"""
         corr = 0+0j
         for exp in self.exponents:

diff --git a/qutip/solver/heom/bofin_fit.py b/qutip/solver/heom/bofin_fit.py
@@ -187,7 +187,7 @@ def _matsubara_coefficients(self):
         return bath
 
     def correlation_function(self, t):
-        """Computes the correlation function from 
+        """Computes the correlation function from
          the exponents"""
         corr = 0+0j
         for exp in self.exponents:

diff --git a/qutip/tests/solver/heom/test_bofin_fit.py b/qutip/tests/solver/heom/test_bofin_fit.py
@@ -237,14 +237,14 @@ def test_matsubara_coefficients(self, correlation_fit):
             np.imag(fittedbath),
             atol=1e-5).all()  # one order below final_rmse
 
-    def test_power_spectrum_approx(self):
+    def test_power_spectrum_approx(self, correlation_fit):
         fc, ud, t, T = correlation_fit
         w = np.linspace(0.1, 10, 1000)
         S = fc.power_spectrum_approx(w)
         S2 = ud.power_spectrum(w, 1/T)
         np.isclose(S, S2, atol=1e-5).all()
 
-    def test_spectral_density_approx(self):
+    def test_spectral_density_approx(self, correlation_fit):
         fc, ud, t, T = correlation_fit
         w = np.linspace(0.1, 10, 1000)
         J = fc.spectral_density_approx(w, 1/T)