File size: 840 Bytes
4cea933
 
 
 
 
2a9d346
4cea933
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
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