2b037a83e8
- Implement migration script to add quetiapine and quetiapine_doses columns to existing CSV data. - Update DataManager to include quetiapine and quetiapine_doses in data handling. - Modify MedTrackerApp to manage quetiapine entries and doses. - Enhance UIManager to include quetiapine in the user interface for medication selection and display. - Update tests to cover new quetiapine functionality, including sample data and DataManager tests.
75 lines
1.9 KiB
Python
75 lines
1.9 KiB
Python
"""
|
|
Fixtures and configuration for pytest tests.
|
|
"""
|
|
import os
|
|
import tempfile
|
|
import pytest
|
|
import pandas as pd
|
|
from unittest.mock import Mock
|
|
import logging
|
|
|
|
|
|
@pytest.fixture
|
|
def temp_csv_file():
|
|
"""Create a temporary CSV file for testing."""
|
|
fd, path = tempfile.mkstemp(suffix='.csv')
|
|
os.close(fd)
|
|
yield path
|
|
# Cleanup
|
|
if os.path.exists(path):
|
|
os.unlink(path)
|
|
|
|
|
|
@pytest.fixture
|
|
def sample_data():
|
|
"""Sample data for testing."""
|
|
return [
|
|
["2024-01-01", 3, 2, 4, 3, 1, "", 0, "", 2, "", 1, "", 0, "", "Test note 1"],
|
|
["2024-01-02", 2, 3, 3, 4, 1, "", 1, "", 2, "", 0, "", 1, "", "Test note 2"],
|
|
["2024-01-03", 4, 1, 5, 2, 0, "", 0, "", 1, "", 1, "", 0, "", ""],
|
|
]
|
|
|
|
|
|
@pytest.fixture
|
|
def sample_dataframe():
|
|
"""Sample DataFrame for testing."""
|
|
return pd.DataFrame({
|
|
'date': ['2024-01-01', '2024-01-02', '2024-01-03'],
|
|
'depression': [3, 2, 4],
|
|
'anxiety': [2, 3, 1],
|
|
'sleep': [4, 3, 5],
|
|
'appetite': [3, 4, 2],
|
|
'bupropion': [1, 1, 0],
|
|
'bupropion_doses': ['', '', ''],
|
|
'hydroxyzine': [0, 1, 0],
|
|
'hydroxyzine_doses': ['', '', ''],
|
|
'gabapentin': [2, 2, 1],
|
|
'gabapentin_doses': ['', '', ''],
|
|
'propranolol': [1, 0, 1],
|
|
'propranolol_doses': ['', '', ''],
|
|
'quetiapine': [0, 1, 0],
|
|
'quetiapine_doses': ['', '', ''],
|
|
'note': ['Test note 1', 'Test note 2', '']
|
|
})
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_logger():
|
|
"""Mock logger for testing."""
|
|
return Mock(spec=logging.Logger)
|
|
|
|
|
|
@pytest.fixture
|
|
def temp_log_dir():
|
|
"""Create a temporary directory for log files."""
|
|
with tempfile.TemporaryDirectory() as temp_dir:
|
|
yield temp_dir
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_env_vars(monkeypatch):
|
|
"""Mock environment variables."""
|
|
monkeypatch.setenv("LOG_LEVEL", "DEBUG")
|
|
monkeypatch.setenv("LOG_PATH", "/tmp/test_logs")
|
|
monkeypatch.setenv("LOG_CLEAR", "False")
|