format using black

tardis-sn · Apr 3, 2024 · d3e81b1 · d3e81b1
1 parent d28fc89
commit d3e81b1
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 5 deletions.
diff --git 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
@@ -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