From adf5e96542ebd65c7d13ca5e9825071183b3ef13 Mon Sep 17 00:00:00 2001 From: TheSiahxyz <164138827+TheSiahxyz@users.noreply.github.com> Date: Wed, 1 Apr 2026 17:06:49 +0900 Subject: fix: lint fixes for integration tests and backtester noqa annotations --- tests/integration/test_portfolio_tracking_flow.py | 59 +++++++++++++++-------- 1 file changed, 40 insertions(+), 19 deletions(-) (limited to 'tests/integration/test_portfolio_tracking_flow.py') diff --git a/tests/integration/test_portfolio_tracking_flow.py b/tests/integration/test_portfolio_tracking_flow.py index 386e78f..80a781c 100644 --- a/tests/integration/test_portfolio_tracking_flow.py +++ b/tests/integration/test_portfolio_tracking_flow.py @@ -1,12 +1,13 @@ """Integration test: order -> portfolio tracker -> position state.""" + import sys from pathlib import Path -sys.path.insert(0, str(Path(__file__).resolve().parents[2] / "services" / "portfolio-manager" / "src")) +sys.path.insert( + 0, str(Path(__file__).resolve().parents[2] / "services" / "portfolio-manager" / "src") +) -import pytest from decimal import Decimal -from datetime import datetime, timezone from shared.models import Order, OrderSide, OrderType, OrderStatus from portfolio_manager.portfolio import PortfolioTracker @@ -17,9 +18,13 @@ def test_portfolio_tracks_buy_sell_cycle(): tracker = PortfolioTracker() buy_order = Order( - signal_id="sig-1", symbol="BTCUSDT", side=OrderSide.BUY, - type=OrderType.MARKET, price=Decimal("50000"), - quantity=Decimal("0.1"), status=OrderStatus.FILLED, + signal_id="sig-1", + symbol="BTCUSDT", + side=OrderSide.BUY, + type=OrderType.MARKET, + price=Decimal("50000"), + quantity=Decimal("0.1"), + status=OrderStatus.FILLED, ) tracker.apply_order(buy_order) @@ -29,9 +34,13 @@ def test_portfolio_tracks_buy_sell_cycle(): assert pos.avg_entry_price == Decimal("50000") sell_order = Order( - signal_id="sig-2", symbol="BTCUSDT", side=OrderSide.SELL, - type=OrderType.MARKET, price=Decimal("55000"), - quantity=Decimal("0.1"), status=OrderStatus.FILLED, + signal_id="sig-2", + symbol="BTCUSDT", + side=OrderSide.SELL, + type=OrderType.MARKET, + price=Decimal("55000"), + quantity=Decimal("0.1"), + status=OrderStatus.FILLED, ) tracker.apply_order(sell_order) @@ -43,16 +52,28 @@ def test_portfolio_weighted_average_on_multiple_buys(): """Multiple buys at different prices should compute weighted average.""" tracker = PortfolioTracker() - tracker.apply_order(Order( - signal_id="s1", symbol="BTCUSDT", side=OrderSide.BUY, - type=OrderType.MARKET, price=Decimal("50000"), - quantity=Decimal("0.1"), status=OrderStatus.FILLED, - )) - tracker.apply_order(Order( - signal_id="s2", symbol="BTCUSDT", side=OrderSide.BUY, - type=OrderType.MARKET, price=Decimal("60000"), - quantity=Decimal("0.1"), status=OrderStatus.FILLED, - )) + tracker.apply_order( + Order( + signal_id="s1", + symbol="BTCUSDT", + side=OrderSide.BUY, + type=OrderType.MARKET, + price=Decimal("50000"), + quantity=Decimal("0.1"), + status=OrderStatus.FILLED, + ) + ) + tracker.apply_order( + Order( + signal_id="s2", + symbol="BTCUSDT", + side=OrderSide.BUY, + type=OrderType.MARKET, + price=Decimal("60000"), + quantity=Decimal("0.1"), + status=OrderStatus.FILLED, + ) + ) pos = tracker.get_position("BTCUSDT") assert pos.quantity == Decimal("0.2") -- cgit v1.2.3