def test_tcn_forecaster_shape_error(self):
    (train_data, val_data, test_data) = create_data()
    forecaster = LSTMForecaster(past_seq_len=24, input_feature_num=2, output_feature_num=1, loss='mae', lr=0.01)
    with pytest.raises(AssertionError):
        forecaster.fit(train_data, epochs=2)
def test_tcn_forecaster_shape_error(self):
    (train_data, val_data, test_data) = create_data()
    forecaster = LSTMForecaster(past_seq_len=24, input_feature_num=2, output_feature_num=1, loss='mae', lr=0.01)
    with pytest.raises(AssertionError):
        forecaster.fit(train_data, epochs=2)