Spaces:
Runtime error
Runtime error
| import pytest | |
| import pytest_asyncio | |
| import os | |
| import sys | |
| import aiosqlite | |
| from collections import namedtuple | |
| sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '../backend'))) | |
| import database | |
| # Mock settings to avoid real huggingface calls | |
| database.settings.hf_token = None | |
| database.DB_PATH = "test_patient_history.db" | |
| async def setup_teardown(): | |
| if os.path.exists(database.DB_PATH): | |
| os.remove(database.DB_PATH) | |
| await database.init_db() | |
| yield | |
| if os.path.exists(database.DB_PATH): | |
| try: | |
| os.remove(database.DB_PATH) | |
| except PermissionError: | |
| pass # Windows file lock fallback | |
| async def test_init_db(): | |
| assert os.path.exists(database.DB_PATH) | |
| async with aiosqlite.connect(database.DB_PATH) as db: | |
| cursor = await db.execute("SELECT name FROM sqlite_master WHERE type='table' AND name='predictions'") | |
| table = await cursor.fetchone() | |
| assert table is not None | |
| async def test_log_and_get_history(): | |
| Data = namedtuple('Data', ['Heart_Rate', 'SpO2_Level', 'Systolic_BP', 'Diastolic_BP', 'Body_Temp']) | |
| data = Data(80.0, 98.0, 120.0, 80.0, 37.0) | |
| await database.log_prediction(data, "Normal", 0.95) | |
| history = await database.get_history() | |
| assert len(history) == 1 | |
| assert history[0]['heart_rate'] == 80.0 | |
| assert history[0]['prediction_label'] == "Normal" | |
| assert history[0]['confidence'] == 0.95 | |
| async def test_empty_history(): | |
| if os.path.exists(database.DB_PATH): | |
| os.remove(database.DB_PATH) | |
| history = await database.get_history() | |
| assert history == [] | |