1 files changed, 32 insertions, 0 deletions
diff --git a/services/portfolio-manager/tests/test_pnl.py b/services/portfolio-manager/tests/test_pnl.py
new file mode 100644
index 0000000..4462adc
--- /dev/null
+++ b/services/portfolio-manager/tests/test_pnl.py
@@ -0,0 +1,32 @@
+"""Tests for PnL calculation functions."""
+from decimal import Decimal
+
+from portfolio_manager.pnl import calculate_realized_pnl, calculate_unrealized_pnl
+
+
+def test_unrealized_pnl_profit() -> None:
+    result = calculate_unrealized_pnl(
+        quantity=Decimal("0.1"),
+        avg_entry_price=Decimal("50000"),
+        current_price=Decimal("55000"),
+    )
+    assert result == Decimal("500")
+
+
+def test_unrealized_pnl_loss() -> None:
+    result = calculate_unrealized_pnl(
+        quantity=Decimal("0.1"),
+        avg_entry_price=Decimal("50000"),
+        current_price=Decimal("45000"),
+    )
+    assert result == Decimal("-500")
+
+
+def test_realized_pnl_single_trade() -> None:
+    result = calculate_realized_pnl(
+        buy_price=Decimal("50000"),
+        sell_price=Decimal("55000"),
+        quantity=Decimal("0.1"),
+        fee=Decimal("5.5"),
+    )
+    assert result == Decimal("494.5")