tinyml-backend / tests /test_api.py
IDKHowToCodeFr's picture
Upload folder using huggingface_hub
2a9d346 verified
Raw History Blame
840 Bytes
import pytest
from fastapi.testclient import TestClient
import sys
import os
sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '../backend')))
from main import app
client = TestClient(app)
def test_history_endpoint():
response = client.get("/history")
assert response.status_code == 200
assert isinstance(response.json(), list)
def test_dataset_endpoint():
response = client.get("/dataset")
assert response.status_code == 200
assert isinstance(response.json(), list)
def test_export_tinyml_random_forest():
response = client.get("/export_tinyml?model_name=rf")
if response.status_code == 200:
data = response.json()
if "error" in data:
assert "not found" in data["error"].lower()
else:
assert "code" in data or "c_code" in data