Fix CSV column header spacing in FlightDataExporter (Issue #864)

Co-authored-by: Gui-FernandesBR <[email protected]>

Author: Copilot

rocketpy/simulation/flight_data_exporter.py

@@ -156,7 +156,7 @@ class attributes which are instances of the Function class. Usage
                     ) from exc
             variable_points = variable_function(time_points)
             exported_matrix += [variable_points]
-            exported_header += f", {variable_function.__outputs__[0]}"
+            exported_header += f",{variable_function.__outputs__[0]}"
 
         exported_matrix = np.array(exported_matrix).T  # Fix matrix orientation
 

tests/unit/test_flight_data_exporter.py

@@ -192,3 +192,63 @@ def test_export_kml_trajectory(flight_calisto_robust, tmp_path):
     assert np.allclose(flight_calisto_robust.latitude[:, 1], lat, atol=1e-3)
     assert np.allclose(flight_calisto_robust.longitude[:, 1], lon, atol=1e-3)
     assert np.allclose(flight_calisto_robust.z[:, 1], z, atol=1e-3)
+
+
+def test_export_data_csv_column_names_no_leading_spaces(flight_calisto, tmp_path):
+    """Test that CSV column headers have no leading spaces after commas.
+
+    This validates that exported CSV files can be easily read with pandas
+    without requiring leading spaces in column names (Issue #864).
+
+    When reading CSVs with pandas, column names should be accessible as
+    'Vz (m/s)' not ' Vz (m/s)' (with leading space).
+
+    Parameters
+    ----------
+    flight_calisto : rocketpy.Flight
+        Flight object to be tested.
+    tmp_path : pathlib.Path
+        Pytest fixture for temporary directories.
+    """
+    file_name = tmp_path / "flight_data_columns.csv"
+    FlightDataExporter(flight_calisto).export_data(
+        str(file_name), "z", "vz", "altitude"
+    )
+
+    # Read the header line directly
+    with open(file_name, "r") as f:
+        header_line = f.readline().strip()
+
+    # Verify header format - should have no spaces after commas
+    # Format should be: # Time (s),Z (m),Vz (m/s),Altitude AGL (m)
+    assert header_line.startswith("# Time (s),")
+    assert ", " not in header_line, "Header should not contain ', ' (comma-space)"
+
+    # Verify with pandas that columns are accessible without leading spaces
+    try:
+        import pandas as pd
+
+        df = pd.read_csv(file_name)
+        columns = df.columns.tolist()
+
+        # First column should be '# Time (s)'
+        assert columns[0] == "# Time (s)"
+
+        # Other columns should NOT have leading spaces
+        for col in columns[1:]:
+            assert not col.startswith(" "), f"Column '{col}' has leading space"
+
+        # Verify columns are accessible with expected names (no leading spaces)
+        assert "Z (m)" in columns
+        assert "Vz (m/s)" in columns
+        assert "Altitude AGL (m)" in columns
+
+        # Verify we can access data using column names without spaces
+        _ = df["# Time (s)"]
+        _ = df["Z (m)"]
+        _ = df["Vz (m/s)"]
+        _ = df["Altitude AGL (m)"]
+
+    except ImportError:
+        # pandas not available, skip pandas-specific test
+        pass