charieswei
/
TradingAgents
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix Black formatting in test_trading_graph.py 

Split long line for mock_graph.stream.return_value

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-Authored-By: Claude <noreply@anthropic.com>
This commit is contained in:
佐藤優一 2025-08-11 13:19:52 +09:00
parent b314db546f
commit 01841ab876
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tests/unit/graph/test_trading_graph.py
View File

@ -266,7 +266,10 @@ class TestTradingAgentsGraph:
"final_trade_decision": "BUY", "final_trade_decision": "BUY",
} }
mock_final_chunk["messages"][0].pretty_print = Mock() mock_final_chunk["messages"][0].pretty_print = Mock()
mock_graph.stream.return_value = [mock_final_chunk, mock_final_chunk] # Multiple chunks mock_graph.stream.return_value = [
mock_final_chunk,
mock_final_chunk,
] # Multiple chunks
with patch("tradingagents.graph.trading_graph.FinancialSituationMemory"): with patch("tradingagents.graph.trading_graph.FinancialSituationMemory"):
with patch("tradingagents.graph.trading_graph.set_config"): with patch("tradingagents.graph.trading_graph.set_config"):