Yuxuan Yan
19 lines
642 B
Python
19 lines
642 B
Python
import pytest
|
|
from data.downloader import download_daily
|
|
|
|
|
|
def test_download_single_stock():
|
|
"""Smoke test: download data for 浦发银行 for a short window."""
|
|
df = download_daily("sh600000", "2024-01-01", "2024-01-31")
|
|
assert df is not None
|
|
assert len(df) > 0
|
|
assert list(df.columns) == ["symbol", "date", "open", "high", "low", "close", "volume", "amount"]
|
|
assert df["close"].notna().all()
|
|
|
|
|
|
def test_download_baostock_fallback():
|
|
"""Test baostock works as secondary source."""
|
|
df = download_daily("sz000001", "2024-06-01", "2024-06-15", source="baostock")
|
|
assert df is not None
|
|
assert len(df) > 0
|