MNT: Running ruff

Author: juliomachad0

rocketpy/motors/motor.py

@@ -1,6 +1,6 @@
 import re
-import xml.etree.ElementTree as ET
 import warnings
+import xml.etree.ElementTree as ET
 from abc import ABC, abstractmethod
 from functools import cached_property
 from os import path

rocketpy/stochastic/stochastic_aero_surfaces.py

@@ -4,12 +4,12 @@
 """
 
 from rocketpy.rocket.aero_surface import (
+    AirBrakes,
     EllipticalFins,
     NoseCone,
     RailButtons,
     Tail,
     TrapezoidalFins,
-    AirBrakes,
 )
 
 from .stochastic_model import StochasticModel

tests/integration/test_environment.py

@@ -28,9 +28,7 @@ def test_set_elevation_open_elevation(
 
 
 @patch("matplotlib.pyplot.show")
-def test_era5_atmosphere(
-    mock_show, example_spaceport_env
-):  # pylint: disable=unused-argument
+def test_era5_atmosphere(mock_show, example_spaceport_env):  # pylint: disable=unused-argument
     """Tests the Reanalysis model with the ERA5 file. It uses an example file
     available in the data/weather folder of the RocketPy repository.
 
@@ -51,9 +49,7 @@ def test_era5_atmosphere(
 
 
 @patch("matplotlib.pyplot.show")
-def test_custom_atmosphere(
-    mock_show, example_plain_env
-):  # pylint: disable=unused-argument
+def test_custom_atmosphere(mock_show, example_plain_env):  # pylint: disable=unused-argument
     """Tests the custom atmosphere model in the environment object.
 
     Parameters
@@ -78,9 +74,7 @@ def test_custom_atmosphere(
 
 
 @patch("matplotlib.pyplot.show")
-def test_standard_atmosphere(
-    mock_show, example_plain_env
-):  # pylint: disable=unused-argument
+def test_standard_atmosphere(mock_show, example_plain_env):  # pylint: disable=unused-argument
     """Tests the standard atmosphere model in the environment object.
 
     Parameters
@@ -132,9 +126,7 @@ def test_windy_atmosphere(example_euroc_env, model_name):
 
 @pytest.mark.slow
 @patch("matplotlib.pyplot.show")
-def test_gfs_atmosphere(
-    mock_show, example_spaceport_env
-):  # pylint: disable=unused-argument
+def test_gfs_atmosphere(mock_show, example_spaceport_env):  # pylint: disable=unused-argument
     """Tests the Forecast model with the GFS file. It does not test the values,
     instead the test checks if the method runs without errors.
 
@@ -151,9 +143,7 @@ def test_gfs_atmosphere(
 
 @pytest.mark.slow
 @patch("matplotlib.pyplot.show")
-def test_nam_atmosphere(
-    mock_show, example_spaceport_env
-):  # pylint: disable=unused-argument
+def test_nam_atmosphere(mock_show, example_spaceport_env):  # pylint: disable=unused-argument
     """Tests the Forecast model with the NAM file.
 
     Parameters
@@ -169,9 +159,7 @@ def test_nam_atmosphere(
 
 @pytest.mark.slow
 @patch("matplotlib.pyplot.show")
-def test_rap_atmosphere(
-    mock_show, example_spaceport_env
-):  # pylint: disable=unused-argument
+def test_rap_atmosphere(mock_show, example_spaceport_env):  # pylint: disable=unused-argument
     today = date.today()
     now = datetime.now(timezone.utc)
     example_spaceport_env.set_date((today.year, today.month, today.day, now.hour))
@@ -181,9 +169,7 @@ def test_rap_atmosphere(
 
 @pytest.mark.slow
 @patch("matplotlib.pyplot.show")
-def test_gefs_atmosphere(
-    mock_show, example_spaceport_env
-):  # pylint: disable=unused-argument
+def test_gefs_atmosphere(mock_show, example_spaceport_env):  # pylint: disable=unused-argument
     """Tests the Ensemble model with the GEFS file.
 
     Parameters
@@ -199,9 +185,7 @@ def test_gefs_atmosphere(
 
 @pytest.mark.slow
 @patch("matplotlib.pyplot.show")
-def test_wyoming_sounding_atmosphere(
-    mock_show, example_plain_env
-):  # pylint: disable=unused-argument
+def test_wyoming_sounding_atmosphere(mock_show, example_plain_env):  # pylint: disable=unused-argument
     """Asserts whether the Wyoming sounding model in the environment
     object behaves as expected with respect to some attributes such
     as pressure, barometric_height, wind_velocity and temperature.
@@ -236,9 +220,7 @@ def test_wyoming_sounding_atmosphere(
 
 @pytest.mark.slow
 @patch("matplotlib.pyplot.show")
-def test_hiresw_ensemble_atmosphere(
-    mock_show, example_spaceport_env
-):  # pylint: disable=unused-argument
+def test_hiresw_ensemble_atmosphere(mock_show, example_spaceport_env):  # pylint: disable=unused-argument
     """Tests the Forecast model with the HIRESW file.
 
     Parameters
@@ -264,9 +246,7 @@ def test_hiresw_ensemble_atmosphere(
 
 @pytest.mark.skip(reason="CMC model is currently not working")
 @patch("matplotlib.pyplot.show")
-def test_cmc_atmosphere(
-    mock_show, example_spaceport_env
-):  # pylint: disable=unused-argument
+def test_cmc_atmosphere(mock_show, example_spaceport_env):  # pylint: disable=unused-argument
     """Tests the Ensemble model with the CMC file.
 
     Parameters

tests/integration/test_flight.py

@@ -195,9 +195,7 @@ def test_export_pressures(flight_calisto_robust):
 
 
 @patch("matplotlib.pyplot.show")
-def test_hybrid_motor_flight(
-    mock_show, flight_calisto_hybrid_modded
-):  # pylint: disable=unused-argument
+def test_hybrid_motor_flight(mock_show, flight_calisto_hybrid_modded):  # pylint: disable=unused-argument
     """Test the flight of a rocket with a hybrid motor. This test only validates
     that a flight simulation can be performed with a hybrid motor; it does not
     validate the results.
@@ -213,9 +211,7 @@ def test_hybrid_motor_flight(
 
 
 @patch("matplotlib.pyplot.show")
-def test_liquid_motor_flight(
-    mock_show, flight_calisto_liquid_modded
-):  # pylint: disable=unused-argument
+def test_liquid_motor_flight(mock_show, flight_calisto_liquid_modded):  # pylint: disable=unused-argument
     """Test the flight of a rocket with a liquid motor. This test only validates
     that a flight simulation can be performed with a liquid motor; it does not
     validate the results.
@@ -232,9 +228,7 @@ def test_liquid_motor_flight(
 
 @pytest.mark.slow
 @patch("matplotlib.pyplot.show")
-def test_time_overshoot(
-    mock_show, calisto_robust, example_spaceport_env
-):  # pylint: disable=unused-argument
+def test_time_overshoot(mock_show, calisto_robust, example_spaceport_env):  # pylint: disable=unused-argument
     """Test the time_overshoot parameter of the Flight class. This basically
     calls the all_info() method for a simulation without time_overshoot and
     checks if it returns None. It is not testing if the values are correct,
@@ -263,9 +257,7 @@ def test_time_overshoot(
 
 
 @patch("matplotlib.pyplot.show")
-def test_simpler_parachute_triggers(
-    mock_show, example_plain_env, calisto_robust
-):  # pylint: disable=unused-argument
+def test_simpler_parachute_triggers(mock_show, example_plain_env, calisto_robust):  # pylint: disable=unused-argument
     """Tests different types of parachute triggers. This is important to ensure
     the code is working as intended, since the parachute triggers can have very
     different format definitions. It will add 3 parachutes using different
@@ -407,9 +399,7 @@ def test_eccentricity_on_flight(  # pylint: disable=unused-argument
 
 
 @patch("matplotlib.pyplot.show")
-def test_air_brakes_flight(
-    mock_show, flight_calisto_air_brakes
-):  # pylint: disable=unused-argument
+def test_air_brakes_flight(mock_show, flight_calisto_air_brakes):  # pylint: disable=unused-argument
     """Test the flight of a rocket with air brakes. This test only validates
     that a flight simulation can be performed with air brakes; it does not
     validate the results.
@@ -429,9 +419,7 @@ def test_air_brakes_flight(
 
 
 @patch("matplotlib.pyplot.show")
-def test_initial_solution(
-    mock_show, example_plain_env, calisto_robust
-):  # pylint: disable=unused-argument
+def test_initial_solution(mock_show, example_plain_env, calisto_robust):  # pylint: disable=unused-argument
     """Tests the initial_solution option of the Flight class. This test simply
     simulates the flight using the initial_solution option and checks if the
     all_info method returns None.
@@ -476,9 +464,7 @@ def test_initial_solution(
 
 
 @patch("matplotlib.pyplot.show")
-def test_empty_motor_flight(
-    mock_show, example_plain_env, calisto_motorless
-):  # pylint: disable=unused-argument
+def test_empty_motor_flight(mock_show, example_plain_env, calisto_motorless):  # pylint: disable=unused-argument
     flight = Flight(
         rocket=calisto_motorless,
         environment=example_plain_env,

tests/integration/test_flight_data_importer.py

@@ -26,9 +26,9 @@ def test_flight_importer_bella_lui():
     )
     assert fd.name == "Bella Lui, EPFL Rocket Team, 2020"
     assert "time" in fd._columns[path], "Can't find 'time' column in fd._columns"
-    assert (
-        "altitude" in fd._columns[path]
-    ), "Can't find 'altitude' column in fd._columns"
+    assert "altitude" in fd._columns[path], (
+        "Can't find 'altitude' column in fd._columns"
+    )
     assert "vz" in fd._columns[path], "Can't find 'vz' column in fd._columns"
     assert np.isclose(fd.altitude(0), 0.201, atol=1e-4)
     assert np.isclose(fd.vz(0), 5.028, atol=1e-4)
@@ -51,7 +51,7 @@ def test_flight_importer_ndrt():
     )
     assert fd.name == "NDRT Rocket team, 2020"
     assert "time" in fd._columns[path], "Can't find 'time' column in fd._columns"
-    assert (
-        "altitude" in fd._columns[path]
-    ), "Can't find 'altitude' column in fd._columns"
+    assert "altitude" in fd._columns[path], (
+        "Can't find 'altitude' column in fd._columns"
+    )
     assert np.isclose(fd.altitude(0), 0)

tests/integration/test_rocket.py

@@ -45,9 +45,7 @@ def test_airfoil(
 
 
 @patch("matplotlib.pyplot.show")
-def test_air_brakes_clamp_on(
-    mock_show, calisto_air_brakes_clamp_on
-):  # pylint: disable=unused-argument
+def test_air_brakes_clamp_on(mock_show, calisto_air_brakes_clamp_on):  # pylint: disable=unused-argument
     """Test the air brakes class with clamp on configuration. This test checks
     the basic attributes and the deployment_level setter. It also checks the
     all_info method.

tests/integration/test_sensor.py

@@ -89,9 +89,7 @@ def test_gnss_receiver(self):
 
 @pytest.mark.parametrize("plane", ["xz", "yz"])
 @patch("matplotlib.pyplot.show")
-def test_draw(
-    mock_show, calisto_with_sensors, plane
-):  # pylint: disable=unused-argument
+def test_draw(mock_show, calisto_with_sensors, plane):  # pylint: disable=unused-argument
     """Test the drawing of the sensors."""
     calisto_with_sensors.draw(plane=plane)