tests/test_train.py

from src.train import train


def test_smoke_train_runs_fast():
    r = train(smoke=True)
    assert 0.5 <= r <= 1.0