import pytest

from tcalc.dataset import TCalcDataset
from tcalc.model import TCalcPredictor
from tcalc.config import TCalcDatasetConfig


@pytest.fixture(scope="session")
def dataset():
    return TCalcDataset(TCalcDatasetConfig())


@pytest.fixture(scope="session")
def tcalc_predictor():
    p = TCalcPredictor(TCalcDatasetConfig())
    p.load_models("default_model")
    return p