54 lines
1.6 KiB
Python
54 lines
1.6 KiB
Python
from tradingagents.agents.scanners.macro_synthesis import (
|
|
_build_candidate_rankings,
|
|
_extract_rankable_tickers,
|
|
_format_horizon_label,
|
|
)
|
|
|
|
|
|
def test_format_horizon_label_supported_values():
|
|
assert _format_horizon_label(30) == "1 month"
|
|
assert _format_horizon_label(60) == "2 months"
|
|
assert _format_horizon_label(90) == "3 months"
|
|
|
|
|
|
def test_format_horizon_label_unsupported_defaults_to_one_month():
|
|
assert _format_horizon_label(45) == "1 month"
|
|
|
|
|
|
def test_extract_rankable_tickers_filters_noise():
|
|
tickers = _extract_rankable_tickers(
|
|
"NVDA and AAPL look strong; GDP and JSON are not tickers. MSFT also appears."
|
|
)
|
|
assert {"NVDA", "AAPL", "MSFT"} <= tickers
|
|
assert "GDP" not in tickers
|
|
assert "JSON" not in tickers
|
|
|
|
|
|
def test_build_candidate_rankings_rewards_overlap():
|
|
state = {
|
|
"gatekeeper_universe_report": "NVDA AAPL MSFT",
|
|
"market_movers_report": "NVDA AAPL",
|
|
"smart_money_report": "NVDA",
|
|
"factor_alignment_report": "NVDA MSFT",
|
|
"drift_opportunities_report": "NVDA AAPL",
|
|
"industry_deep_dive_report": "MSFT",
|
|
}
|
|
ranked = _build_candidate_rankings(state)
|
|
|
|
assert ranked[0]["ticker"] == "NVDA"
|
|
assert ranked[0]["score"] > ranked[1]["score"]
|
|
|
|
|
|
def test_build_candidate_rankings_excludes_names_outside_gatekeeper():
|
|
state = {
|
|
"gatekeeper_universe_report": "NVDA AAPL",
|
|
"market_movers_report": "NVDA TSLA",
|
|
"drift_opportunities_report": "TSLA",
|
|
}
|
|
|
|
ranked = _build_candidate_rankings(state)
|
|
|
|
tickers = {row["ticker"] for row in ranked}
|
|
assert "NVDA" in tickers
|
|
assert "TSLA" not in tickers
|