File size: 1,002 Bytes
9596562
 
 
 
 
2a9d346
9596562
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
import pytest
import asyncio
import sys
import os

sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '../backend')))
from simulator import PatientDataSimulator

@pytest.mark.asyncio
async def test_simulator_initialization():
    sim = PatientDataSimulator(initial_hr=80.0)
    assert sim.hr == 80.0
    assert sim.spo2 == 98.0
    assert sim.sys_bp == 120.0
    assert sim.dia_bp == 80.0
    assert sim.temp == 37.0

@pytest.mark.asyncio
async def test_simulator_shift_state():
    sim = PatientDataSimulator()
    import random
    random.seed(42)
    sim.target_hr = 150.0
    sim._step_physics()
    assert sim.hr > 75.0 

@pytest.mark.asyncio
async def test_simulator_run_yields_dict():
    sim = PatientDataSimulator()
    generator = sim.run()
    data = await anext(generator)
    assert isinstance(data, dict)
    assert "Heart_Rate" in data
    assert "SpO2_Level" in data
    assert "Systolic_BP" in data
    assert "Diastolic_BP" in data
    assert "Body_Temp" in data