MNT: ruff pylint

Author: MateusStano

rocketpy/prints/aero_surface_prints.py

@@ -1,5 +1,6 @@
 from abc import ABC, abstractmethod
 
+
 class _AeroSurfacePrints(ABC):
     def __init__(self, aero_surface):
         self.aero_surface = aero_surface

rocketpy/rocket/aero_surface/fins/_geometry.py

@@ -89,11 +89,7 @@ def evaluate_geometrical_parameters(self):
         Af = Yr * owner.span / 2
         AR = 2 * owner.span**2 / Af
         gamma_c = np.arctan(
-            (
-                self.sweep_length
-                + 0.5 * self.tip_chord
-                - 0.5 * owner.root_chord
-            )
+            (self.sweep_length + 0.5 * self.tip_chord - 0.5 * owner.root_chord)
             / owner.span
         )
         Yma = (owner.span / 3) * (owner.root_chord + 2 * self.tip_chord) / Yr
@@ -114,25 +110,17 @@ def evaluate_geometrical_parameters(self):
             * owner.rocket_radius**2
         ) / 12
         roll_damping_interference_factor = 1 + (
-            ((tau - lambda_) / tau)
-            - ((1 - lambda_) / (tau - 1)) * np.log(tau)
+            ((tau - lambda_) / tau) - ((1 - lambda_) / (tau - 1)) * np.log(tau)
         ) / (
             ((tau + 1) * (tau - lambda_)) / 2
             - ((1 - lambda_) * (tau**3 - 1)) / (3 * (tau - 1))
         )
         roll_forcing_interference_factor = (1 / np.pi**2) * (
             (np.pi**2 / 4) * ((tau + 1) ** 2 / tau**2)
-            + (
-                np.pi
-                * (tau**2 + 1) ** 2
-                / (tau**2 * (tau - 1) ** 2)
-            )
+            + (np.pi * (tau**2 + 1) ** 2 / (tau**2 * (tau - 1) ** 2))
             * np.arcsin((tau**2 - 1) / (tau**2 + 1))
             - (2 * np.pi * (tau + 1)) / (tau * (tau - 1))
-            + (
-                (tau**2 + 1) ** 2
-                / (tau**2 * (tau - 1) ** 2)
-            )
+            + ((tau**2 + 1) ** 2 / (tau**2 * (tau - 1) ** 2))
             * (np.arcsin((tau**2 - 1) / (tau**2 + 1))) ** 2
             - (4 * (tau + 1))
             / (tau * (tau - 1))
@@ -149,12 +137,8 @@ def evaluate_geometrical_parameters(self):
         owner.tau = tau
         owner.lift_interference_factor = lift_interference_factor
         owner.λ = lambda_  # pylint: disable=non-ascii-name
-        owner.roll_damping_interference_factor = (
-            roll_damping_interference_factor
-        )
-        owner.roll_forcing_interference_factor = (
-            roll_forcing_interference_factor
-        )
+        owner.roll_damping_interference_factor = roll_damping_interference_factor
+        owner.roll_forcing_interference_factor = roll_forcing_interference_factor
 
         self.evaluate_shape()
 
@@ -192,13 +176,9 @@ def get_data(self, include_outputs=False):
                     "AR": self.owner.AR,
                     "gamma_c": self.owner.gamma_c,
                     "Yma": self.owner.Yma,
-                    "roll_geometrical_constant": (
-                        self.owner.roll_geometrical_constant
-                    ),
+                    "roll_geometrical_constant": (self.owner.roll_geometrical_constant),
                     "tau": self.owner.tau,
-                    "lift_interference_factor": (
-                        self.owner.lift_interference_factor
-                    ),
+                    "lift_interference_factor": (self.owner.lift_interference_factor),
                     "roll_damping_interference_factor": (
                         self.owner.roll_damping_interference_factor
                     ),
@@ -278,8 +258,7 @@ def evaluate_geometrical_parameters(self):  # pylint: disable=too-many-statement
                     * owner.rocket_radius**2
                     * np.sqrt(-(owner.span**2) + owner.rocket_radius**2)
                     * np.arctan(
-                        owner.span
-                        / np.sqrt(-(owner.span**2) + owner.rocket_radius**2)
+                        owner.span / np.sqrt(-(owner.span**2) + owner.rocket_radius**2)
                     )
                     - np.pi
                     * owner.rocket_radius**2
@@ -289,27 +268,17 @@ def evaluate_geometrical_parameters(self):  # pylint: disable=too-many-statement
                 2
                 * owner.span
                 * (-(owner.span**2) + owner.rocket_radius**2)
-                * (
-                    owner.span**2 / 3
-                    + np.pi * owner.span * owner.rocket_radius / 4
-                )
+                * (owner.span**2 / 3 + np.pi * owner.span * owner.rocket_radius / 4)
             )
         else:
             roll_damping_interference_factor = (28 - 3 * np.pi) / (4 + 3 * np.pi)
 
         roll_forcing_interference_factor = (1 / np.pi**2) * (
             (np.pi**2 / 4) * ((tau + 1) ** 2 / tau**2)
-            + (
-                np.pi
-                * (tau**2 + 1) ** 2
-                / (tau**2 * (tau - 1) ** 2)
-            )
+            + (np.pi * (tau**2 + 1) ** 2 / (tau**2 * (tau - 1) ** 2))
             * np.arcsin((tau**2 - 1) / (tau**2 + 1))
             - (2 * np.pi * (tau + 1)) / (tau * (tau - 1))
-            + (
-                (tau**2 + 1) ** 2
-                / (tau**2 * (tau - 1) ** 2)
-            )
+            + ((tau**2 + 1) ** 2 / (tau**2 * (tau - 1) ** 2))
             * (np.arcsin((tau**2 - 1) / (tau**2 + 1))) ** 2
             - (4 * (tau + 1))
             / (tau * (tau - 1))
@@ -324,12 +293,8 @@ def evaluate_geometrical_parameters(self):  # pylint: disable=too-many-statement
         owner.roll_geometrical_constant = roll_geometrical_constant
         owner.tau = tau
         owner.lift_interference_factor = lift_interference_factor
-        owner.roll_damping_interference_factor = (
-            roll_damping_interference_factor
-        )
-        owner.roll_forcing_interference_factor = (
-            roll_forcing_interference_factor
-        )
+        owner.roll_damping_interference_factor = roll_damping_interference_factor
+        owner.roll_forcing_interference_factor = roll_forcing_interference_factor
 
         self.evaluate_shape()
 
@@ -389,12 +354,12 @@ def infer_dimensions(shape_points):
         span = max(ys) - min(ys)
         return root_chord, span
 
-    def evaluate_geometrical_parameters(self):  # pylint: disable=too-many-statements
+    def evaluate_geometrical_parameters(
+        self,
+    ):  # pylint: disable=too-many-statements,too-many-locals,invalid-name
         """Calculates and saves free-form fin geometric parameters."""
         owner = self.owner
 
-        # pylint: disable=invalid-name
-        # pylint: disable=too-many-locals
         Af = 0
         for i in range(len(self.shape_points) - 1):
             x1, y1 = self.shape_points[i]
@@ -410,17 +375,10 @@ def evaluate_geometrical_parameters(self):  # pylint: disable=too-many-statement
 
         roll_forcing_interference_factor = (1 / np.pi**2) * (
             (np.pi**2 / 4) * ((tau + 1) ** 2 / tau**2)
-            + (
-                np.pi
-                * (tau**2 + 1) ** 2
-                / (tau**2 * (tau - 1) ** 2)
-            )
+            + (np.pi * (tau**2 + 1) ** 2 / (tau**2 * (tau - 1) ** 2))
             * np.arcsin((tau**2 - 1) / (tau**2 + 1))
             - (2 * np.pi * (tau + 1)) / (tau * (tau - 1))
-            + (
-                (tau**2 + 1) ** 2
-                / (tau**2 * (tau - 1) ** 2)
-            )
+            + ((tau**2 + 1) ** 2 / (tau**2 * (tau - 1) ** 2))
             * (np.arcsin((tau**2 - 1) / (tau**2 + 1))) ** 2
             - (4 * (tau + 1))
             / (tau * (tau - 1))
@@ -547,13 +505,9 @@ def get_data(self, include_outputs=False):
                     "Yma": self.owner.Yma,
                     "mac_length": self.owner.mac_length,
                     "mac_lead": self.owner.mac_lead,
-                    "roll_geometrical_constant": (
-                        self.owner.roll_geometrical_constant
-                    ),
+                    "roll_geometrical_constant": (self.owner.roll_geometrical_constant),
                     "tau": self.owner.tau,
-                    "lift_interference_factor": (
-                        self.owner.lift_interference_factor
-                    ),
+                    "lift_interference_factor": (self.owner.lift_interference_factor),
                     "roll_forcing_interference_factor": (
                         self.owner.roll_forcing_interference_factor
                     ),

rocketpy/rocket/aero_surface/fins/elliptical_fins.py

@@ -183,9 +183,7 @@ def evaluate_center_of_pressure(self):
     def to_dict(self, **kwargs):
         data = super().to_dict(**kwargs)
         data.update(
-            self.geometry.get_data(
-                include_outputs=kwargs.get("include_outputs", False)
-            )
+            self.geometry.get_data(include_outputs=kwargs.get("include_outputs", False))
         )
         return data
 

rocketpy/rocket/aero_surface/fins/free_form_fins.py

@@ -1,6 +1,7 @@
 from rocketpy.plots.aero_surface_plots import _FreeFormFinsPlots
 from rocketpy.prints.aero_surface_prints import _FreeFormFinsPrints
 from rocketpy.rocket.aero_surface.fins._geometry import _FreeFormGeometry
+
 from .fins import Fins
 
 
@@ -172,9 +173,7 @@ def shape_points(self):
     def to_dict(self, **kwargs):
         data = super().to_dict(**kwargs)
         data.update(
-            self.geometry.get_data(
-                include_outputs=kwargs.get("include_outputs", False)
-            )
+            self.geometry.get_data(include_outputs=kwargs.get("include_outputs", False))
         )
         return data
 

rocketpy/rocket/aero_surface/fins/trapezoidal_fin.py

@@ -1,6 +1,7 @@
 from rocketpy.plots.aero_surface_plots import _TrapezoidalFinPlots
 from rocketpy.prints.aero_surface_prints import _TrapezoidalFinPrints
 from rocketpy.rocket.aero_surface.fins._geometry import _TrapezoidalGeometry
+
 from .fin import Fin
 
 

rocketpy/rocket/aero_surface/fins/trapezoidal_fins.py

@@ -1,6 +1,7 @@
 from rocketpy.plots.aero_surface_plots import _TrapezoidalFinsPlots
 from rocketpy.prints.aero_surface_prints import _TrapezoidalFinsPrints
 from rocketpy.rocket.aero_surface.fins._geometry import _TrapezoidalGeometry
+
 from .fins import Fins
 
 
@@ -227,9 +228,7 @@ def evaluate_center_of_pressure(self):
     def to_dict(self, **kwargs):
         data = super().to_dict(**kwargs)
         data.update(
-            self.geometry.get_data(
-                include_outputs=kwargs.get("include_outputs", False)
-            )
+            self.geometry.get_data(include_outputs=kwargs.get("include_outputs", False))
         )
         return data
 

tests/unit/rocket/aero_surface/test_individual_fins.py

@@ -44,7 +44,7 @@ def test_individual_fin_info_returns_none(request, fixture_name, expected_class)
         "calisto_free_form_fin",
     ],
 )
