Create test_ensemble.py

test_ensemble.py

@@ -0,0 +1,78 @@
+#!/usr/bin/env python3
+"""
+Basic testing script for the Enhanced Ensemble Model
+"""
+import unittest
+from PIL import Image
+import numpy as np
+from app import EnhancedEnsembleMemeAnalyzer
+
+class TestEnhancedEnsemble(unittest.TestCase):
+    
+    @classmethod
+    def setUpClass(cls):
+        """Initialize the analyzer once for all tests"""
+        cls.analyzer = EnhancedEnsembleMemeAnalyzer()
+        
+        # Create a simple test image
+        cls.test_image = Image.fromarray(np.random.randint(0, 255, (224, 224, 3), dtype=np.uint8))
+    
+    def test_sentiment_analysis(self):
+        """Test sentiment analysis functionality"""
+        
+        # Test positive sentiment
+        positive_result = self.analyzer.analyze_sentiment("I love this content! It's amazing!")
+        self.assertIn(positive_result["label"], ["POSITIVE", "NEUTRAL"])
+        self.assertGreater(positive_result["score"], 0)
+        
+        # Test negative sentiment
+        negative_result = self.analyzer.analyze_sentiment("This is terrible and offensive content")
+        self.assertIn(negative_result["label"], ["NEGATIVE", "NEUTRAL"])
+        self.assertGreater(negative_result["score"], 0)
+    
+    def test_ocr_extraction(self):
+        """Test OCR text extraction"""
+        result = self.analyzer.extract_text_from_image(self.test_image)
+        self.assertIsInstance(result, str)
+    
+    def test_multimodal_classification(self):
+        """Test multimodal content classification"""
+        result = self.analyzer.classify_multimodal_content(self.test_image, "test text")
+        
+        self.assertIn("is_hateful", result)
+        self.assertIn("hate_probability", result)
+        self.assertIn("confidence", result)
+        self.assertIsInstance(result["is_hateful"], bool)
+        self.assertGreaterEqual(result["hate_probability"], 0)
+        self.assertLessEqual(result["hate_probability"], 1)
+    
+    def test_ensemble_prediction(self):
+        """Test ensemble prediction functionality"""
+        
+        # Mock sentiment result
+        sentiment_result = {
+            "label": "NEGATIVE",
+            "score": 0.85,
+            "probabilities": [0.85, 0.10, 0.05]
+        }
+        
+        # Mock multimodal result
+        multimodal_result = {
+            "is_hateful": True,
+            "hate_probability": 0.75,
+            "safe_probability": 0.25,
+            "confidence": 0.80,
+            "detailed_scores": []
+        }
+        
+        ensemble_result = self.analyzer.ensemble_prediction(
+            sentiment_result, multimodal_result, "test text"
+        )
+        
+        self.assertIn("risk_level", ensemble_result)
+        self.assertIn("risk_score", ensemble_result)
+        self.assertIn("confidence", ensemble_result)
+        self.assertIn(ensemble_result["risk_level"], ["HIGH", "MEDIUM", "LOW", "SAFE"])
+
+if __name__ == "__main__":
+    unittest.main()