File size: 977 Bytes
bd73929 a68a14c |
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 31 32 |
import sys
import os
sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
from app.interface import predict
def test_predict_zero_shot():
result = predict("Tu es gentil.", model_type="zero-shot")
assert isinstance(result, str)
assert "Résultat de la classification" in result
def test_predict_few_shot():
result = predict("Tu es débile.", model_type="few-shot")
assert isinstance(result, str)
assert "Résultat de la classification" in result
def test_predict_empty_input():
try:
result = predict("", model_type="zero-shot")
except ValueError as e:
assert "at least one sequence" in str(e)
def test_predict_invalid_model():
try:
predict("Texte test", model_type="unknown")
except ValueError as e:
assert "Modèle inconnu" in str(e)
def test_create_interface():
from app.interface import create_interface
iface = create_interface()
assert iface is not None |