test_installation.py

#!/usr/bin/env python3
"""
Test script to verify installation and basic functionality
"""

import sys
import os

# Add src to path
sys.path.insert(0, os.path.join(os.path.dirname(__file__), 'src'))

def test_imports():
    """Test that all modules can be imported"""
    print("Testing imports...")
    
    try:
        from core import (
            MetadataExtractor,
            SteganographyDetector,
            ManipulationDetector,
            ImageAnalyzer,
            HashGenerator
        )
        print("✓ Core modules imported successfully")
    except ImportError as e:
        print(f"✗ Failed to import core modules: {e}")
        return False
    
    try:
        from reports import HTMLReportGenerator, JSONExporter
        print("✓ Report modules imported successfully")
    except ImportError as e:
        print(f"✗ Failed to import report modules: {e}")
        return False
    
    try:
        from utils import FileHandler, ImageValidator, GeoParser
        print("✓ Utility modules imported successfully")
    except ImportError as e:
        print(f"✗ Failed to import utility modules: {e}")
        return False
    
    try:
        from forensics import (
            ErrorLevelAnalysis,
            NoiseAnalyzer,
            CloneDetector,
            StatisticalAnalyzer
        )
        print("✓ Forensics modules imported successfully")
    except ImportError as e:
        print(f"✗ Failed to import forensics modules: {e}")
        return False
    
    try:
        from ml import MLStegoDetector
        print("✓ ML modules imported successfully")
    except ImportError as e:
        print(f"✗ Failed to import ML modules: {e}")
        return False
    
    return True

def test_basic_functionality():
    """Test basic functionality"""
    print("\nTesting basic functionality...")
    
    try:
        from core import MetadataExtractor, HashGenerator
        from utils import ImageValidator
        
        # Test validator
        validator = ImageValidator()
        print("✓ ImageValidator initialized")
        
        # Test metadata extractor
        extractor = MetadataExtractor()
        print("✓ MetadataExtractor initialized")
        
        # Test hash generator
        hasher = HashGenerator()
        print("✓ HashGenerator initialized")
        
        return True
        
    except Exception as e:
        print(f"✗ Error during functionality test: {e}")
        return False

def check_dependencies():
    """Check if required dependencies are available"""
    print("\nChecking dependencies...")
    
    required = [
        'PIL',
        'cv2',
        'numpy',
        'click',
        'plotly',
        'exifread',
        'imagehash'
    ]
    
    missing = []
    
    for module in required:
        try:
            __import__(module)
            print(f"✓ {module} is installed")
        except ImportError:
            print(f"✗ {module} is NOT installed")
            missing.append(module)
    
    if missing:
        print(f"\n⚠ Missing dependencies: {', '.join(missing)}")
        print("Please run: pip install -r requirements.txt")
        return False
    
    return True

def main():
    """Run all tests"""
    print("=" * 60)
    print("PixIntelligence - Installation Test")
    print("=" * 60)
    
    all_passed = True
    
    # Check dependencies
    if not check_dependencies():
        all_passed = False
        print("\n⚠ Some dependencies are missing. Install them first.")
        return
    
    # Test imports
    if not test_imports():
        all_passed = False
    
    # Test functionality
    if not test_basic_functionality():
        all_passed = False
    
    print("\n" + "=" * 60)
    if all_passed:
        print("✓ All tests passed! The suite is ready to use.")
        print("\nTry running:")
        print("  python src/main.py --help")
        print("  python src/main.py info")
    else:
        print("✗ Some tests failed. Please check the errors above.")
    print("=" * 60)

if __name__ == "__main__":
    main()