feat: expand header stats with total funds and profit-loss ratio

Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
dekun
2026-07-07 02:14:18 +08:00
parent 659c0969fe
commit 814fe67a21
10 changed files with 138 additions and 10 deletions
+33
View File
@@ -0,0 +1,33 @@
"""instance_embed_context_lib 顶栏统计。"""
from __future__ import annotations
import unittest
from lib.instance.instance_embed_context_lib import (
profit_loss_ratio_from_averages,
profit_loss_ratio_from_trades,
total_funds_usdt,
)
class TestHeaderStatsLib(unittest.TestCase):
def test_profit_loss_ratio_from_averages(self):
self.assertEqual(profit_loss_ratio_from_averages(9.0, -3.0), 3.0)
self.assertIsNone(profit_loss_ratio_from_averages(9.0, 0))
def test_profit_loss_ratio_from_trades(self):
trades = [
{"effective_pnl_amount": 10},
{"effective_pnl_amount": 8},
{"effective_pnl_amount": -4},
{"effective_pnl_amount": -2},
]
self.assertEqual(profit_loss_ratio_from_trades(trades), 3.0)
def test_total_funds_usdt(self):
self.assertEqual(total_funds_usdt(100.5, 59.27), 159.77)
self.assertIsNone(total_funds_usdt(None, 10))
if __name__ == "__main__":
unittest.main()