test_gradio.py

"""
Script de test pour l'interface Gradio
"""

from mcp_server import analyze_herbicide_trends, get_available_plots

def test_function_calls():
    """Test des fonctions individuelles"""
    print("🧪 Test des fonctions...")
    
    # Test get_available_plots
    print("\n📋 Test get_available_plots():")
    plots = get_available_plots()
    print(f"Parcelles disponibles: {len(plots)} - {plots[:5]}...")
    
    # Test analyze_herbicide_trends avec différents paramètres
    print("\n📈 Test analyze_herbicide_trends():")
    
    # Test 1: Période normale
    print("Test 1: Période 2020-2024, toutes parcelles")
    fig, summary = analyze_herbicide_trends([2020, 2024], "Toutes")
    if fig is not None:
        print("✅ Graphique généré avec succès")
        print("📊 Résumé:", summary[:200] + "...")
    else:
        print("❌ Erreur:", summary)
    
    # Test 2: Parcelle spécifique
    print("\nTest 2: Période 2020-2024, parcelle spécifique")
    fig, summary = analyze_herbicide_trends([2020, 2024], "Champ ferme W du sol")
    if fig is not None:
        print("✅ Graphique généré avec succès")
    else:
        print("❌ Erreur:", summary)
    
    # Test 3: Format année simple
    print("\nTest 3: Année simple")
    fig, summary = analyze_herbicide_trends(2023, "Toutes")
    if fig is not None:
        print("✅ Graphique généré avec succès")
    else:
        print("❌ Erreur:", summary)

if __name__ == "__main__":
    test_function_calls()