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

Enhance test for dataprovider

Browse Source
This commit is contained in:
Matthias
2024-04-05 07:07:12 +02:00
parent e0f1b1e48e
commit 28e4711ee9
1 changed files with 12 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/data/test_dataprovider.py
+12 -3
View File
@@ -231,8 +231,8 @@ def test_emit_df(mocker, default_conf, ohlcv_history):
def test_refresh(mocker, default_conf): def test_refresh(mocker, default_conf):
refresh_mock = MagicMock() refresh_mock = mocker.patch(f"{EXMS}.refresh_latest_ohlcv")
mocker.patch(f"{EXMS}.refresh_latest_ohlcv", refresh_mock) mock_refresh_trades = mocker.patch(f"{EXMS}.refresh_latest_trades")
exchange = get_patched_exchange(mocker, default_conf, id="binance") exchange = get_patched_exchange(mocker, default_conf, id="binance")
timeframe = default_conf["timeframe"] timeframe = default_conf["timeframe"]
@@ -242,7 +242,7 @@ def test_refresh(mocker, default_conf):
dp = DataProvider(default_conf, exchange) dp = DataProvider(default_conf, exchange)
dp.refresh(pairs) dp.refresh(pairs)
assert mock_refresh_trades.call_count == 0
assert refresh_mock.call_count == 1 assert refresh_mock.call_count == 1
assert len(refresh_mock.call_args[0]) == 1 assert len(refresh_mock.call_args[0]) == 1
assert len(refresh_mock.call_args[0][0]) == len(pairs) assert len(refresh_mock.call_args[0][0]) == len(pairs)
@@ -250,11 +250,20 @@ def test_refresh(mocker, default_conf):
refresh_mock.reset_mock() refresh_mock.reset_mock()
dp.refresh(pairs, pairs_non_trad) dp.refresh(pairs, pairs_non_trad)
assert mock_refresh_trades.call_count == 0
assert refresh_mock.call_count == 1 assert refresh_mock.call_count == 1
assert len(refresh_mock.call_args[0]) == 1 assert len(refresh_mock.call_args[0]) == 1
assert len(refresh_mock.call_args[0][0]) == len(pairs) + len(pairs_non_trad) assert len(refresh_mock.call_args[0][0]) == len(pairs) + len(pairs_non_trad)
assert refresh_mock.call_args[0][0] == pairs + pairs_non_trad assert refresh_mock.call_args[0][0] == pairs + pairs_non_trad
# Test with public trades
refresh_mock.reset_mock()
refresh_mock.reset_mock()
default_conf['exchange']['use_public_trades'] = True
dp.refresh(pairs, pairs_non_trad)
assert mock_refresh_trades.call_count == 1
assert refresh_mock.call_count == 1
def test_orderbook(mocker, default_conf, order_book_l2): def test_orderbook(mocker, default_conf, order_book_l2):
api_mock = MagicMock() api_mock = MagicMock()