IDKHowToCodeFr/tinyml-backend
1
1import pytest2import asyncio3import sys4import os5 6sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '../backend')))7from simulator import PatientDataSimulator8 9@pytest.mark.asyncio10async def test_simulator_initialization():11 sim = PatientDataSimulator(initial_hr=80.0)12 assert sim.hr == 80.013 assert sim.spo2 == 98.014 assert sim.sys_bp == 120.015 assert sim.dia_bp == 80.016 assert sim.temp == 37.017 18@pytest.mark.asyncio19async def test_simulator_shift_state():20 sim = PatientDataSimulator()21 import random22 random.seed(42)23 sim.target_hr = 150.024 sim._step_physics()25 assert sim.hr > 75.0 26 27@pytest.mark.asyncio28async def test_simulator_run_yields_dict():29 sim = PatientDataSimulator()30 generator = sim.run()31 data = await anext(generator)32 assert isinstance(data, dict)33 assert "Heart_Rate" in data34 assert "SpO2_Level" in data35 assert "Systolic_BP" in data36 assert "Diastolic_BP" in data37 assert "Body_Temp" in data38 