mohammadian7/freqtrade
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

tests: improve assert accuracy

Browse Source
This commit is contained in:
Matthias
2024-11-30 13:50:12 +01:00
parent 88d77658f8
commit 2b350eb872
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/strategy/test_strategy_safe_wrapper.py
+5 -4
View File
@@ -49,20 +49,21 @@ def test_strategy_safe_wrapper(value):
@pytest.mark.usefixtures("init_persistence") @pytest.mark.usefixtures("init_persistence")
def test_strategy_safe_wrapper_trade_copy(fee): def test_strategy_safe_wrapper_trade_copy(fee):
create_mock_trades(fee) create_mock_trades(fee)
trade_ = Trade.get_open_trades()[0]
def working_method(trade): def working_method(trade):
assert len(trade.orders) > 0 assert len(trade.orders) > 0
assert trade.orders assert trade.orders
trade.orders = [] trade.orders = []
assert len(trade.orders) == 0 assert len(trade.orders) == 0
assert id(trade_) != id(trade)
return trade return trade
trade = Trade.get_open_trades()[0]
# Don't assert anything before strategy_wrapper. # Don't assert anything before strategy_wrapper.
# This ensures that relationship loading works correctly. # This ensures that relationship loading works correctly.
ret = strategy_safe_wrapper(working_method, message="DeadBeef")(trade=trade) ret = strategy_safe_wrapper(working_method, message="DeadBeef")(trade=trade_)
assert isinstance(ret, Trade) assert isinstance(ret, Trade)
assert id(trade) != id(ret) assert id(trade_) != id(ret)
# Did not modify the original order # Did not modify the original order
assert len(trade.orders) > 0 assert len(trade_.orders) > 0
assert len(ret.orders) == 0 assert len(ret.orders) == 0