mirror of
https://github.com/simonlin1212/TradingAgents-astock.git
synced 2026-08-31 01:23:38 +00:00
c567f8aee9
- 适配A股数据源(mootdx+腾讯财经+akshare)，全免费无积分墙 - 新增3个A股特化分析师(政策/游资/解禁)，共7位AI分析师 - A股交易规则约束(T+1/涨跌停/最小手数/ST标识) - 内置Streamlit Web UI + PDF报告导出 - 支持MiniMax/DeepSeek/智谱/通义/OpenAI/Anthropic等多家LLM - 沪深300基准替代SPY
94 lines
3.4 KiB
Python
94 lines
3.4 KiB
Python
"""Data quality gate test: run all 14 a_stock.py endpoints on 000858 五粮液.
|
|
|
|
Checks: return type, non-empty, data freshness, key fields present.
|
|
"""
|
|
|
|
import sys
|
|
import time
|
|
import traceback
|
|
|
|
sys.path.insert(0, ".")
|
|
|
|
TICKER = "300750"
|
|
TRADE_DATE = "2026-05-12"
|
|
START_DATE = "2026-04-01"
|
|
END_DATE = "2026-05-12"
|
|
|
|
from tradingagents.dataflows.a_stock import (
|
|
get_stock_data,
|
|
get_indicators,
|
|
get_fundamentals,
|
|
get_balance_sheet,
|
|
get_cashflow,
|
|
get_income_statement,
|
|
get_news,
|
|
get_global_news,
|
|
get_insider_transactions,
|
|
get_profit_forecast,
|
|
get_hot_stocks,
|
|
get_northbound_flow,
|
|
get_concept_blocks,
|
|
get_fund_flow,
|
|
get_dragon_tiger_board,
|
|
get_lockup_expiry,
|
|
get_industry_comparison,
|
|
)
|
|
|
|
TESTS = [
|
|
("1. get_stock_data", lambda: get_stock_data(TICKER, START_DATE, END_DATE)),
|
|
("2. get_indicators", lambda: get_indicators(TICKER, "rsi", END_DATE, 30)),
|
|
("3. get_fundamentals", lambda: get_fundamentals(TICKER, TRADE_DATE)),
|
|
("4. get_balance_sheet", lambda: get_balance_sheet(TICKER, "quarterly", TRADE_DATE)),
|
|
("5. get_cashflow", lambda: get_cashflow(TICKER, "quarterly", TRADE_DATE)),
|
|
("6. get_income_statement", lambda: get_income_statement(TICKER, "quarterly", TRADE_DATE)),
|
|
("7. get_news", lambda: get_news(TICKER, START_DATE, END_DATE)),
|
|
("8. get_global_news", lambda: get_global_news(TRADE_DATE, 7, 10)),
|
|
("9. get_insider_transactions", lambda: get_insider_transactions(TICKER)),
|
|
("10. get_profit_forecast", lambda: get_profit_forecast(TICKER, TRADE_DATE)),
|
|
("11. get_hot_stocks", lambda: get_hot_stocks(TRADE_DATE)),
|
|
("12. get_northbound_flow", lambda: get_northbound_flow(TRADE_DATE, True)),
|
|
("13. get_concept_blocks", lambda: get_concept_blocks(TICKER)),
|
|
("14. get_fund_flow", lambda: get_fund_flow(TICKER, TRADE_DATE, True)),
|
|
("15. get_dragon_tiger_board", lambda: get_dragon_tiger_board(TICKER, TRADE_DATE)),
|
|
("16. get_lockup_expiry", lambda: get_lockup_expiry(TICKER, TRADE_DATE)),
|
|
("17. get_industry_comparison", lambda: get_industry_comparison(TICKER, TRADE_DATE)),
|
|
]
|
|
|
|
print(f"{'='*70}")
|
|
print(f"数据质量实测 | {TICKER} 五粮液 | {TRADE_DATE}")
|
|
print(f"{'='*70}\n")
|
|
|
|
results = []
|
|
|
|
for name, fn in TESTS:
|
|
print(f"--- {name} ---")
|
|
t0 = time.time()
|
|
try:
|
|
out = fn()
|
|
elapsed = time.time() - t0
|
|
length = len(out) if isinstance(out, str) else len(str(out))
|
|
preview = out[:800] if isinstance(out, str) else str(out)[:800]
|
|
print(f" OK | {elapsed:.1f}s | {length} chars")
|
|
print(f" >>> {preview}")
|
|
if length < 50:
|
|
results.append((name, "WARN", f"too short ({length} chars)"))
|
|
else:
|
|
results.append((name, "OK", f"{length} chars, {elapsed:.1f}s"))
|
|
except Exception as e:
|
|
elapsed = time.time() - t0
|
|
print(f" FAIL | {elapsed:.1f}s | {type(e).__name__}: {e}")
|
|
traceback.print_exc()
|
|
results.append((name, "FAIL", str(e)[:100]))
|
|
print()
|
|
|
|
print(f"\n{'='*70}")
|
|
print("SUMMARY")
|
|
print(f"{'='*70}")
|
|
ok = sum(1 for _, s, _ in results if s == "OK")
|
|
warn = sum(1 for _, s, _ in results if s == "WARN")
|
|
fail = sum(1 for _, s, _ in results if s == "FAIL")
|
|
print(f"OK: {ok} | WARN: {warn} | FAIL: {fail} / {len(results)} total\n")
|
|
for name, status, detail in results:
|
|
marker = "✓" if status == "OK" else ("⚠" if status == "WARN" else "✗")
|
|
print(f" {marker} {name}: {detail}")
|