style: format verification scripts and pin pytz>=2020.1 for CI stability

Author: GuilhermeAsura

requirements.txt

@@ -3,6 +3,6 @@ scipy>=1.0
 matplotlib>=3.9.0  # Released May 15th 2024
 netCDF4>=1.6.4
 requests>=2.25.0
-pytz
+pytz>=2020.1
 simplekml
 dill

tests/animation_verification/main.py

@@ -4,9 +4,10 @@
 from rocket_stl import create_rocket_stl
 from rocket_setup import get_calisto_rocket
 
+
 def run_simulation_and_test_animation():
     print("🚀 Setting up simulation (Calisto Example)...")
-    
+
     # 1. Setup Environment
     env = Environment(latitude=32.990254, longitude=-106.974998, elevation=1400)
     env.set_date((2025, 12, 5, 12))
@@ -23,40 +24,40 @@ def run_simulation_and_test_animation():
     test_flight = Flight(
         rocket=calisto, environment=env, rail_length=5.2, inclination=85, heading=0
     )
-    
+
     print(f"✅ Flight simulated successfully! Apogee: {test_flight.apogee:.2f} m")
 
     # 4. Test Animation Methods
     stl_file = "rocket_model.stl"
-    # Note: Depending on where you run this, you might need to adjust imports 
+    # Note: Depending on where you run this, you might need to adjust imports
     # or ensure create_rocket_stl is available in scope.
     create_rocket_stl(stl_file, length=300, radius=50)
 
     print("\n🎥 Testing animate_trajectory()...")
-    
+
     try:
         test_flight.animate_trajectory(
-            file_name=stl_file, 
+            file_name=stl_file,
             stop=15.0,
             time_step=0.05,
-            azimuth=-45,   # Rotates view 45 degrees left
+            azimuth=-45,  # Rotates view 45 degrees left
             elevation=30,  # Tilts view 30 degrees up
-            zoom=1.2
-        ) 
+            zoom=1.2,
+        )
         print("✅ animate_trajectory() executed successfully.")
     except Exception as e:
         print(f"❌ animate_trajectory() Failed: {e}")
         traceback.print_exc()
 
     print("\n🔄 Testing animate_rotate()...")
-    
+
     try:
         test_flight.animate_rotate(
-            file_name=stl_file, 
+            file_name=stl_file,
             time_step=1.0,
-            azimuth=-45,   # Rotates view 45 degrees left
+            azimuth=-45,  # Rotates view 45 degrees left
             elevation=30,  # Tilts view 30 degrees up
-            zoom=1.2
+            zoom=1.2,
         )
         print("✅ animate_rotate() executed successfully.")
     except Exception as e:
@@ -67,5 +68,6 @@ def run_simulation_and_test_animation():
     if os.path.exists(stl_file):
         os.remove(stl_file)
 
+
 if __name__ == "__main__":
-    run_simulation_and_test_animation()
+    run_simulation_and_test_animation()

tests/animation_verification/rocket_setup.py

@@ -10,6 +10,7 @@
 AIRFOIL_PATH = os.path.join(DATA_DIR, "airfoils/NACA0012-radians.txt")
 MOTOR_PATH = os.path.join(DATA_DIR, "motors/cesaroni/Cesaroni_M1670.eng")
 
+
 def get_motor():
     """Locates the motor file and returns a configured SolidMotor object."""
     # We can now point directly to the file without searching
@@ -35,10 +36,11 @@ def get_motor():
         coordinate_system_orientation="nozzle_to_combustion_chamber",
     )
 
+
 def get_calisto_rocket():
     """Configures and returns the Calisto Rocket object."""
     motor = get_motor()
-    
+
     calisto = Rocket(
         radius=127 / 2000,
         mass=14.426,
@@ -73,10 +75,20 @@ def get_calisto_rocket():
 
     # Parachutes
     calisto.add_parachute(
-        name="Main", cd_s=10.0, trigger=800, sampling_rate=105, lag=1.5, noise=(0, 8.3, 0.5)
+        name="Main",
+        cd_s=10.0,
+        trigger=800,
+        sampling_rate=105,
+        lag=1.5,
+        noise=(0, 8.3, 0.5),
     )
     calisto.add_parachute(
-        name="Drogue", cd_s=1.0, trigger="apogee", sampling_rate=105, lag=1.5, noise=(0, 8.3, 0.5)
+        name="Drogue",
+        cd_s=1.0,
+        trigger="apogee",
+        sampling_rate=105,
+        lag=1.5,
+        noise=(0, 8.3, 0.5),
     )
-    
-    return calisto
+
+    return calisto