From 8008e2a192a85d7197660825598f6c22f3cc02a3 Mon Sep 17 00:00:00 2001
From: Atharva Arya <aryaatharva18@gmail.com>
Date: Mon, 19 Aug 2024 14:47:09 +0530
Subject: [PATCH] Run ruff and black

---
 tardis/simulation/tests/test_simulation.py | 34 +++++++++++-----------
 1 file changed, 17 insertions(+), 17 deletions(-)

diff --git a/tardis/simulation/tests/test_simulation.py b/tardis/simulation/tests/test_simulation.py
index a34d568f82c..f1920429ea2 100644
--- a/tardis/simulation/tests/test_simulation.py
+++ b/tardis/simulation/tests/test_simulation.py
@@ -1,21 +1,12 @@
-import os
-
-from pathlib import Path
-
+import astropy.units as u
+import numpy as np
+import pandas as pd
 import pytest
-import logging
 
+import tardis
 from tardis.io.configuration.config_reader import Configuration
 from tardis.simulation import Simulation
-from tardis import run_tardis
-import pandas.testing as pdt
 
-import numpy as np
-import pandas as pd
-
-import astropy.units as u
-import tardis
-from tardis.tests.fixtures.regression_data import RegressionData
 
 @pytest.fixture(scope="module")
 def config(example_configuration_dir):
@@ -23,6 +14,7 @@ def config(example_configuration_dir):
         example_configuration_dir / "tardis_configv1_verysimple.yml"
     )
 
+
 @pytest.fixture(scope="module")
 def simulation_one_loop(config, atomic_data_fname):
     config.atom_data = atomic_data_fname
@@ -35,6 +27,7 @@ def simulation_one_loop(config, atomic_data_fname):
     sim.run_final()
     return sim
 
+
 @pytest.mark.parametrize(
     "attr",
     [
@@ -52,6 +45,7 @@ def test_plasma_state_iterations(simulation_one_loop, attr, regression_data):
     expected = regression_data.sync_dataframe(actual)
     pd.testing.assert_frame_equal(actual, expected, rtol=1e-5, atol=1e-8)
 
+
 @pytest.mark.parametrize(
     "attr",
     [
@@ -65,14 +59,20 @@ def test_plasma_state_iterations(simulation_one_loop, attr, regression_data):
 )
 def test_plasma_estimates(simulation_one_loop, attr, regression_data):
     if attr in ["nu_bar_estimator", "j_estimator"]:
-        actual = getattr(simulation_one_loop.transport.transport_state.radfield_mc_estimators, attr)
+        actual = getattr(
+            simulation_one_loop.transport.transport_state.radfield_mc_estimators,
+            attr,
+        )
     elif attr in ["t_radiative", "dilution_factor"]:
         actual = getattr(simulation_one_loop.simulation_state, attr)
     elif attr in ["output_nus", "output_energies"]:
-        actual = getattr(simulation_one_loop.transport.transport_state.packet_collection, attr)
+        actual = getattr(
+            simulation_one_loop.transport.transport_state.packet_collection,
+            attr,
+        )
     else:
         actual = getattr(simulation_one_loop.transport, attr)
-    
+
     if hasattr(actual, "value"):
         actual = actual.value
     actual = pd.Series(actual)
@@ -122,4 +122,4 @@ def test_plasma_state_storer_reshape(
 
 
 def test_version_tag(simulation_without_loop):
-    assert simulation_without_loop.version == tardis.__version__
\ No newline at end of file
+    assert simulation_without_loop.version == tardis.__version__