dekun
111 lines
4.2 KiB
Python
111 lines
4.2 KiB
Python
"""多笔买/卖回 VWAP 与真盈亏."""
|
|
from __future__ import annotations
|
|
|
|
import sqlite3
|
|
import unittest
|
|
|
|
|
|
class SpotVwapBridgeTests(unittest.TestCase):
|
|
def test_buy_and_close_sell_vwap_multi_fill(self):
|
|
from lib.options.options_spot_bridge_lib import (
|
|
BRIDGE_CLOSED,
|
|
add_bridge_buy_fill,
|
|
add_bridge_close_sell_fill,
|
|
bridge_buy_vwap,
|
|
bridge_sell_vwap,
|
|
ensure_bridge_table,
|
|
insert_bridge,
|
|
update_bridge,
|
|
)
|
|
|
|
conn = sqlite3.connect(":memory:")
|
|
conn.row_factory = sqlite3.Row
|
|
ensure_bridge_table(conn)
|
|
bid = insert_bridge(conn, underlying="BTC", status="holding", budget_usdt=2200)
|
|
add_bridge_buy_fill(conn, bid, usdt_spent=1000.0, coin_bought=0.5)
|
|
add_bridge_buy_fill(conn, bid, usdt_spent=1200.0, coin_bought=0.6)
|
|
# 2200 / 1.1 = 2000
|
|
self.assertAlmostEqual(float(bridge_buy_vwap(conn, bid)), 2000.0, places=4)
|
|
|
|
add_bridge_close_sell_fill(conn, bid, usdt_recovered=800.0, coin_sold=0.4)
|
|
add_bridge_close_sell_fill(conn, bid, usdt_recovered=1200.0, coin_sold=0.6)
|
|
# 2000 / 1.0 = 2000
|
|
self.assertAlmostEqual(float(bridge_sell_vwap(conn, bid)), 2000.0, places=4)
|
|
update_bridge(conn, bid, status=BRIDGE_CLOSED, close=True)
|
|
self.assertAlmostEqual(float(bridge_sell_vwap(conn, bid)), 2000.0, places=4)
|
|
conn.close()
|
|
|
|
def test_true_pnl_uses_bridge_vwap_not_index(self):
|
|
from lib.options.options_db import init_options_tables
|
|
from lib.options.options_pnl_settle_lib import resolve_true_pnl_usdt_from_trade
|
|
from lib.options.options_spot_bridge_lib import (
|
|
BRIDGE_CLOSED,
|
|
add_bridge_buy_fill,
|
|
add_bridge_close_sell_fill,
|
|
ensure_bridge_table,
|
|
insert_bridge,
|
|
update_bridge,
|
|
)
|
|
|
|
conn = sqlite3.connect(":memory:")
|
|
conn.row_factory = sqlite3.Row
|
|
init_options_tables(conn)
|
|
ensure_bridge_table(conn)
|
|
bid = insert_bridge(conn, underlying="BTC", status="holding", budget_usdt=2200)
|
|
add_bridge_buy_fill(conn, bid, usdt_spent=2200.0, coin_bought=1.1)
|
|
add_bridge_close_sell_fill(conn, bid, usdt_recovered=1600.0, coin_sold=0.8)
|
|
add_bridge_close_sell_fill(conn, bid, usdt_recovered=400.0, coin_sold=0.2)
|
|
update_bridge(conn, bid, status=BRIDGE_CLOSED, close=True)
|
|
# buy_vwap=2000, sell_vwap=2000; coin_pnl=0.01
|
|
# A = 0.01*2000 - (0.01*2000 - 0.01*2000) = 20
|
|
# 若误用指数 90000/80000 会得到 700
|
|
pnl, src = resolve_true_pnl_usdt_from_trade(
|
|
premium_paid=0.01,
|
|
premium_received=0.02,
|
|
realized_pnl=0.01,
|
|
open_index_px=90000.0,
|
|
close_index_px=80000.0,
|
|
premium_ccy="BTC",
|
|
conn=conn,
|
|
bridge_id=bid,
|
|
)
|
|
self.assertEqual(src, "premium_move")
|
|
self.assertAlmostEqual(float(pnl), 20.0, places=2)
|
|
conn.close()
|
|
|
|
def test_pending_sell_blocks_true_pnl(self):
|
|
from lib.options.options_pnl_settle_lib import resolve_true_pnl_usdt_from_trade
|
|
from lib.options.options_spot_bridge_lib import (
|
|
BRIDGE_PENDING_SELL,
|
|
add_bridge_buy_fill,
|
|
add_bridge_close_sell_fill,
|
|
ensure_bridge_table,
|
|
insert_bridge,
|
|
update_bridge,
|
|
)
|
|
|
|
conn = sqlite3.connect(":memory:")
|
|
conn.row_factory = sqlite3.Row
|
|
ensure_bridge_table(conn)
|
|
bid = insert_bridge(conn, underlying="ETH", status="holding", budget_usdt=3000)
|
|
add_bridge_buy_fill(conn, bid, usdt_spent=3000.0, coin_bought=1.0)
|
|
add_bridge_close_sell_fill(conn, bid, usdt_recovered=1500.0, coin_sold=0.5)
|
|
update_bridge(conn, bid, status=BRIDGE_PENDING_SELL)
|
|
pnl, src = resolve_true_pnl_usdt_from_trade(
|
|
premium_paid=0.01,
|
|
premium_received=0.02,
|
|
realized_pnl=0.01,
|
|
open_index_px=3000.0,
|
|
close_index_px=3000.0,
|
|
premium_ccy="ETH",
|
|
conn=conn,
|
|
bridge_id=bid,
|
|
)
|
|
self.assertEqual(src, "pending_sell")
|
|
self.assertIsNone(pnl)
|
|
conn.close()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|