diff --git a/tardis/montecarlo/montecarlo_numba/tests/test_base.py b/tardis/montecarlo/montecarlo_numba/tests/test_base.py index 3f1687d1cd7..0a91e6dad24 100644 --- a/tardis/montecarlo/montecarlo_numba/tests/test_base.py +++ b/tardis/montecarlo/montecarlo_numba/tests/test_base.py @@ -50,8 +50,12 @@ def test_montecarlo_main_loop( # Load compare data from refdata - expected_nu = expected_hdf_store["/simulation/transport/transport_state/output_nu"] - expected_energy = expected_hdf_store["/simulation/transport/transport_state/output_energy"] + expected_nu = expected_hdf_store[ + "/simulation/transport/transport_state/output_nu" + ] + expected_energy = expected_hdf_store[ + "/simulation/transport/transport_state/output_energy" + ] expected_nu_bar_estimator = expected_hdf_store[ "/simulation/transport/transport_state/nu_bar_estimator" ] @@ -98,8 +102,12 @@ def test_montecarlo_main_loop_vpacket_log( montecarlo_main_loop_simulation ) - expected_nu = expected_hdf_store["/simulation/transport/transport_state/output_nu"] - expected_energy = expected_hdf_store["/simulation/transport/transport_state/output_energy"] + expected_nu = expected_hdf_store[ + "/simulation/transport/transport_state/output_nu" + ] + expected_energy = expected_hdf_store[ + "/simulation/transport/transport_state/output_energy" + ] expected_nu_bar_estimator = expected_hdf_store[ "/simulation/transport/transport_state/nu_bar_estimator" ] diff --git a/tardis/tests/test_tardis_full_formal_integral.py b/tardis/tests/test_tardis_full_formal_integral.py index 06df740d333..f28203045f1 100644 --- a/tardis/tests/test_tardis_full_formal_integral.py +++ b/tardis/tests/test_tardis_full_formal_integral.py @@ -80,7 +80,9 @@ def test_j_blue_estimators(self, transport, refdata): ) def test_spectrum(self, transport, refdata): - luminosity = u.Quantity(refdata("transport_state/spectrum/luminosity"), "erg /s") + luminosity = u.Quantity( + refdata("transport_state/spectrum/luminosity"), "erg /s" + ) assert_quantity_allclose( transport.transport_state.spectrum.luminosity, luminosity