mirror of
https://github.com/freqtrade/freqtrade.git
synced 2024-09-20 09:31:12 +00:00
371 lines
12 KiB
Python
371 lines
12 KiB
Python
from datetime import datetime, timezone
|
|
from unittest.mock import MagicMock
|
|
|
|
import pandas as pd
|
|
import pytest
|
|
|
|
from freqtrade.data.converter import ohlcv_to_dataframe
|
|
from freqtrade.enums import CandleType
|
|
from freqtrade.exceptions import OperationalException
|
|
from freqtrade.plugins.pairlist.PercentChangePairList import PercentChangePairList
|
|
from freqtrade.plugins.pairlistmanager import PairListManager
|
|
from tests.conftest import (
|
|
EXMS,
|
|
generate_test_data_raw,
|
|
get_patched_exchange,
|
|
get_patched_freqtradebot,
|
|
)
|
|
|
|
|
|
@pytest.fixture(scope="function")
|
|
def rpl_config(default_conf):
|
|
default_conf["stake_currency"] = "USDT"
|
|
|
|
default_conf["exchange"]["pair_whitelist"] = [
|
|
"ETH/USDT",
|
|
"XRP/USDT",
|
|
]
|
|
default_conf["exchange"]["pair_blacklist"] = ["BLK/USDT"]
|
|
|
|
return default_conf
|
|
|
|
|
|
def test_volume_change_pair_list_init_exchange_support(mocker, rpl_config):
|
|
rpl_config["pairlists"] = [
|
|
{
|
|
"method": "PercentChangePairList",
|
|
"number_assets": 2,
|
|
"sort_key": "percentage",
|
|
"min_value": 0,
|
|
"refresh_period": 86400,
|
|
}
|
|
]
|
|
|
|
with pytest.raises(
|
|
OperationalException,
|
|
match=r"Exchange does not support dynamic whitelist in this configuration. "
|
|
r"Please edit your config and either remove PercentChangePairList, "
|
|
r"or switch to using candles. and restart the bot.",
|
|
):
|
|
get_patched_freqtradebot(mocker, rpl_config)
|
|
|
|
|
|
def test_volume_change_pair_list_init_wrong_refresh_period(mocker, rpl_config):
|
|
rpl_config["pairlists"] = [
|
|
{
|
|
"method": "PercentChangePairList",
|
|
"number_assets": 2,
|
|
"sort_key": "percentage",
|
|
"min_value": 0,
|
|
"refresh_period": 1800,
|
|
"lookback_days": 4,
|
|
}
|
|
]
|
|
|
|
with pytest.raises(
|
|
OperationalException,
|
|
match=r"Refresh period of 1800 seconds is smaller than one "
|
|
r"timeframe of 1d. Please adjust refresh_period "
|
|
r"to at least 86400 and restart the bot.",
|
|
):
|
|
get_patched_freqtradebot(mocker, rpl_config)
|
|
|
|
|
|
def test_volume_change_pair_list_init_wrong_lookback_period(mocker, rpl_config):
|
|
rpl_config["pairlists"] = [
|
|
{
|
|
"method": "PercentChangePairList",
|
|
"number_assets": 2,
|
|
"sort_key": "percentage",
|
|
"min_value": 0,
|
|
"refresh_period": 86400,
|
|
"lookback_days": 3,
|
|
"lookback_period": 3,
|
|
}
|
|
]
|
|
|
|
with pytest.raises(
|
|
OperationalException,
|
|
match=r"Ambiguous configuration: lookback_days "
|
|
r"and lookback_period both set in pairlist config. "
|
|
r"Please set lookback_days only or lookback_period "
|
|
r"and lookback_timeframe and restart the bot.",
|
|
):
|
|
get_patched_freqtradebot(mocker, rpl_config)
|
|
|
|
rpl_config["pairlists"] = [
|
|
{
|
|
"method": "PercentChangePairList",
|
|
"number_assets": 2,
|
|
"sort_key": "percentage",
|
|
"min_value": 0,
|
|
"refresh_period": 86400,
|
|
"lookback_days": 1001,
|
|
}
|
|
]
|
|
|
|
with pytest.raises(
|
|
OperationalException,
|
|
match=r"ChangeFilter requires lookback_period to not exceed"
|
|
r" exchange max request size \(1000\)",
|
|
):
|
|
get_patched_freqtradebot(mocker, rpl_config)
|
|
|
|
|
|
def test_volume_change_pair_list_init_wrong_config(mocker, rpl_config):
|
|
rpl_config["pairlists"] = [
|
|
{
|
|
"method": "PercentChangePairList",
|
|
"sort_key": "percentage",
|
|
"min_value": 0,
|
|
"refresh_period": 86400,
|
|
}
|
|
]
|
|
|
|
with pytest.raises(
|
|
OperationalException,
|
|
match=r"`number_assets` not specified. Please check your configuration "
|
|
r'for "pairlist.config.number_assets"',
|
|
):
|
|
get_patched_freqtradebot(mocker, rpl_config)
|
|
|
|
|
|
def test_gen_pairlist_with_valid_change_pair_list_config(mocker, rpl_config, tickers, time_machine):
|
|
rpl_config["pairlists"] = [
|
|
{
|
|
"method": "PercentChangePairList",
|
|
"number_assets": 2,
|
|
"sort_key": "percentage",
|
|
"min_value": 0,
|
|
"refresh_period": 86400,
|
|
"lookback_days": 4,
|
|
}
|
|
]
|
|
start = datetime(2024, 8, 1, 0, 0, 0, 0, tzinfo=timezone.utc)
|
|
time_machine.move_to(start, tick=False)
|
|
|
|
mock_ohlcv_data = {
|
|
("ETH/USDT", "1d", CandleType.SPOT): pd.DataFrame(
|
|
ohlcv_to_dataframe(
|
|
generate_test_data_raw("1d", 100, start.strftime("%Y-%m-%d"), random_seed=12),
|
|
"1d",
|
|
pair="ETH/USDT",
|
|
fill_missing=True,
|
|
)
|
|
),
|
|
("BTC/USDT", "1d", CandleType.SPOT): pd.DataFrame(
|
|
ohlcv_to_dataframe(
|
|
generate_test_data_raw("1d", 100, start.strftime("%Y-%m-%d"), random_seed=13),
|
|
"1d",
|
|
pair="BTC/USDT",
|
|
fill_missing=True,
|
|
)
|
|
),
|
|
("XRP/USDT", "1d", CandleType.SPOT): pd.DataFrame(
|
|
ohlcv_to_dataframe(
|
|
generate_test_data_raw("1d", 100, start.strftime("%Y-%m-%d"), random_seed=14),
|
|
"1d",
|
|
pair="XRP/USDT",
|
|
fill_missing=True,
|
|
)
|
|
),
|
|
("NEO/USDT", "1d", CandleType.SPOT): pd.DataFrame(
|
|
ohlcv_to_dataframe(
|
|
generate_test_data_raw("1d", 100, start.strftime("%Y-%m-%d"), random_seed=15),
|
|
"1d",
|
|
pair="NEO/USDT",
|
|
fill_missing=True,
|
|
)
|
|
),
|
|
("TKN/USDT", "1d", CandleType.SPOT): pd.DataFrame(
|
|
# Make sure always have highest percentage
|
|
{
|
|
"timestamp": [
|
|
"2024-07-01 00:00:00",
|
|
"2024-07-01 01:00:00",
|
|
"2024-07-01 02:00:00",
|
|
"2024-07-01 03:00:00",
|
|
"2024-07-01 04:00:00",
|
|
"2024-07-01 05:00:00",
|
|
],
|
|
"open": [100, 102, 101, 103, 104, 105],
|
|
"high": [102, 103, 102, 104, 105, 106],
|
|
"low": [99, 101, 100, 102, 103, 104],
|
|
"close": [101, 102, 103, 104, 105, 106],
|
|
"volume": [1000, 1500, 2000, 2500, 3000, 3500],
|
|
}
|
|
),
|
|
}
|
|
|
|
mocker.patch(f"{EXMS}.refresh_latest_ohlcv", MagicMock(return_value=mock_ohlcv_data))
|
|
|
|
exchange = get_patched_exchange(mocker, rpl_config, exchange="binance")
|
|
pairlistmanager = PairListManager(exchange, rpl_config)
|
|
|
|
remote_pairlist = PercentChangePairList(
|
|
exchange, pairlistmanager, rpl_config, rpl_config["pairlists"][0], 0
|
|
)
|
|
|
|
result = remote_pairlist.gen_pairlist(tickers)
|
|
|
|
assert len(result) == 2
|
|
assert result == ["NEO/USDT", "TKN/USDT"]
|
|
|
|
|
|
def test_filter_pairlist_with_empty_ticker(mocker, rpl_config, tickers, time_machine):
|
|
rpl_config["pairlists"] = [
|
|
{
|
|
"method": "PercentChangePairList",
|
|
"number_assets": 2,
|
|
"sort_key": "percentage",
|
|
"min_value": 0,
|
|
"refresh_period": 86400,
|
|
"sort_direction": "asc",
|
|
"lookback_days": 4,
|
|
}
|
|
]
|
|
start = datetime(2024, 8, 1, 0, 0, 0, 0, tzinfo=timezone.utc)
|
|
time_machine.move_to(start, tick=False)
|
|
|
|
mock_ohlcv_data = {
|
|
("ETH/USDT", "1d", CandleType.SPOT): pd.DataFrame(
|
|
{
|
|
"timestamp": [
|
|
"2024-07-01 00:00:00",
|
|
"2024-07-01 01:00:00",
|
|
"2024-07-01 02:00:00",
|
|
"2024-07-01 03:00:00",
|
|
"2024-07-01 04:00:00",
|
|
"2024-07-01 05:00:00",
|
|
],
|
|
"open": [100, 102, 101, 103, 104, 105],
|
|
"high": [102, 103, 102, 104, 105, 106],
|
|
"low": [99, 101, 100, 102, 103, 104],
|
|
"close": [101, 102, 103, 104, 105, 105],
|
|
"volume": [1000, 1500, 2000, 2500, 3000, 3500],
|
|
}
|
|
),
|
|
("XRP/USDT", "1d", CandleType.SPOT): pd.DataFrame(
|
|
{
|
|
"timestamp": [
|
|
"2024-07-01 00:00:00",
|
|
"2024-07-01 01:00:00",
|
|
"2024-07-01 02:00:00",
|
|
"2024-07-01 03:00:00",
|
|
"2024-07-01 04:00:00",
|
|
"2024-07-01 05:00:00",
|
|
],
|
|
"open": [100, 102, 101, 103, 104, 105],
|
|
"high": [102, 103, 102, 104, 105, 106],
|
|
"low": [99, 101, 100, 102, 103, 104],
|
|
"close": [101, 102, 103, 104, 105, 104],
|
|
"volume": [1000, 1500, 2000, 2500, 3000, 3400],
|
|
}
|
|
),
|
|
}
|
|
|
|
mocker.patch(f"{EXMS}.refresh_latest_ohlcv", MagicMock(return_value=mock_ohlcv_data))
|
|
exchange = get_patched_exchange(mocker, rpl_config, exchange="binance")
|
|
pairlistmanager = PairListManager(exchange, rpl_config)
|
|
|
|
remote_pairlist = PercentChangePairList(
|
|
exchange, pairlistmanager, rpl_config, rpl_config["pairlists"][0], 0
|
|
)
|
|
|
|
result = remote_pairlist.filter_pairlist(rpl_config["exchange"]["pair_whitelist"], {})
|
|
|
|
assert len(result) == 2
|
|
assert result == ["XRP/USDT", "ETH/USDT"]
|
|
|
|
|
|
def test_filter_pairlist_with_max_value_set(mocker, rpl_config, tickers, time_machine):
|
|
rpl_config["pairlists"] = [
|
|
{
|
|
"method": "PercentChangePairList",
|
|
"number_assets": 2,
|
|
"sort_key": "percentage",
|
|
"min_value": 0,
|
|
"max_value": 15,
|
|
"refresh_period": 86400,
|
|
"lookback_days": 4,
|
|
}
|
|
]
|
|
|
|
start = datetime(2024, 8, 1, 0, 0, 0, 0, tzinfo=timezone.utc)
|
|
time_machine.move_to(start, tick=False)
|
|
|
|
mock_ohlcv_data = {
|
|
("ETH/USDT", "1d", CandleType.SPOT): pd.DataFrame(
|
|
{
|
|
"timestamp": [
|
|
"2024-07-01 00:00:00",
|
|
"2024-07-01 01:00:00",
|
|
"2024-07-01 02:00:00",
|
|
"2024-07-01 03:00:00",
|
|
"2024-07-01 04:00:00",
|
|
"2024-07-01 05:00:00",
|
|
],
|
|
"open": [100, 102, 101, 103, 104, 105],
|
|
"high": [102, 103, 102, 104, 105, 106],
|
|
"low": [99, 101, 100, 102, 103, 104],
|
|
"close": [101, 102, 103, 104, 105, 106],
|
|
"volume": [1000, 1500, 2000, 1800, 2400, 2500],
|
|
}
|
|
),
|
|
("XRP/USDT", "1d", CandleType.SPOT): pd.DataFrame(
|
|
{
|
|
"timestamp": [
|
|
"2024-07-01 00:00:00",
|
|
"2024-07-01 01:00:00",
|
|
"2024-07-01 02:00:00",
|
|
"2024-07-01 03:00:00",
|
|
"2024-07-01 04:00:00",
|
|
"2024-07-01 05:00:00",
|
|
],
|
|
"open": [100, 102, 101, 103, 104, 105],
|
|
"high": [102, 103, 102, 104, 105, 106],
|
|
"low": [99, 101, 100, 102, 103, 104],
|
|
"close": [101, 102, 103, 104, 105, 101],
|
|
"volume": [1000, 1500, 2000, 2500, 3000, 3500],
|
|
}
|
|
),
|
|
}
|
|
|
|
mocker.patch(f"{EXMS}.refresh_latest_ohlcv", MagicMock(return_value=mock_ohlcv_data))
|
|
exchange = get_patched_exchange(mocker, rpl_config, exchange="binance")
|
|
pairlistmanager = PairListManager(exchange, rpl_config)
|
|
|
|
remote_pairlist = PercentChangePairList(
|
|
exchange, pairlistmanager, rpl_config, rpl_config["pairlists"][0], 0
|
|
)
|
|
|
|
result = remote_pairlist.filter_pairlist(rpl_config["exchange"]["pair_whitelist"], {})
|
|
|
|
assert len(result) == 1
|
|
assert result == ["ETH/USDT"]
|
|
|
|
|
|
def test_gen_pairlist_from_tickers(mocker, rpl_config, tickers):
|
|
rpl_config["pairlists"] = [
|
|
{
|
|
"method": "PercentChangePairList",
|
|
"number_assets": 2,
|
|
"sort_key": "percentage",
|
|
"min_value": 0,
|
|
}
|
|
]
|
|
|
|
mocker.patch(f"{EXMS}.exchange_has", MagicMock(return_value=True))
|
|
|
|
exchange = get_patched_exchange(mocker, rpl_config, exchange="binance")
|
|
pairlistmanager = PairListManager(exchange, rpl_config)
|
|
|
|
remote_pairlist = PercentChangePairList(
|
|
exchange, pairlistmanager, rpl_config, rpl_config["pairlists"][0], 0
|
|
)
|
|
|
|
result = remote_pairlist.gen_pairlist(tickers.return_value)
|
|
|
|
assert len(result) == 1
|
|
assert result == ["ETH/USDT"]
|