2018-01-28 07:38:41 +00:00
|
|
|
# pragma pylint: disable=missing-docstring, protected-access, C0103
|
2018-01-15 08:35:11 +00:00
|
|
|
import logging
|
2018-03-24 20:56:20 +00:00
|
|
|
import os
|
|
|
|
|
2018-03-25 14:28:04 +00:00
|
|
|
import pytest
|
|
|
|
|
2018-06-23 09:14:31 +00:00
|
|
|
from freqtrade.strategy import import_strategy
|
|
|
|
from freqtrade.strategy.default_strategy import DefaultStrategy
|
2018-03-24 20:56:20 +00:00
|
|
|
from freqtrade.strategy.interface import IStrategy
|
2018-03-24 17:11:21 +00:00
|
|
|
from freqtrade.strategy.resolver import StrategyResolver
|
2018-01-15 08:35:11 +00:00
|
|
|
|
|
|
|
|
2018-06-23 09:14:31 +00:00
|
|
|
def test_import_strategy(caplog):
|
|
|
|
caplog.set_level(logging.DEBUG)
|
2018-07-16 05:11:17 +00:00
|
|
|
default_config = {}
|
2018-06-23 09:14:31 +00:00
|
|
|
|
2018-07-16 05:11:17 +00:00
|
|
|
strategy = DefaultStrategy(default_config)
|
2018-06-23 09:14:31 +00:00
|
|
|
strategy.some_method = lambda *args, **kwargs: 42
|
|
|
|
|
|
|
|
assert strategy.__module__ == 'freqtrade.strategy.default_strategy'
|
|
|
|
assert strategy.some_method() == 42
|
|
|
|
|
2018-07-16 05:11:17 +00:00
|
|
|
imported_strategy = import_strategy(strategy, default_config)
|
2018-06-23 09:14:31 +00:00
|
|
|
|
2018-06-23 09:57:26 +00:00
|
|
|
assert dir(strategy) == dir(imported_strategy)
|
|
|
|
|
2018-06-23 09:14:31 +00:00
|
|
|
assert imported_strategy.__module__ == 'freqtrade.strategy'
|
|
|
|
assert imported_strategy.some_method() == 42
|
|
|
|
|
|
|
|
assert (
|
|
|
|
'freqtrade.strategy',
|
|
|
|
logging.DEBUG,
|
|
|
|
'Imported strategy freqtrade.strategy.default_strategy.DefaultStrategy '
|
|
|
|
'as freqtrade.strategy.DefaultStrategy',
|
|
|
|
) in caplog.record_tuples
|
|
|
|
|
|
|
|
|
2018-01-15 08:35:11 +00:00
|
|
|
def test_search_strategy():
|
2018-07-16 05:11:17 +00:00
|
|
|
default_config = {}
|
2018-03-24 21:16:42 +00:00
|
|
|
default_location = os.path.join(os.path.dirname(
|
|
|
|
os.path.realpath(__file__)), '..', '..', 'strategy'
|
|
|
|
)
|
|
|
|
assert isinstance(
|
2018-07-16 05:11:17 +00:00
|
|
|
StrategyResolver._search_strategy(
|
|
|
|
default_location,
|
|
|
|
config=default_config,
|
|
|
|
strategy_name='DefaultStrategy'
|
|
|
|
),
|
|
|
|
IStrategy
|
2018-03-24 21:16:42 +00:00
|
|
|
)
|
2018-07-16 05:11:17 +00:00
|
|
|
assert StrategyResolver._search_strategy(
|
|
|
|
default_location,
|
|
|
|
config=default_config,
|
|
|
|
strategy_name='NotFoundStrategy'
|
|
|
|
) is None
|
2018-01-15 08:35:11 +00:00
|
|
|
|
|
|
|
|
|
|
|
def test_load_strategy(result):
|
2018-06-23 09:14:31 +00:00
|
|
|
resolver = StrategyResolver({'strategy': 'TestStrategy'})
|
2018-03-24 22:20:21 +00:00
|
|
|
assert hasattr(resolver.strategy, 'populate_indicators')
|
|
|
|
assert 'adx' in resolver.strategy.populate_indicators(result)
|
2018-01-15 08:35:11 +00:00
|
|
|
|
|
|
|
|
2018-06-23 12:42:22 +00:00
|
|
|
def test_load_strategy_invalid_directory(result, caplog):
|
2018-03-25 14:28:04 +00:00
|
|
|
resolver = StrategyResolver()
|
|
|
|
extra_dir = os.path.join('some', 'path')
|
2018-06-23 12:42:22 +00:00
|
|
|
resolver._load_strategy('TestStrategy', extra_dir)
|
|
|
|
|
|
|
|
assert (
|
|
|
|
'freqtrade.strategy.resolver',
|
|
|
|
logging.WARNING,
|
|
|
|
'Path "{}" does not exist'.format(extra_dir),
|
|
|
|
) in caplog.record_tuples
|
2018-03-25 14:28:04 +00:00
|
|
|
|
|
|
|
assert hasattr(resolver.strategy, 'populate_indicators')
|
|
|
|
assert 'adx' in resolver.strategy.populate_indicators(result)
|
|
|
|
|
|
|
|
|
|
|
|
def test_load_not_found_strategy():
|
|
|
|
strategy = StrategyResolver()
|
|
|
|
with pytest.raises(ImportError,
|
|
|
|
match=r'Impossible to load Strategy \'NotFoundStrategy\'.'
|
|
|
|
r' This class does not exist or contains Python code errors'):
|
2018-07-16 05:11:17 +00:00
|
|
|
strategy._load_strategy(strategy_name='NotFoundStrategy', config={})
|
2018-02-02 09:01:09 +00:00
|
|
|
|
|
|
|
|
2018-01-15 08:35:11 +00:00
|
|
|
def test_strategy(result):
|
2018-03-24 22:20:21 +00:00
|
|
|
resolver = StrategyResolver({'strategy': 'DefaultStrategy'})
|
2018-01-15 08:35:11 +00:00
|
|
|
|
2018-03-24 22:20:21 +00:00
|
|
|
assert hasattr(resolver.strategy, 'minimal_roi')
|
|
|
|
assert resolver.strategy.minimal_roi[0] == 0.04
|
2018-01-15 08:35:11 +00:00
|
|
|
|
2018-03-24 22:20:21 +00:00
|
|
|
assert hasattr(resolver.strategy, 'stoploss')
|
|
|
|
assert resolver.strategy.stoploss == -0.10
|
2018-01-15 08:35:11 +00:00
|
|
|
|
2018-03-24 22:20:21 +00:00
|
|
|
assert hasattr(resolver.strategy, 'populate_indicators')
|
|
|
|
assert 'adx' in resolver.strategy.populate_indicators(result)
|
2018-01-15 08:35:11 +00:00
|
|
|
|
2018-03-24 22:20:21 +00:00
|
|
|
assert hasattr(resolver.strategy, 'populate_buy_trend')
|
|
|
|
dataframe = resolver.strategy.populate_buy_trend(resolver.strategy.populate_indicators(result))
|
2018-01-15 08:35:11 +00:00
|
|
|
assert 'buy' in dataframe.columns
|
|
|
|
|
2018-03-24 22:20:21 +00:00
|
|
|
assert hasattr(resolver.strategy, 'populate_sell_trend')
|
|
|
|
dataframe = resolver.strategy.populate_sell_trend(resolver.strategy.populate_indicators(result))
|
2018-01-15 08:35:11 +00:00
|
|
|
assert 'sell' in dataframe.columns
|
|
|
|
|
|
|
|
|
|
|
|
def test_strategy_override_minimal_roi(caplog):
|
2018-01-31 17:37:38 +00:00
|
|
|
caplog.set_level(logging.INFO)
|
2018-01-15 08:35:11 +00:00
|
|
|
config = {
|
2018-03-24 19:44:04 +00:00
|
|
|
'strategy': 'DefaultStrategy',
|
2018-01-15 08:35:11 +00:00
|
|
|
'minimal_roi': {
|
|
|
|
"0": 0.5
|
|
|
|
}
|
|
|
|
}
|
2018-03-24 22:20:21 +00:00
|
|
|
resolver = StrategyResolver(config)
|
2018-01-15 08:35:11 +00:00
|
|
|
|
2018-03-24 22:20:21 +00:00
|
|
|
assert hasattr(resolver.strategy, 'minimal_roi')
|
|
|
|
assert resolver.strategy.minimal_roi[0] == 0.5
|
2018-03-24 17:11:21 +00:00
|
|
|
assert ('freqtrade.strategy.resolver',
|
2018-01-15 08:35:11 +00:00
|
|
|
logging.INFO,
|
|
|
|
'Override strategy \'minimal_roi\' with value in config file.'
|
|
|
|
) in caplog.record_tuples
|
|
|
|
|
|
|
|
|
|
|
|
def test_strategy_override_stoploss(caplog):
|
2018-01-31 17:37:38 +00:00
|
|
|
caplog.set_level(logging.INFO)
|
2018-01-15 08:35:11 +00:00
|
|
|
config = {
|
2018-03-24 19:44:04 +00:00
|
|
|
'strategy': 'DefaultStrategy',
|
2018-01-15 08:35:11 +00:00
|
|
|
'stoploss': -0.5
|
|
|
|
}
|
2018-03-24 22:20:21 +00:00
|
|
|
resolver = StrategyResolver(config)
|
2018-01-15 08:35:11 +00:00
|
|
|
|
2018-03-24 22:20:21 +00:00
|
|
|
assert hasattr(resolver.strategy, 'stoploss')
|
|
|
|
assert resolver.strategy.stoploss == -0.5
|
2018-03-24 17:11:21 +00:00
|
|
|
assert ('freqtrade.strategy.resolver',
|
2018-01-15 08:35:11 +00:00
|
|
|
logging.INFO,
|
2018-01-20 22:40:41 +00:00
|
|
|
'Override strategy \'stoploss\' with value in config file: -0.5.'
|
|
|
|
) in caplog.record_tuples
|
|
|
|
|
|
|
|
|
|
|
|
def test_strategy_override_ticker_interval(caplog):
|
2018-01-31 17:37:38 +00:00
|
|
|
caplog.set_level(logging.INFO)
|
|
|
|
|
2018-01-20 22:40:41 +00:00
|
|
|
config = {
|
2018-03-24 19:44:04 +00:00
|
|
|
'strategy': 'DefaultStrategy',
|
2018-01-20 22:40:41 +00:00
|
|
|
'ticker_interval': 60
|
|
|
|
}
|
2018-03-24 22:20:21 +00:00
|
|
|
resolver = StrategyResolver(config)
|
2018-01-20 22:40:41 +00:00
|
|
|
|
2018-03-24 22:20:21 +00:00
|
|
|
assert hasattr(resolver.strategy, 'ticker_interval')
|
|
|
|
assert resolver.strategy.ticker_interval == 60
|
2018-03-24 17:11:21 +00:00
|
|
|
assert ('freqtrade.strategy.resolver',
|
2018-01-20 22:40:41 +00:00
|
|
|
logging.INFO,
|
|
|
|
'Override strategy \'ticker_interval\' with value in config file: 60.'
|
2018-01-15 08:35:11 +00:00
|
|
|
) in caplog.record_tuples
|