ruff format: More updates to tests

This commit is contained in:
Matthias
2024-05-12 15:38:09 +02:00
parent 23427bec08
commit 099b1fc8c4
5 changed files with 852 additions and 795 deletions

View File

@@ -13,17 +13,22 @@ from tests.conftest import EXMS, log_has, log_has_re, patch_exchange
def test_import_kraken_trades_from_csv(testdatadir, tmp_path, caplog, default_conf_usdt, mocker):
with pytest.raises(OperationalException, match="This function is only for the kraken exchange"):
import_kraken_trades_from_csv(default_conf_usdt, 'feather')
import_kraken_trades_from_csv(default_conf_usdt, "feather")
default_conf_usdt['exchange']['name'] = 'kraken'
default_conf_usdt["exchange"]["name"] = "kraken"
patch_exchange(mocker, id='kraken')
mocker.patch(f'{EXMS}.markets', PropertyMock(return_value={
'BCH/EUR': {'symbol': 'BCH/EUR', 'id': 'BCHEUR', 'altname': 'BCHEUR'},
}))
dstfile = tmp_path / 'BCH_EUR-trades.feather'
patch_exchange(mocker, id="kraken")
mocker.patch(
f"{EXMS}.markets",
PropertyMock(
return_value={
"BCH/EUR": {"symbol": "BCH/EUR", "id": "BCHEUR", "altname": "BCHEUR"},
}
),
)
dstfile = tmp_path / "BCH_EUR-trades.feather"
assert not dstfile.is_file()
default_conf_usdt['datadir'] = tmp_path
default_conf_usdt["datadir"] = tmp_path
# There's 2 files in this tree, containing a total of 2 days.
# tests/testdata/kraken/
# └── trades_csv
@@ -31,29 +36,31 @@ def test_import_kraken_trades_from_csv(testdatadir, tmp_path, caplog, default_co
# └── incremental_q2
# └── BCHEUR.csv <-- 2023-01-02
copytree(testdatadir / 'kraken/trades_csv', tmp_path / 'trades_csv')
copytree(testdatadir / "kraken/trades_csv", tmp_path / "trades_csv")
import_kraken_trades_from_csv(default_conf_usdt, 'feather')
import_kraken_trades_from_csv(default_conf_usdt, "feather")
assert log_has("Found csv files for BCHEUR.", caplog)
assert log_has("Converting pairs: BCH/EUR.", caplog)
assert log_has_re(r"BCH/EUR: 340 trades.* 2023-01-01.* 2023-01-02.*", caplog)
assert dstfile.is_file()
dh = get_datahandler(tmp_path, 'feather')
trades = dh.trades_load('BCH_EUR', TradingMode.SPOT)
dh = get_datahandler(tmp_path, "feather")
trades = dh.trades_load("BCH_EUR", TradingMode.SPOT)
assert len(trades) == 340
assert trades['date'].min().to_pydatetime() == datetime(2023, 1, 1, 0, 3, 56,
tzinfo=timezone.utc)
assert trades['date'].max().to_pydatetime() == datetime(2023, 1, 2, 23, 17, 3,
tzinfo=timezone.utc)
assert trades["date"].min().to_pydatetime() == datetime(
2023, 1, 1, 0, 3, 56, tzinfo=timezone.utc
)
assert trades["date"].max().to_pydatetime() == datetime(
2023, 1, 2, 23, 17, 3, tzinfo=timezone.utc
)
# ID is not filled
assert len(trades.loc[trades['id'] != '']) == 0
assert len(trades.loc[trades["id"] != ""]) == 0
caplog.clear()
default_conf_usdt['pairs'] = ['XRP/EUR']
default_conf_usdt["pairs"] = ["XRP/EUR"]
# Filtered to non-existing pair
import_kraken_trades_from_csv(default_conf_usdt, 'feather')
import_kraken_trades_from_csv(default_conf_usdt, "feather")
assert log_has("Found csv files for BCHEUR.", caplog)
assert log_has("No data found for pairs XRP/EUR.", caplog)