No object is returned

tardis-sn · Oct 16, 2024 · 876026f · 876026f
1 parent f379754
commit 876026f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/tardis/spectrum/tests/test_spectrum_solver.py b/tardis/spectrum/tests/test_spectrum_solver.py
@@ -34,8 +34,7 @@ def simulation(
         request.cls.regression_data = RegressionData(request)
         request.cls.regression_data.sync_hdf_store(simulation)
 
-        yield simulation
-        request.cls.regression_data.close()
+        return simulation
 
     def get_expected_data(self, key: str):
         return pd.read_hdf(self.regression_data.fpath, key)

diff --git a/tardis/tests/test_tardis_full.py b/tardis/tests/test_tardis_full.py
@@ -57,8 +57,7 @@ def simulation(
         request.cls.regression_data = RegressionData(request)
         request.cls.regression_data.sync_hdf_store(simulation)
 
-        yield simulation
-        request.cls.regression_data.close()
+        return simulation
 
     def get_expected_data(self, key: str):
         return pd.read_hdf(self.regression_data.fpath, key)