test_quick.py

#!/usr/bin/env python3
"""Quick test of the bias detection functionality."""

import sys
import os
sys.path.append('app')

from app.utils import predict_bias_zero_shot

def test_basic_functionality():
    """Test basic bias prediction."""
    
    # Test different types of text
    test_cases = [
        "This progressive policy will help working families and reduce inequality.",
        "Conservative values and traditional family structures are important for society.",
        "The economic data shows mixed results with both positive and negative indicators."
    ]
    
    print("🧪 Testing News Bias Detector...\n")
    
    for i, text in enumerate(test_cases, 1):
        print(f"Test {i}: {text[:50]}...")
        try:
            result = predict_bias_zero_shot(text)
            print(f"  → Prediction: {result['label']} (confidence: {result['prob']:.2f})")
            print(f"  → Bias score: {result['score']:.2f}")
            print()
        except Exception as e:
            print(f"  → Error: {e}")
            print()
    
    print("✅ Basic functionality test completed!")

if __name__ == "__main__":
    test_basic_functionality()