Add explicit test for stoploss_from_open
This commit is contained in:
parent
c03c3a5706
commit
d0045673fa
@ -169,6 +169,36 @@ def test_stoploss_from_open(side, profitrange):
|
|||||||
assert pytest.approx(stop_price) == expected_stop_price
|
assert pytest.approx(stop_price) == expected_stop_price
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.parametrize("side,rel_stop,curr_profit,leverage,expected", [
|
||||||
|
# profit range for long is [-1, inf] while for shorts is [-inf, 1]
|
||||||
|
("long", 0, -1, 1, 1),
|
||||||
|
("long", 0, 0.1, 1, 0.09090909),
|
||||||
|
("long", -0.1, 0.1, 1, 0.18181818),
|
||||||
|
("long", 0.1, 0.2, 1, 0.08333333),
|
||||||
|
("long", 0.1, 0.5, 1, 0.266666666),
|
||||||
|
("long", 0.1, 5, 1, 0.816666666), # 500% profit, set stoploss to 10% above open price
|
||||||
|
|
||||||
|
("short", 0, 0.1, 1, 0.1111111),
|
||||||
|
("short", -0.1, 0.1, 1, 0.2222222),
|
||||||
|
("short", 0.1, 0.2, 1, 0.125),
|
||||||
|
("short", 0.1, 1, 1, 1),
|
||||||
|
])
|
||||||
|
def test_stoploss_from_open_leverage(side, rel_stop, curr_profit, leverage, expected):
|
||||||
|
|
||||||
|
stoploss = stoploss_from_open(rel_stop, curr_profit, side == 'short')
|
||||||
|
assert pytest.approx(stoploss) == expected
|
||||||
|
open_rate = 100
|
||||||
|
if stoploss != 1:
|
||||||
|
if side == 'long':
|
||||||
|
current_rate = open_rate * (1 + curr_profit)
|
||||||
|
stop = current_rate * (1 - stoploss)
|
||||||
|
assert pytest.approx(stop) == open_rate * (1 + rel_stop)
|
||||||
|
else:
|
||||||
|
current_rate = open_rate * (1 - curr_profit)
|
||||||
|
stop = current_rate * (1 + stoploss)
|
||||||
|
assert pytest.approx(stop) == open_rate * (1 - rel_stop)
|
||||||
|
|
||||||
|
|
||||||
def test_stoploss_from_absolute():
|
def test_stoploss_from_absolute():
|
||||||
assert pytest.approx(stoploss_from_absolute(90, 100)) == 1 - (90 / 100)
|
assert pytest.approx(stoploss_from_absolute(90, 100)) == 1 - (90 / 100)
|
||||||
assert pytest.approx(stoploss_from_absolute(90, 100)) == 0.1
|
assert pytest.approx(stoploss_from_absolute(90, 100)) == 0.1
|
||||||
|
Loading…
Reference in New Issue
Block a user