from datetime import datetime from pandas import DataFrame from freqtrade.persistence.models import Trade from .strats.strategy_test_v2 import StrategyTestV2 def test_strategy_test_v2_structure(): assert hasattr(StrategyTestV2, 'minimal_roi') assert hasattr(StrategyTestV2, 'stoploss') assert hasattr(StrategyTestV2, 'timeframe') assert hasattr(StrategyTestV2, 'populate_indicators') assert hasattr(StrategyTestV2, 'populate_buy_trend') assert hasattr(StrategyTestV2, 'populate_sell_trend') def test_strategy_test_v2(result, fee): strategy = StrategyTestV2({}) metadata = {'pair': 'ETH/BTC'} assert type(strategy.minimal_roi) is dict assert type(strategy.stoploss) is float assert type(strategy.timeframe) is str indicators = strategy.populate_indicators(result, metadata) 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 trade = Trade( open_rate=19_000, amount=0.1, pair='ETH/BTC', fee_open=fee.return_value ) assert strategy.confirm_trade_entry(pair='ETH/BTC', order_type='limit', amount=0.1, rate=20000, time_in_force='gtc', current_time=datetime.utcnow()) is True 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', current_time=datetime.utcnow()) is True # TODO-lev: Test for shorts? assert strategy.custom_stoploss(pair='ETH/BTC', trade=trade, current_time=datetime.now(), current_rate=20_000, current_profit=0.05) == strategy.stoploss