Team Ai
Apppublic

IDKHowToCodeFr/tinyml-backend

sourceHugging Faceupdated 10m agoView on Hugging Face
1likes
test_simulator.py38 linesDownload Raw Back to tests
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