employee-future-prediction/application/tests/test_create_service.py

import requests


def test_service():
    prediction = requests.post(
        "http://127.0.0.1:3000/predict",
        headers={"content-type": "application/json"},
        data='{"City": "Pune", "PaymentTier": 0, "Age": 0, "Gender": "Female", "EverBenched": "No", "ExperienceInCurrentDomain": 0}',
    ).text

    assert prediction[0] in ["0", "1"]