freqtrade_origin/tests/strategy/test_default_strategy.py

55 lines
2.1 KiB
Python
Raw Normal View History

2020-12-19 16:59:49 +00:00
from datetime import datetime
2020-12-20 10:17:50 +00:00
2021-11-14 01:50:53 +00:00
import pytest
2018-01-15 08:35:11 +00:00
from pandas import DataFrame
2018-03-17 21:44:47 +00:00
2020-12-20 10:17:50 +00:00
from freqtrade.persistence.models import Trade
from .strats.strategy_test_v3 import StrategyTestV3
2018-01-15 08:35:11 +00:00
2021-11-14 08:02:25 +00:00
def test_strategy_test_v3_structure():
assert hasattr(StrategyTestV3, 'minimal_roi')
assert hasattr(StrategyTestV3, 'stoploss')
assert hasattr(StrategyTestV3, 'timeframe')
assert hasattr(StrategyTestV3, 'populate_indicators')
assert hasattr(StrategyTestV3, 'populate_buy_trend')
assert hasattr(StrategyTestV3, 'populate_sell_trend')
2018-01-15 08:35:11 +00:00
2018-01-18 05:50:12 +00:00
2021-11-14 01:50:53 +00:00
@pytest.mark.parametrize('is_short,side', [
(True, 'short'),
(False, 'long'),
])
2021-11-14 08:02:25 +00:00
def test_strategy_test_v3(result, fee, is_short, side):
strategy = StrategyTestV3({})
2018-01-15 08:35:11 +00:00
metadata = {'pair': 'ETH/BTC'}
2018-01-15 08:35:11 +00:00
assert type(strategy.minimal_roi) is dict
assert type(strategy.stoploss) is float
2020-06-02 07:36:04 +00:00
assert type(strategy.timeframe) is str
indicators = strategy.populate_indicators(result, metadata)
2018-01-15 08:35:11 +00:00
assert type(indicators) is DataFrame
assert type(strategy.populate_buy_trend(indicators, metadata)) is DataFrame
assert type(strategy.populate_sell_trend(indicators, metadata)) is DataFrame
2020-12-19 16:59:49 +00:00
trade = Trade(
open_rate=19_000,
amount=0.1,
pair='ETH/BTC',
2021-11-14 01:50:53 +00:00
fee_open=fee.return_value,
is_short=is_short
2020-12-19 16:59:49 +00:00
)
assert strategy.confirm_trade_entry(pair='ETH/BTC', order_type='limit', amount=0.1,
rate=20000, time_in_force='gtc',
2021-11-14 01:50:53 +00:00
current_time=datetime.utcnow(),
side=side) is True
2020-12-19 16:59:49 +00:00
assert strategy.confirm_trade_exit(pair='ETH/BTC', trade=trade, order_type='limit', amount=0.1,
rate=20000, time_in_force='gtc', sell_reason='roi',
2021-11-14 01:50:53 +00:00
current_time=datetime.utcnow(),
side=side) is True
2020-12-19 16:59:49 +00:00
2020-12-20 10:17:50 +00:00
assert strategy.custom_stoploss(pair='ETH/BTC', trade=trade, current_time=datetime.now(),
current_rate=20_000, current_profit=0.05) == strategy.stoploss