validate_dependencies.py

#!/usr/bin/env python3
"""Validate that all required dependencies are installed and working correctly.

This script checks that all Python packages required for the Computational Physics
2016 scripts are properly installed and can be imported successfully.
"""

import sys
import importlib


def check_dependency(package_name, display_name=None):
    """Check if a package can be imported and return version info."""
    if display_name is None:
        display_name = package_name
    
    try:
        module = importlib.import_module(package_name)
        version = getattr(module, '__version__', 'unknown')
        print(f"✓ {display_name}: {version}")
        return True
    except ImportError as e:
        print(f"✗ {display_name}: Failed to import ({e})")
        return False


def main():
    """Validate all dependencies."""
    print("Validating Python dependencies for Computational Physics 2016...")
    print("=" * 60)
    
    # Core scientific computing packages
    dependencies = [
        ('numpy', 'NumPy'),
        ('matplotlib', 'Matplotlib'),
        ('scipy', 'SciPy'),
        ('sympy', 'SymPy'),
    ]
    
    # Additional packages that might be used
    optional_dependencies = [
        ('cycler', 'Cycler'),
        ('PIL', 'Pillow'),
        ('dateutil', 'python-dateutil'),
    ]
    
    success_count = 0
    total_count = len(dependencies)
    
    print("\nCore Dependencies:")
    print("-" * 30)
    for package, display in dependencies:
        if check_dependency(package, display):
            success_count += 1
    
    print("\nOptional Dependencies:")
    print("-" * 30)
    for package, display in optional_dependencies:
        check_dependency(package, display)
    
    print("\n" + "=" * 60)
    if success_count == total_count:
        print(f"✓ All {total_count} core dependencies are working correctly!")
        print("Your environment is ready for Computational Physics 2016.")
        
        # Test matplotlib backend
        try:
            import matplotlib.pyplot as plt
            plt.figure(figsize=(1, 1))
            plt.close()
            print("✓ Matplotlib backend is working correctly.")
        except Exception as e:
            print(f"⚠ Matplotlib backend warning: {e}")
            print("  This may affect plot display but scripts should still run.")
        
        return 0
    else:
        print(f"✗ {total_count - success_count} core dependencies failed!")
        print("Please install missing dependencies using:")
        print("  pip install -r stable_requirements.txt")
        print("  or")
        print("  conda env create -f environment.yml")
        return 1


if __name__ == "__main__":
    sys.exit(main())