diff --git a/tardis/tests/integration_tests/test_integration.py b/tardis/tests/integration_tests/test_integration.py index 9aae36d17e1..82c1d0407bc 100644 --- a/tardis/tests/integration_tests/test_integration.py +++ b/tardis/tests/integration_tests/test_integration.py @@ -120,8 +120,8 @@ def setup(self, request, reference, data_path, pytestconfig): # We now do a run with prepared config and get the simulation object. self.result = Simulation.from_config(tardis_config, atom_data=self.atom_data) + capmanager.suspend_global_capture(True) - capmanager.suspendcapture(True) # If current test run is just for collecting reference data, store the # output model to HDF file, save it at specified path. Skip all tests. # Else simply perform the run and move further for performing @@ -139,7 +139,7 @@ def setup(self, request, reference, data_path, pytestconfig): pytest.skip("Reference data saved at {0}".format( data_path['reference_path'] )) - capmanager.resumecapture() + capmanager.resume_global_capture() # Get the reference data through the fixture. self.reference = reference @@ -234,4 +234,4 @@ def plot_spectrum(self): deviation, color='black') deviation_ax.set_xlabel("Wavelength [Angstrom]") - return plt.gcf() + return plt.gcf() \ No newline at end of file