test_requirements.py

#!/usr/bin/env python3
"""
Test script for AI Chess Scanner
Verifies all components work without requiring GUI display
"""

import sys
import os

# Test 1: Check Python version
print("=" * 60)
print("TEST 1: Python Version")
print("=" * 60)
print(f"Python: {sys.version}")
required_version = (3, 8)
if sys.version_info < required_version:
    print(f"❌ FAILED: Python 3.8+ required")
    sys.exit(1)
print("✅ PASSED\n")

# Test 2: Check dependencies
print("=" * 60)
print("TEST 2: Required Dependencies")
print("=" * 60)

dependencies = {
    'PIL': 'Pillow',
    'cv2': 'opencv-python',
    'chess': 'python-chess',
    'chess.engine': 'python-chess'
}

missing = []
for module, package in dependencies.items():
    try:
        __import__(module)
        print(f"✅ {package:20} - OK")
    except ImportError:
        print(f"❌ {package:20} - MISSING")
        missing.append(package)

if missing:
    print(f"\n❌ FAILED: Missing packages: {', '.join(missing)}")
    sys.exit(1)
print("✅ PASSED\n")

# Test 3: Check Stockfish
print("=" * 60)
print("TEST 3: Stockfish Engine")
print("=" * 60)

import chess
import chess.engine

stockfish_paths = [
    "/opt/homebrew/bin/stockfish",  # macOS Apple Silicon
    "/usr/local/bin/stockfish",     # macOS Intel
    "/usr/games/stockfish",         # Linux
    "/usr/bin/stockfish",           # Linux
]

found_stockfish = None
for path in stockfish_paths:
    if os.path.exists(path):
        found_stockfish = path
        print(f"✅ Found Stockfish at: {path}")
        break

if not found_stockfish:
    print(f"❌ FAILED: Stockfish not found")
    print("Install with: brew install stockfish (macOS) or sudo apt install stockfish (Linux)")
    sys.exit(1)

# Test engine initialization
try:
    with chess.engine.SimpleEngine.popen_uci(found_stockfish) as engine:
        print(f"✅ Stockfish engine initialized successfully")
        print(f"✅ PASSED\n")
except Exception as e:
    print(f"❌ FAILED: {str(e)}\n")
    sys.exit(1)

# Test 4: FEN Validation
print("=" * 60)
print("TEST 4: Chess FEN Validation")
print("=" * 60)

test_fens = [
    ("Starting Position", "rnbqkbnr/pppppppp/8/8/8/8/PPPPPPPP/RNBQKBNR w KQkq - 0 1", True),
    ("After 1.e4", "rnbqkbnr/pppppppp/8/8/4P3/8/PPPP1PPP/RNBQKBNR b KQkq e3 0 1", True),
    ("Invalid FEN", "invalid/fen/string", False),
]

for name, fen, should_pass in test_fens:
    try:
        chess.Board(fen)
        result = "✅" if should_pass else "❌"
        print(f"{result} {name}: {fen}")
    except ValueError as e:
        result = "✅" if not should_pass else "❌"
        print(f"{result} {name}: Invalid")

print("✅ PASSED\n")

# Test 5: Basic Position Analysis
print("=" * 60)
print("TEST 5: Basic Position Analysis")
print("=" * 60)

board = chess.Board("rnbqkbnr/pppppppp/8/8/4P3/8/PPPP1PPP/RNBQKBNR b KQkq e3 0 1")
print(f"FEN: {board.fen()}")
print(f"Turn: {'White' if board.turn else 'Black'}")
print(f"Legal Moves Available: {board.legal_moves.count()}")
print(f"In Check: {board.is_check()}")

try:
    with chess.engine.SimpleEngine.popen_uci(found_stockfish) as engine:
        info = engine.analyse(board, chess.engine.Limit(time=1.0))
        
        if "pv" in info and info["pv"]:
            best_move = info["pv"][0]
            move_san = board.san(best_move)
            print(f"\n🎯 Best Move: {move_san}")
            
            if "score" in info:
                score = info["score"]
                if score.is_mate():
                    mate_in = score.mate()
                    print(f"⚠️  Checkmate in {abs(mate_in)} moves")
                else:
                    # Extract the centipawn score
                    try:
                        # Try using relative() method first
                        cp_val = score.relative.cp
                    except (AttributeError, TypeError):
                        try:
                            # For white's perspective
                            cp_val = score.white().cp
                        except (AttributeError, TypeError):
                            # Fallback: convert to string and parse
                            cp_str = str(score)
                            try:
                                cp_val = int(float(cp_str) * 100)
                            except:
                                cp_val = 0
                    
                    eval_score = cp_val / 100.0
                    print(f"📊 Evaluation: {eval_score:+.2f} pawns")
            
            print("✅ PASSED\n")
        else:
            print("❌ FAILED: No best move found\n")
except Exception as e:
    print(f"❌ FAILED: {str(e)}\n")
    sys.exit(1)

# Final Summary
print("=" * 60)
print("ALL TESTS PASSED! ✅")
print("=" * 60)
print("\n✨ Your Chess Scanner is ready to run!")
print("   Launch with: python script.py")
print("\nNote: The GUI requires a display environment.")
print("If running on a server, use ssh -X for X11 forwarding.")