File size: 1,865 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 |
#!/usr/bin/env python3
# filepath: /Users/yagoairm2/Desktop/agents/final project/HF_Agents_Final_Project/tests/test_chess_analysis.py
"""
Test the non-hardcoded chess image analysis implementation
"""
import os
import sys
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"))
def main():
"""Test the chess image analysis with our new implementation"""
print("Testing chess image analysis with OpenCV and chess engine")
# Path to the test chess image
test_image = str(Path(__file__).parent.parent / "data/downloaded_files" / "cca530fc-4052-43b2-b130-b30968d8aa44.png")
if not os.path.exists(test_image):
print(f"Error: Test image not found at {test_image}")
return
print(f"Processing chess image: {test_image}")
# Import here to avoid dependency issues
from image_processing_tool import ImageProcessor
# Create image processor
processor = ImageProcessor()
# Process the image directly with our new implementation
result = processor.analyze_chess_position(test_image)
# Display the results
if isinstance(result, dict):
print("\nChess Position Analysis Results:")
for key, value in result.items():
print(f"{key}: {value}")
# Extract the move recommendation for the question answer
if "recommended_move" in result:
print("\nQuestion: Review the chess position provided in the image. It is black's turn. Provide the correct next move for black which guarantees a win. Please provide your response in algebraic notation.")
print(f"Answer: {result['recommended_move']}")
else:
print("\nUnexpected result format:")
print(result)
if __name__ == "__main__":
main()
|