adelorenzo/stable
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update test_freqtradebot.py

Browse Source
This commit is contained in:
Kavinkumar 2022-03-08 11:23:01 +05:30
parent 7ded76d6b4
commit 5ad1c59a6b
1 changed files with 95 additions and 74 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

51
tests/test_freqtradebot.py
View File

@ -4804,7 +4804,7 @@ def test_position_adjust2(mocker, default_conf_usdt, fee) -> None:
assert trade.is_open is False assert trade.is_open is False
@pytest.mark.parametrize('orders, profit, res', [ @pytest.mark.parametrize('orders, res', [
( (
( (
# side ampunt, price # side ampunt, price
@ -4813,7 +4813,16 @@ def test_position_adjust2(mocker, default_conf_usdt, fee) -> None:
('sell', 50, 12), ('sell', 50, 12),
('sell', 100, 20), ('sell', 100, 20),
('sell', 50, 5), ('sell', 50, 5),
), 336.625, ()), ),
(
# amount, open_rate, stake_amount, realized_profit, unrealized_profit
(100.0, 10.0, 1000.0, 0.0, -5.0),
(200.0, 12.5, 2500.0, 0.0, 486.25),
(150.0, 12.686616791354945, 1902.9925187032418, -153.375, -112.25),
(50.0, -1.7406483790523748, -87.03241895261874, 588.5, 1084.75),
(50.0, -1.7406483790523748, -87.03241895261874, 588.5, 336.625),
)
),
( (
( (
('buy', 100, 3), ('buy', 100, 3),
@ -4822,9 +4831,18 @@ def test_position_adjust2(mocker, default_conf_usdt, fee) -> None:
('buy', 150, 15), ('buy', 150, 15),
('sell', 100, 19), ('sell', 100, 19),
('sell', 150, 23), ('sell', 150, 23),
), 3175.75, ()), ),
(
(100.0, 3.0, 300.0, 0.0, -1.5),
(200.0, 5.0, 1000.0, 0.0, 394.0),
(100.0, -0.9451371571072332, -94.51371571072332, 395.5, 1192.0),
(250.0, 8.621945137157107, 2155.4862842892767, 395.5, 1579.75),
(150.0, 1.766417290108061, 264.96259351620915, 787.0, 2577.25),
(150.0, 1.766417290108061, 264.96259351620915, 787.0, 3175.75),
)
),
]) ])
def test_position_adjust3(mocker, default_conf_usdt, fee, orders, profit, res) -> None: def test_position_adjust3(mocker, default_conf_usdt, fee, orders, res) -> None:
default_conf_usdt.update({ default_conf_usdt.update({
"position_adjustment_enable": True, "position_adjustment_enable": True,
@ -4832,14 +4850,13 @@ def test_position_adjust3(mocker, default_conf_usdt, fee, orders, profit, res) -
"stake_amount": 200.0, "stake_amount": 200.0,
"dry_run_wallet": 1000.0, "dry_run_wallet": 1000.0,
}) })
if 1:
patch_RPCManager(mocker) patch_RPCManager(mocker)
patch_exchange(mocker) patch_exchange(mocker)
patch_wallet(mocker, free=10000) patch_wallet(mocker, free=10000)
freqtrade = FreqtradeBot(default_conf_usdt) freqtrade = FreqtradeBot(default_conf_usdt)
trade = None trade = None
freqtrade.strategy.confirm_trade_entry = MagicMock(return_value=True) freqtrade.strategy.confirm_trade_entry = MagicMock(return_value=True)
for idx2, order in enumerate(orders): for idx, order in enumerate(orders):
amount = order[1] amount = order[1]
price = order[2] price = order[2]
price_mock = MagicMock(return_value=price) price_mock = MagicMock(return_value=price)
@ -4869,8 +4886,8 @@ def test_position_adjust3(mocker, default_conf_usdt, fee, orders, profit, res) -
'amount': amount, 'amount': amount,
'filled': amount, 'filled': amount,
'ft_is_open': False, 'ft_is_open': False,
'id': f'60{idx2}', 'id': f'60{idx}',
'order_id': f'60{idx2}' 'order_id': f'60{idx}'
} }
mocker.patch('freqtrade.exchange.Exchange.create_order', mocker.patch('freqtrade.exchange.Exchange.create_order',
MagicMock(return_value=closed_successful_order)) MagicMock(return_value=closed_successful_order))
@ -4883,25 +4900,29 @@ def test_position_adjust3(mocker, default_conf_usdt, fee, orders, profit, res) -
trade=trade, limit=price, trade=trade, limit=price,
sell_reason=SellCheckTuple(sell_type=SellType.STOP_LOSS), sell_reason=SellCheckTuple(sell_type=SellType.STOP_LOSS),
sub_trade_amt=amount) sub_trade_amt=amount)
# Should create an closed trade with an no open order id
# Order is filled and trade is open
orders1 = Order.query.all() orders1 = Order.query.all()
assert orders1 assert orders1
assert len(orders1) == idx2 + 1 assert len(orders1) == idx + 1
trade = Trade.query.first() trade = Trade.query.first()
assert trade assert trade
if idx2 < len(orders) - 1: if idx < len(orders) - 1:
assert trade.is_open is True assert trade.is_open is True
assert trade.open_order_id is None assert trade.open_order_id is None
assert trade.amount == res[idx][0]
assert trade.open_rate == res[idx][1]
assert trade.stake_amount == res[idx][2]
assert pytest.approx(trade.realized_profit) == res[idx][3]
assert trade.calc_profit(order[2]) == res[idx][4]
# Make sure the closed order is found as the second order.
order_obj = trade.select_order(order[0], False) order_obj = trade.select_order(order[0], False)
assert order_obj.order_id == f'60{idx2}' assert order_obj.order_id == f'60{idx}'
trade = Trade.query.first() trade = Trade.query.first()
assert trade assert trade
assert trade.open_order_id is None assert trade.open_order_id is None
assert trade.close_profit_abs == profit assert trade.close_profit_abs == res[idx][4]
assert trade.is_open is False assert trade.is_open is False