import pytest

from tcalc.dataset import TCalcDataset
from tcalc.model import TCalcPredictor
from tcalc.utils import read_yaml


@pytest.fixture(scope="session")
def dataset():
    return TCalcDataset(read_yaml("config/dataset.yaml"))


@pytest.fixture(scope="session")
def tcalc_predictor():
    p = TCalcPredictor(read_yaml("config/dataset.yaml"))
    p.load_models("default_model")
    return p