109 lines
3.2 KiB
Python
109 lines
3.2 KiB
Python
"""验证 TuShare 拉数流程与因子计算的集成行为。"""
|
|
from __future__ import annotations
|
|
|
|
import sys
|
|
import types
|
|
from datetime import date
|
|
|
|
import pytest
|
|
|
|
# 某些环境下 pandas 可能存在二进制依赖问题，这里提供最小桩避免导入失败
|
|
try: # pragma: no cover - 测试运行环境中若 pandas 可用则直接复用
|
|
import pandas as _pd # type: ignore
|
|
except Exception: # pragma: no cover - stub fallback
|
|
pandas_stub = types.ModuleType("pandas")
|
|
|
|
class _DummyFrame: # pylint: disable=too-few-public-methods
|
|
empty = True
|
|
|
|
def __init__(self, *args, **kwargs): # noqa: D401
|
|
"""轻量占位，避免测试期调用实际逻辑。"""
|
|
|
|
def to_dict(self, *_args, **_kwargs):
|
|
return {}
|
|
|
|
def reindex(self, *_args, **_kwargs):
|
|
return self
|
|
|
|
def where(self, *_args, **_kwargs):
|
|
return self
|
|
|
|
pandas_stub.DataFrame = _DummyFrame
|
|
pandas_stub.Series = _DummyFrame
|
|
pandas_stub.concat = lambda *args, **kwargs: _DummyFrame() # type: ignore[arg-type]
|
|
pandas_stub.Timestamp = lambda *args, **kwargs: None # type: ignore[assignment]
|
|
pandas_stub.to_datetime = lambda value, **kwargs: value # type: ignore[assignment]
|
|
pandas_stub.isna = lambda value: False # type: ignore[assignment]
|
|
pandas_stub.notna = lambda value: True # type: ignore[assignment]
|
|
|
|
sys.modules.setdefault("pandas", pandas_stub)
|
|
else: # pragma: no cover
|
|
sys.modules.setdefault("pandas", _pd)
|
|
|
|
from app.ingest.tushare import FetchJob, run_ingestion
|
|
from app.utils import alerts
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def clear_alerts():
|
|
alerts.clear_warnings()
|
|
yield
|
|
alerts.clear_warnings()
|
|
|
|
|
|
def test_run_ingestion_triggers_factor_range(monkeypatch):
|
|
job = FetchJob(
|
|
name="daily_job",
|
|
start=date(2025, 1, 10),
|
|
end=date(2025, 1, 11),
|
|
ts_codes=("000001.SZ",),
|
|
)
|
|
|
|
coverage_called = {}
|
|
|
|
def fake_coverage(*args, **kwargs):
|
|
coverage_called["args"] = (args, kwargs)
|
|
|
|
monkeypatch.setattr("app.ingest.tushare.ensure_data_coverage", fake_coverage)
|
|
|
|
captured: dict = {}
|
|
|
|
def fake_compute(start, end, **kwargs):
|
|
captured["start"] = start
|
|
captured["end"] = end
|
|
captured["kwargs"] = kwargs
|
|
return []
|
|
|
|
monkeypatch.setattr("app.ingest.tushare.compute_factor_range", fake_compute)
|
|
|
|
run_ingestion(job, include_limits=False)
|
|
|
|
assert "args" in coverage_called
|
|
assert captured["start"] == job.start
|
|
assert captured["end"] == job.end
|
|
assert captured["kwargs"] == {"ts_codes": job.ts_codes, "skip_existing": False}
|
|
|
|
|
|
def test_run_ingestion_skips_factors_for_non_daily(monkeypatch):
|
|
job = FetchJob(
|
|
name="weekly_job",
|
|
start=date(2025, 1, 10),
|
|
end=date(2025, 1, 17),
|
|
granularity="weekly",
|
|
ts_codes=None,
|
|
)
|
|
|
|
monkeypatch.setattr("app.ingest.tushare.ensure_data_coverage", lambda *_, **__: None)
|
|
|
|
invoked = {"count": 0}
|
|
|
|
def fake_compute(*args, **kwargs):
|
|
invoked["count"] += 1
|
|
return []
|
|
|
|
monkeypatch.setattr("app.ingest.tushare.compute_factor_range", fake_compute)
|
|
|
|
run_ingestion(job)
|
|
|
|
assert invoked["count"] == 0
|