Improve test

This commit is contained in:
Matthias
2024-02-24 14:05:25 +01:00
parent 9dd5967275
commit 2704f6e758
+7 -1
View File
@@ -789,10 +789,11 @@ def test_VolatilityFilter_sort(mocker, whitelist_conf, time_machine, sort_direct
('TKN/BTC', '1d', CandleType.SPOT): df2, ('TKN/BTC', '1d', CandleType.SPOT): df2,
} }
ohlcv_mock = MagicMock(return_value=ohlcv_data)
mocker.patch.multiple( mocker.patch.multiple(
EXMS, EXMS,
exchange_has=MagicMock(return_value=True), exchange_has=MagicMock(return_value=True),
refresh_latest_ohlcv=MagicMock(return_value=ohlcv_data), refresh_latest_ohlcv=ohlcv_mock,
) )
exchange = get_patched_exchange(mocker, whitelist_conf) exchange = get_patched_exchange(mocker, whitelist_conf)
@@ -801,11 +802,16 @@ def test_VolatilityFilter_sort(mocker, whitelist_conf, time_machine, sort_direct
assert exchange.ohlcv_candle_limit.call_count == 1 assert exchange.ohlcv_candle_limit.call_count == 1
plm.refresh_pairlist() plm.refresh_pairlist()
assert ohlcv_mock.call_count == 1
assert exchange.ohlcv_candle_limit.call_count == 1 assert exchange.ohlcv_candle_limit.call_count == 1
assert plm.whitelist == ( assert plm.whitelist == (
['ETH/BTC', 'TKN/BTC'] if sort_direction == 'asc' else ['TKN/BTC', 'ETH/BTC'] ['ETH/BTC', 'TKN/BTC'] if sort_direction == 'asc' else ['TKN/BTC', 'ETH/BTC']
) )
plm.refresh_pairlist()
assert exchange.ohlcv_candle_limit.call_count == 1
assert ohlcv_mock.call_count == 1
def test_ShuffleFilter_init(mocker, whitelist_conf, caplog) -> None: def test_ShuffleFilter_init(mocker, whitelist_conf, caplog) -> None:
whitelist_conf['pairlists'] = [ whitelist_conf['pairlists'] = [