import pytest def test_model_prediction(): pass def test_model_training(): pass