Format using black

tardis-sn · Apr 1, 2024 · dc2cd8f · dc2cd8f
1 parent cc6d2ec
commit dc2cd8f
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 9 deletions.
diff --git a/tardis/simulation/tests/test_simulation.py b/tardis/simulation/tests/test_simulation.py
@@ -58,9 +58,7 @@ def simulation_one_loop(
             "t_radiative",
             "dilution_factor",
         ]
-        simulation.transport.hdf_properties = [
-            "transport_state"
-        ]
+        simulation.transport.hdf_properties = ["transport_state"]
         simulation.to_hdf(
             tardis_ref_data, "", "test_simulation", overwrite=True
         )
@@ -117,8 +115,6 @@ def test_plasma_estimates(simulation_one_loop, refdata, name):
     pdt.assert_series_equal(actual, refdata_keyname, rtol=1e-5, atol=1e-8)
 
 
-
-
 @pytest.mark.parametrize(
     "name",
     [

diff --git a/tardis/tests/test_tardis_full.py b/tardis/tests/test_tardis_full.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

diff --git a/tardis/tests/test_tardis_full_formal_integral.py b/tardis/tests/test_tardis_full_formal_integral.py
@@ -58,9 +58,7 @@ def transport(
         if not generate_reference:
             return simulation.transport
         else:
-            simulation.transport.hdf_properties = [
-                "transport_state"
-            ]
+            simulation.transport.hdf_properties = ["transport_state"]
             simulation.transport.to_hdf(
                 tardis_ref_data, "", self.name, overwrite=True
             )