"""
Pipeline Tests

Description: Pipeline tests ensures that the functionalities of the feature processing pipelines
work as expected to avoid feeding models incorrect data.


"""

from sklearn.pipeline import Pipeline
import pandas as pd
import numpy as np
import os
from absl import logging

from forecasting_toolkit.models.pipelines.train.pre_release import (
    gbt_regressor_baseline_v2,
    histgbm_regressor_baseline_v1
)



def test_gbt_regressor_pipeline_v2():
    """ test gbt model by loading it to make sure all is well"""
    logging.set_verbosity(logging.DEBUG)
    model = gbt_regressor_baseline_v2()
    assert isinstance(model, Pipeline)