diff options
| author | TheSiahxyz <164138827+TheSiahxyz@users.noreply.github.com> | 2026-04-01 17:06:49 +0900 |
|---|---|---|
| committer | TheSiahxyz <164138827+TheSiahxyz@users.noreply.github.com> | 2026-04-01 17:06:49 +0900 |
| commit | adf5e96542ebd65c7d13ca5e9825071183b3ef13 (patch) | |
| tree | 699800730cc9f68ab775bea5722ba55cf08b304a /tests/integration/test_portfolio_tracking_flow.py | |
| parent | b8dc7344ff99eb23d5f003795f17cdba3b89c40b (diff) | |
fix: lint fixes for integration tests and backtester noqa annotations
Diffstat (limited to 'tests/integration/test_portfolio_tracking_flow.py')
| -rw-r--r-- | tests/integration/test_portfolio_tracking_flow.py | 59 |
1 files changed, 40 insertions, 19 deletions
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") |
