tests/test_models.py

import os
os.environ["MOCK_INFERENCE"] = "true"

from fastapi.testclient import TestClient
from app import app

client = TestClient(app)

def test_models_list():
    response = client.get("/v1/models")
    assert response.status_code == 200
    json_data = response.json()
    assert json_data["object"] == "list"
    assert len(json_data["data"]) == 3
    
    model_ids = [m["id"] for m in json_data["data"]]
    assert "plain-llama-3-8b" in model_ids
    assert "plain-llama-3.2-3b" in model_ids
    assert "plain-qwen-0.5b" in model_ids
    
    for item in json_data["data"]:
        assert "paired_rif_model" in item
        assert item["object"] == "model"
        assert item["owned_by"] == "vijnana-ai"
        assert item["engine"] == "plain-transformer"