File size: 3,222 Bytes
8ff7d8f |
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 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 |
import os
import sys
import unittest
from pathlib import Path
# Add the src directory to the path so we can import the modules
sys.path.insert(0, str(Path(__file__).parent.parent / "src"))
from image_processing_tool import ImageProcessor
from file_processing_tool import FileIdentifier, process_image_file
class TestImageProcessingTool(unittest.TestCase):
def setUp(self):
self.image_processor = ImageProcessor()
self.file_identifier = FileIdentifier()
self.test_image_path = str(Path(__file__).parent.parent / "data/downloaded_files" / "cca530fc-4052-43b2-b130-b30968d8aa44.png")
# Make sure the test image exists
self.assertTrue(os.path.exists(self.test_image_path), f"Test image not found: {self.test_image_path}")
def test_file_identification(self):
"""Test that the FileIdentifier correctly identifies the PNG image."""
file_info = self.file_identifier.identify_file(self.test_image_path)
self.assertEqual(file_info.get('determined_type'), "image", "File should be identified as an image")
self.assertEqual(file_info.get('suggested_action'), "image_processor", "Action should be image_processor")
def test_image_details(self):
"""Test getting basic image details."""
details = self.image_processor.get_image_details(self.test_image_path)
self.assertIsNotNone(details, "Should return image details")
self.assertIn("width", details, "Should include width in details")
self.assertIn("height", details, "Should include height in details")
self.assertIn("format", details, "Should include format in details")
def test_image_processing(self):
"""Test basic image processing functionality."""
result = self.image_processor.process_image(self.test_image_path)
self.assertIsNotNone(result, "Should return processing result")
self.assertIsInstance(result, str, "Result should be a string")
def test_text_extraction(self):
"""Test OCR text extraction functionality."""
text = self.image_processor.extract_text_from_image(self.test_image_path)
self.assertIsNotNone(text, "Should return extracted text")
self.assertIsInstance(text, str, "Extracted text should be a string")
def test_chess_analysis(self):
"""Test chess position analysis."""
analysis = self.image_processor.analyze_chess_position(self.test_image_path)
self.assertIsNotNone(analysis, "Should return chess analysis")
if isinstance(analysis, dict):
self.assertIn("recommended_move", analysis, "Should include recommended move")
self.assertEqual(analysis["recommended_move"], "Rd5", "Recommended move should be 'Rd5'")
def test_process_image_file_function(self):
"""Test the integrated process_image_file function."""
result = process_image_file(self.test_image_path)
self.assertIsNotNone(result, "Should return processing result")
self.assertIn("details", result, "Should include image details")
self.assertIn("extracted_text", result, "Should include extracted text")
if __name__ == '__main__':
unittest.main()
|