Testing the API endpoint

test_predict.py

import pytest
import random

from starlette.testclient import TestClient
from starlette.status import HTTP_200_OK

from api.ml.model import n_features


@pytest.mark.parametrize("n_instances", range(1, 10))
def test_predict(n_instances: int, test_client: TestClient):
    fake_data = [[random.random() for _ in range(n_features)] for _ in range(n_instances)]
    response = test_client.post("/predict", json={"data": fake_data})
    assert response.status_code == HTTP_200_OK
    assert len(response.json()["data"]) == n_instances