-def test_individual_fin_draw_returns_none(mock_show, request, fixture_name):
+def test_individual_fin_draw_returns_none(mock_show, request, fixture_name):  # pylint: disable=unused-argument
     """Ensure draw() executes for all individual fin classes."""
     # Arrange
     fin = request.getfixturevalue(fixture_name)
@@ -96,7 +96,9 @@ def test_trapezoidal_fin_setters_update_geometry(calisto_trapezoidal_fin):
 def test_trapezoidal_fin_rejects_inconsistent_sweep_inputs():
     """Ensure trapezoidal fin rejects sweep_length with sweep_angle together."""
     # Arrange / Act / Assert
-    with pytest.raises(ValueError, match="Cannot use sweep_length and sweep_angle together"):
+    with pytest.raises(
+        ValueError, match="Cannot use sweep_length and sweep_angle together"
+    ):
         TrapezoidalFin(
             angular_position=0,
             root_chord=0.12,
@@ -197,7 +199,9 @@ def test_individual_fin_to_dict_contains_expected_keys(
         ),
     ],
 )
-def test_individual_fin_from_dict_roundtrip(request, fixture_name, fin_class, comparisons):
+def test_individual_fin_from_dict_roundtrip(
+    request, fixture_name, fin_class, comparisons
+):
     """Ensure each individual fin can be reconstructed with from_dict."""
     # Arrange
     fin = request.getfixturevalue(fixture_name)