From 35aa61c651217663406c9cd6df404f85338b2d68 Mon Sep 17 00:00:00 2001
From: TheSiahxyz <164138827+TheSiahxyz@users.noreply.github.com>
Date: Thu, 2 Apr 2026 14:17:43 +0900
Subject: style: fix lint and formatting issues across news collector and
 shared

---
 .../strategy-engine/tests/test_stock_selector.py   | 40 ++++++++++++----------
 1 file changed, 21 insertions(+), 19 deletions(-)

(limited to 'services/strategy-engine/tests/test_stock_selector.py')

diff --git a/services/strategy-engine/tests/test_stock_selector.py b/services/strategy-engine/tests/test_stock_selector.py
index a2f5bca..ff9d09c 100644
--- a/services/strategy-engine/tests/test_stock_selector.py
+++ b/services/strategy-engine/tests/test_stock_selector.py
@@ -1,12 +1,8 @@
 """Tests for stock selector engine."""
 
-import pytest
-from unittest.mock import AsyncMock, MagicMock, patch
+from unittest.mock import AsyncMock, MagicMock
 from datetime import datetime, timezone
-from decimal import Decimal
 
-from shared.models import OrderSide
-from shared.sentiment_models import SymbolScore, MarketSentiment, SelectedStock, Candidate
 
 from strategy_engine.stock_selector import (
     SentimentCandidateSource,
@@ -17,10 +13,12 @@ from strategy_engine.stock_selector import (
 
 async def test_sentiment_candidate_source():
     mock_db = MagicMock()
-    mock_db.get_top_symbol_scores = AsyncMock(return_value=[
-        {"symbol": "AAPL", "composite": 0.8, "news_count": 5},
-        {"symbol": "NVDA", "composite": 0.6, "news_count": 3},
-    ])
+    mock_db.get_top_symbol_scores = AsyncMock(
+        return_value=[
+            {"symbol": "AAPL", "composite": 0.8, "news_count": 5},
+            {"symbol": "NVDA", "composite": 0.6, "news_count": 3},
+        ]
+    )
 
     source = SentimentCandidateSource(mock_db)
     candidates = await source.get_candidates()
@@ -64,15 +62,19 @@ def test_parse_llm_selections_with_markdown():
 
 async def test_selector_blocks_on_risk_off():
     mock_db = MagicMock()
-    mock_db.get_latest_market_sentiment = AsyncMock(return_value={
-        "fear_greed": 15,
-        "fear_greed_label": "Extreme Fear",
-        "vix": 35.0,
-        "fed_stance": "neutral",
-        "market_regime": "risk_off",
-        "updated_at": datetime.now(timezone.utc),
-    })
-
-    selector = StockSelector(db=mock_db, broker=MagicMock(), alpaca=MagicMock(), anthropic_api_key="test")
+    mock_db.get_latest_market_sentiment = AsyncMock(
+        return_value={
+            "fear_greed": 15,
+            "fear_greed_label": "Extreme Fear",
+            "vix": 35.0,
+            "fed_stance": "neutral",
+            "market_regime": "risk_off",
+            "updated_at": datetime.now(timezone.utc),
+        }
+    )
+
+    selector = StockSelector(
+        db=mock_db, broker=MagicMock(), alpaca=MagicMock(), anthropic_api_key="test"
+    )
     result = await selector.select()
     assert result == []
-- 
cgit v1.2.3