Source code for test_push_agent_agent.test

"""
Tests the force orthogonal to the contact surface, representing a damped spring interaction between two agents.

Tests cover:
    - Time and position continuity for each agent
    - Constant y position during x-axis push
    - Constant orientation (theta) during push
    - Near-zero angular velocity (omega) during push
    - Stationary phase (with near constant x, y, theta and near zero vx, vy, omega)
"""

# Copyright  2025  Institute of Light and Matter, CNRS UMR 5306, University Claude Bernard Lyon 1
# Contributors: Oscar DUFOUR, Maxime STAPELLE, Alexandre NICOLAS

# This software is a computer program designed to generate a realistic crowd from anthropometric data and
# simulate the mechanical interactions that occur within it and with obstacles.

# This software is governed by the CeCILL-B license under French law and abiding by the rules of distribution
# of free software.  You can  use, modify and/ or redistribute the software under the terms of the CeCILL-B
# license as circulated by CEA, CNRS and INRIA at the following URL "http://www.cecill.info".

# As a counterpart to the access to the source code and  rights to copy, modify and redistribute granted by
# the license, users are provided only with a limited warranty  and the software's author,  the holder of the
# economic rights,  and the successive licensors  have only  limited liability.

# In this respect, the user's attention is drawn to the risks associated with loading,  using,  modifying
# and/or developing or reproducing the software by the user in light of its specific status of free software,
# that may mean  that it is complicated to manipulate,  and  that  also therefore means  that it is reserved
# for developers  and  experienced professionals having in-depth computer knowledge. Users are therefore
# encouraged to load and test the software's suitability as regards their requirements in conditions enabling
# the security of their systems and/or data to be ensured and,  more generally, to use and operate it in the
# same conditions as regards security.

# The fact that you are presently reading this means that you have had knowledge of the CeCILL-B license and that
# you accept its terms.

import subprocess
from pathlib import Path

import numpy as np
import pandas as pd
import pytest

from configuration.backup import xml_to_Chaos

#: Tolerance for the constancy of the decisional time step used throughout the simulation (s).
TIME_TOL = 1e-4
#: Maximum allowed spatial jump (m) between consecutive time steps for each agent.
MAX_SPATIAL_JUMP = 1
#: Tolerance for the constancy of y coordinate during the push on the x-axis and during the stationary phase (m).
DELTA_Y_TOL = 1e-2
#: Tolerance for near-zero velocities along x during stationary phase (m/s).
VX_TOL = 1e-2
#: Tolerance for near-zero velocities along y during stationary phase (m/s).
VY_TOL = 1e-2
#: Tolerance for near-zero angular velocities during stationary phase (rad/s).
OMEGA_TOL = 1e-2
#: Tolerance for constancy of x position during stationary phase (m).
DELTA_X_TOL = 1e-2
#: Tolerance for constancy of orientation during push and stationary phase (radians).
DELTA_THETA_TOL = 1e-2


[docs] @pytest.fixture(scope="session") def df() -> pd.DataFrame: """ Export to CSV the XML files and load the time series once per test session. Returns ------- pd.DataFrame DataFrame containing all time series. """ subprocess.run( ["uv", "run", "python", "run_simulation.py"], check=True, ) filenameCSV = "all_trajectories.csv" # Name of the final CSV file we’ll generate PathXML = Path("inputXML") # Folder path where the XML files are located PathCSV = Path("inputCSV") # Folder path where CSV files will be saved PathCSV.mkdir(parents=True, exist_ok=True) # Create directories if it doesn't exist xml_to_Chaos.export_XML_to_CSV(PathCSV, PathXML) return pd.read_csv(PathCSV / filenameCSV)
[docs] def test_time_and_position_continuity(df: pd.DataFrame) -> None: """ Test time and position continuity for each agent. Parameters ---------- df : pd.DataFrame DataFrame containing all time series. """ required_cols = {"ID", "t", "x", "y"} missing = required_cols - set(df.columns) assert not missing, f"Missing expected columns: {missing}" # agent IDs with irregular time steps violations_missing_time: list[int] = [] # (agent_id, list of jump distances > MAX_SPATIAL_JUMP) violations_big_jump: list[tuple[int, list[float]]] = [] for agent_id, g in df.sort_values("t").groupby("ID"): t = g["t"].to_numpy() dt = np.diff(t) ddt = np.diff(dt) if not np.all(np.abs(ddt) < TIME_TOL): violations_missing_time.append(int(agent_id)) x = g["x"].to_numpy() y = g["y"].to_numpy() dist = np.sqrt(np.diff(x) ** 2 + np.diff(y) ** 2) bad_jump_idx = np.where(dist > MAX_SPATIAL_JUMP)[0] if bad_jump_idx.size > 0: violations_big_jump.append((int(agent_id), dist[bad_jump_idx].tolist())) assert not violations_missing_time, f"Irregular time steps: {violations_missing_time}" assert not violations_big_jump, f"Large spatial jumps: {violations_big_jump}"
[docs] def test_push_on_x_axis_only(df: pd.DataFrame) -> None: """ Test that during the push on the x-axis, y and theta remain constant and omega ~ 0. Parameters ---------- df : pd.DataFrame DataFrame containing all time series. """ required_cols = {"ID", "y", "theta", "omega"} missing = required_cols - set(df.columns) assert not missing, f"Missing expected columns: {missing}" violations_y: list[tuple[int, float]] = [] violations_theta: list[tuple[int, float]] = [] violations_omega: list[tuple[int, float]] = [] for agent_id, g in df.groupby("ID"): y_range = g["y"].max() - g["y"].min() if y_range > DELTA_Y_TOL: violations_y.append((agent_id, float(y_range))) theta_range = g["theta"].max() - g["theta"].min() if theta_range > DELTA_THETA_TOL: violations_theta.append((agent_id, float(theta_range))) max_abs_omega = float(g["omega"].abs().max()) if max_abs_omega > OMEGA_TOL: violations_omega.append((agent_id, max_abs_omega)) assert not violations_y, f"y not constant: {violations_y}" assert not violations_theta, f"theta not constant: {violations_theta}" assert not violations_omega, f"omega not ~0: {violations_omega}"
[docs] def test_stationary_phase(df: pd.DataFrame) -> None: """ Test that during the last 5% of the simulation the velocity is approximately zero. Parameters ---------- df : pd.DataFrame DataFrame containing all time series. """ required_cols = {"ID", "vx", "vy", "x", "y", "theta", "t"} missing = required_cols - set(df.columns) assert not missing, f"Missing expected columns: {missing}" violations_stationary: list[tuple[int, float, float]] = [] for agent_id, g in df.groupby("ID"): t_max = g["t"].max() stationary_phase = g[g["t"] >= t_max * 0.95] max_vx = float(stationary_phase["vx"].abs().max()) max_vy = float(stationary_phase["vy"].abs().max()) x_range = np.abs(stationary_phase["x"].max() - stationary_phase["x"].min()) y_range = np.abs(stationary_phase["y"].max() - stationary_phase["y"].min()) theta_range = np.abs(stationary_phase["theta"].max() - stationary_phase["theta"].min()) if max_vx > VX_TOL or max_vy > VY_TOL or x_range > DELTA_X_TOL or y_range > DELTA_Y_TOL or theta_range > DELTA_THETA_TOL: violations_stationary.append((agent_id, max_vx, max_vy)) assert not violations_stationary, f"Non-zero velocity in stationary phase: {violations_stationary}"