Function test_hazard_classification

test_python_pipeline.py:83–121

Test hazard classification functionality

()

    return True

def test_hazard_classification():
    """Test hazard classification functionality"""
    print("\n🔄 Testing hazard classification...")
    
    from processors.classifier import HazardClassifier, classify_post_enhanced
    
    test_cases = [
        {
            "text": "Tsunami warning issued for coastal areas. Evacuate immediately!",
            "expected_hazard": "Tsunami",
            "expected_urgency": "high"
        },
        {
            "text": "Heavy rainfall causing flooding in the city",
            "expected_hazard": "Flooding",
            "expected_urgency": "medium"
        },
        {
            "text": "Beautiful sunset at the beach today",
            "expected_hazard": "General",
            "expected_urgency": "low"
        }
    ]
    
    classifier = HazardClassifier()
    
    for i, case in enumerate(test_cases):
        hazard_type, urgency, confidence = classify_post_enhanced(case["text"])
        print(f"  Test {i+1}: '{case['text'][:40]}...'")
        print(f"    Hazard: {hazard_type} (expected: {case['expected_hazard']})")
        print(f"    Urgency: {urgency} (expected: {case['expected_urgency']})")
        print(f"    Confidence: {confidence:.2f}")
        
        if hazard_type == case["expected_hazard"]:
            print(f"    ✅ Hazard classification correct")
        else:
            print(f"    ⚠️  Hazard classification different than expected")
    
    return True

def test_sentiment_analysis():
    """Test sentiment analysis functionality"""