make format

Author: Gui-FernandesBR

rocketpy/simulation/flight.py

@@ -4176,7 +4176,10 @@ def export_kml(
             color=color,
             altitude_mode=altitude_mode,
         )
-    def animate_trajectory(self, file_name, start=0, stop=None, time_step=0.1, **kwargs):
+
+    def animate_trajectory(
+        self, file_name, start=0, stop=None, time_step=0.1, **kwargs
+    ):
         """
         6-DOF Animation of the flight trajectory using Vedo.
 
@@ -4208,7 +4211,7 @@ def animate_trajectory(self, file_name, start=0, stop=None, time_step=0.1, **kwa
         ------
         ImportError
             If the 'vedo' package is not installed.
-            
+
         Notes
         -----
         This feature requires the 'vedo' package. Install it with:
@@ -4250,8 +4253,10 @@ def animate_trajectory(self, file_name, start=0, stop=None, time_step=0.1, **kwa
         rocket = Mesh(file_name).c("green")
         rocket.pos(self.x(start), self.y(start), 0).add_trail(n=len(self.x[:, 1]))
         # Create trail
-        trail_points = [[self.x(t), self.y(t), self.z(t) - self.env.elevation] 
-                        for t in np.arange(start, stop, time_step)]
+        trail_points = [
+            [self.x(t), self.y(t), self.z(t) - self.env.elevation]
+            for t in np.arange(start, stop, time_step)
+        ]
         trail = Line(trail_points, c="k", alpha=0.5)
         # Setup Plotter
         plt = Plotter(axes=1, interactive=False)
@@ -4260,21 +4265,21 @@ def animate_trajectory(self, file_name, start=0, stop=None, time_step=0.1, **kwa
         # Animation Loop
         for t in np.arange(start, stop, time_step):
             # Calculate rotation angle and vector from quaternions
-            # Note: This simple rotation logic mimics the old branch. 
-            # Ideally, vedo handles orientation via matrix, but we stick 
+            # Note: This simple rotation logic mimics the old branch.
+            # Ideally, vedo handles orientation via matrix, but we stick
             # to the provided logic for now.
-            
+
             # e0 is the scalar part of the quaternion
-            angle = np.arccos(2 * self.e0(t)**2 - 1) 
+            angle = np.arccos(2 * self.e0(t) ** 2 - 1)
             k = np.sin(angle / 2) if np.sin(angle / 2) != 0 else 1
-            
+
             # Update position and rotation
             # Adjusting for ground elevation
             rocket.pos(self.x(t), self.y(t), self.z(t) - self.env.elevation)
             rocket.rotate_x(self.e1(t) / k)
             rocket.rotate_y(self.e2(t) / k)
             rocket.rotate_z(self.e3(t) / k)
-            
+
             # update the scene
             plt.show(world, rocket, trail)
 
@@ -4283,7 +4288,7 @@ def animate_trajectory(self, file_name, start=0, stop=None, time_step=0.1, **kwa
             while time.time() - start_pause < time_step:
                 plt.render()
 
-            if getattr(plt, 'escaped', False):
+            if getattr(plt, "escaped", False):
                 break
 
         plt.interactive().close()
@@ -4311,7 +4316,7 @@ def animate_rotate(self, file_name, start=0, stop=None, time_step=0.1, **kwargs)
             - elevation (float): Rotation in degrees above the horizon.
             - roll (float): Rotation in degrees around the view axis.
             - zoom (float): Zoom level (default 1).
-            
+
         Returns
         -------
         None
@@ -4329,13 +4334,13 @@ def animate_rotate(self, file_name, start=0, stop=None, time_step=0.1, **kwargs)
                 "Install it with:\n"
                 "    pip install rocketpy[animation]\n"
             ) from e
-        
+
         # Enable interaction if needed
         try:
             settings.allow_interaction = True
         except AttributeError:
             pass  # Not available in newer versions of vedo
-        
+
         if stop is None:
             stop = self.t_final
 
@@ -4355,9 +4360,9 @@ def animate_rotate(self, file_name, start=0, stop=None, time_step=0.1, **kwargs)
         plt.show(world, rocket, __doc__, viewup="z", **kwargs)
 
         for t in np.arange(start, stop, time_step):
-            angle = np.arccos(2 * self.e0(t)**2 - 1)
+            angle = np.arccos(2 * self.e0(t) ** 2 - 1)
             k = np.sin(angle / 2) if np.sin(angle / 2) != 0 else 1
-            
+
             # Keep position static (relative start) to observe only rotation
             rocket.pos(self.x(start), self.y(start), 0)
             rocket.rotate_x(self.e1(t) / k)
@@ -4366,12 +4371,12 @@ def animate_rotate(self, file_name, start=0, stop=None, time_step=0.1, **kwargs)
 
             plt.show(world, rocket)
 
-            if getattr(plt, 'escaped', False):
+            if getattr(plt, "escaped", False):
                 break
 
         plt.interactive().close()
         return None
-    
+
     def info(self):
         """Prints out a summary of the data available about the Flight."""
         self.prints.all()

tests/animation_verification/main.py

@@ -1,8 +1,10 @@
 import os
 import traceback
-from rocketpy import Environment, Flight
-from rocket_stl import create_rocket_stl
+
 from rocket_setup import get_calisto_rocket
+from rocket_stl import create_rocket_stl
+
+from rocketpy import Environment, Flight
 
 
 def run_simulation_and_test_animation():

tests/animation_verification/rocket_setup.py

@@ -1,5 +1,6 @@
 import os
-from rocketpy import SolidMotor, Rocket
+
+from rocketpy import Rocket, SolidMotor
 
 CURRENT_DIR = os.path.dirname(os.path.abspath(__file__))
 ROOT_DIR = os.path.dirname(os.path.dirname(CURRENT_DIR))

tests/animation_verification/rocket_stl.py

@@ -1,7 +1,8 @@
 import math
+import struct
+
 import numpy as np
 from stl import mesh  # Requires numpy-stl package: pip install numpy-stl
-import struct
 
 
 def create_rocket_stl(filename="rocket_model.stl", length=100, radius=10):