17 lines
383 B
Python
17 lines
383 B
Python
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 |