def test_rsi_confirmation_predict(dollar_bar_dataframe):
    rsi_threshold = 45
    rsi_timeperiod = 25
    rsi_confirmation_period = 2
    rsi_moving_average_window = 2
    bar_arrival_rsi_strategy = RSIConfirmation(rsi_timeperiod, rsi_threshold, rsi_confirmation_period, rsi_moving_average_window)
    preds = bar_arrival_rsi_strategy.predict(dollar_bar_dataframe)
    assert set(preds).issubset(bar_arrival_rsi_strategy.classes_)
def test_rsi_confirmation_predict(dollar_bar_dataframe):
    rsi_threshold = 45
    rsi_timeperiod = 25
    rsi_confirmation_period = 2
    rsi_moving_average_window = 2
    bar_arrival_rsi_strategy = RSIConfirmation(rsi_timeperiod, rsi_threshold, rsi_confirmation_period, rsi_moving_average_window)
    preds = bar_arrival_rsi_strategy.predict(dollar_bar_dataframe)
    assert set(preds).issubset(bar_arrival_rsi_strategy.classes_)