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

View File

@ -266,7 +266,10 @@ class TestTradingAgentsGraph:
"final_trade_decision": "BUY",
}
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.set_config"):