dekun
61 lines
1.5 KiB
Python
61 lines
1.5 KiB
Python
"""hub_calculator_lib 测算逻辑。"""
|
|
|
|
from hub_calculator_lib import calc_roll_calculator, calc_trend_calculator
|
|
|
|
|
|
def test_trend_calculator_long_basic():
|
|
data, err = calc_trend_calculator(
|
|
direction="long",
|
|
capital_usdt=1000,
|
|
risk_percent=5,
|
|
leverage=5,
|
|
entry_price=100,
|
|
stop_loss=95,
|
|
add_upper=110,
|
|
take_profit=120,
|
|
dca_legs=3,
|
|
contract_size=1,
|
|
)
|
|
assert err is None
|
|
assert data is not None
|
|
assert data["risk_budget_u"] == 50.0
|
|
assert len(data["rows"]) >= 2
|
|
assert data["rows"][0]["label"] == "首仓"
|
|
assert data["first_profit_u"] is not None
|
|
assert data["first_profit_u"] > 0
|
|
|
|
|
|
def test_trend_calculator_short_rejects_bad_bounds():
|
|
data, err = calc_trend_calculator(
|
|
direction="short",
|
|
capital_usdt=1000,
|
|
risk_percent=5,
|
|
leverage=5,
|
|
entry_price=100,
|
|
stop_loss=90,
|
|
add_upper=110,
|
|
take_profit=80,
|
|
dca_legs=3,
|
|
)
|
|
assert data is None
|
|
assert err is not None
|
|
|
|
|
|
def test_roll_calculator_long():
|
|
data, err = calc_roll_calculator(
|
|
direction="long",
|
|
capital_usdt=1000,
|
|
risk_percent=5,
|
|
qty_existing=10,
|
|
entry_existing=100,
|
|
take_profit=120,
|
|
add_price=105,
|
|
new_stop_loss=98,
|
|
legs_done=0,
|
|
)
|
|
assert err is None
|
|
assert data is not None
|
|
assert data["add_contracts"] > 0
|
|
assert data["qty_after"] > 10
|
|
assert data["profit_at_tp_u"] is not None
|