mlops-architecture/tests/test_train.py

"""Tests for training script."""
from src.train import train


def test_train_returns_model_info():
    result = train()
    assert result["model"] == "test-model"
    assert "accuracy" in result
    assert result["accuracy"] > 0