def test(): return _predict._test()
def test_test(mocker, output, expected): mocker.patch("src.app.ml.active_predictor.active_predictor.predict", return_value=output) result = _test(data=MockData()) assert nested_floats_almost_equal(result["prediction"], expected["prediction"])