feat: add axial_acceleration with tests

Author: hogatata

rocketpy/simulation/flight.py

@@ -2620,6 +2620,15 @@ def max_acceleration(self):
         max_acceleration_time_index = np.argmax(self.acceleration[:, 1])
         return self.acceleration[max_acceleration_time_index, 1]
 
+    @cached_property
+    def axial_acceleration(self):
+        """Axial acceleration magnitude as a function of time."""
+        return (
+            self.ax * self.attitude_vector_x
+            + self.ay * self.attitude_vector_y
+            + self.az * self.attitude_vector_z
+        )
+
     @funcify_method("Time (s)", "Horizontal Speed (m/s)")
     def horizontal_speed(self):
         """Rocket horizontal speed as a Function of time."""

tests/conftest.py

@@ -1,6 +1,6 @@
+import matplotlib
 import netCDF4
 import numpy as np
-import matplotlib
 import pytest
 
 # Configure matplotlib to use non-interactive backend for tests

tests/unit/simulation/test_flight.py

@@ -413,6 +413,37 @@ def test_max_values(flight_calisto_robust):
     assert pytest.approx(285.94948, rel=rtol) == test.max_speed
 
 
+@pytest.mark.parametrize(
+    "flight_time_attr",
+    ["t_initial", "out_of_rail_time", "apogee_time", "t_final"],
+)
+def test_axial_acceleration(flight_calisto_custom_wind, flight_time_attr):
+    """Tests the axial_acceleration property by manually calculating the
+    dot product of the acceleration vector and the attitude vector at
+    specific time steps.
+
+    Parameters
+    ----------
+    flight_calisto_custom_wind : rocketpy.Flight
+        Flight object to be tested.
+    flight_time_attr : str
+        The name of the attribute of the flight object that contains the time
+        of the point to be tested.
+    """
+    flight = flight_calisto_custom_wind
+    t = getattr(flight, flight_time_attr)
+
+    calculated_axial_acc = flight.axial_acceleration(t)
+
+    expected_axial_acc = (
+        flight.ax(t) * flight.attitude_vector_x(t)
+        + flight.ay(t) * flight.attitude_vector_y(t)
+        + flight.az(t) * flight.attitude_vector_z(t)
+    )
+
+    assert pytest.approx(expected_axial_acc, abs=1e-9) == calculated_axial_acc
+
+
 def test_effective_rail_length(flight_calisto_robust, flight_calisto_nose_to_tail):
     """Tests the effective rail length of the flight simulation. The expected
     values are calculated by hand, and should be valid as long as the rail