/
maruga
/
mlops
Обзор
Документация
Войти
/
maruga
/
mlops
Код
Запросы
0
Задачи
Вики
Пакеты
0
Релизы
0
CI/CD
Аналитика
Безопасность
master
tests/conftest.py
45 строк
1 KB
s546371
modified: .gitlab-ci.yml
29 мар 2026, 21:35
29 мар 2026, 21:35
47c4e99
Код
Авторство
О чём код?
"""Модуль содержит фикстуры для тестов""" import pandas as pd import pytest from fastapi.testclient import TestClient @pytest.fixture(autouse=True) def mock_loaded_models(mocker): """Автоматически мокирует модели для всех тестов""" class MockRanker: def predict_proba(self, x): return [[0.3, 0.7]] mocker.patch("src.loaded_models.ranker", MockRanker()) mock_features = pd.DataFrame( { "itemid": [98113, 98114], "views": [100, 150], "purchases": [10, 15], "ctr": [0.1, 0.1], "categoryid": [5, 5], "available": [1, 1], } ) mocker.patch("src.loaded_models.item_features", mock_features) @pytest.fixture def client(): """Фикстура тестового клиента""" from src.main import app return TestClient(app) @pytest.fixture def sample_events_csv(tmp_path, request): """Фикстура с примером событий.""" csv_data = f"timestamp,event,visitorid,itemid,transactionid,\n{request.param}" file_path = tmp_path / "events.csv" file_path.write_text(csv_data) return file_path