test_setup.py

#!/usr/bin/env python3
"""
Quick Test Script for QUENCH++
Verifies environment setup and dependencies.
"""

import sys
import importlib

def check_import(module_name, package_name=None):
    """Check if a module can be imported."""
    try:
        importlib.import_module(module_name)
        print(f"✓ {package_name or module_name}")
        return True
    except ImportError:
        print(f"✗ {package_name or module_name} - NOT INSTALLED")
        return False

def main():
    print("="*60)
    print("QUENCH++ Environment Test")
    print("="*60)
    
    required = [
        ('numpy', 'numpy'),
        ('pandas', 'pandas'),
        ('scipy', 'scipy'),
        ('matplotlib', 'matplotlib'),
        ('seaborn', 'seaborn'),
        ('torch', 'pytorch'),
        ('transformers', 'transformers'),
        ('yaml', 'pyyaml'),
        ('tqdm', 'tqdm'),
    ]
    
    optional = [
        ('openai', 'openai'),
        ('jupyter', 'jupyter'),
    ]
    
    print("\nRequired Dependencies:")
    required_ok = all(check_import(m, p) for m, p in required)
    
    print("\nOptional Dependencies:")
    for m, p in optional:
        check_import(m, p)
    
    print("\n" + "="*60)
    if required_ok:
        print("✅ Environment setup complete!")
        print("\nNext steps:")
        print("1. Configure API keys in config/config.yaml")
        print("2. Place QUENCH dataset in data/original/quench.json")
        print("3. Start with notebooks/eng-hin.ipynb")
    else:
        print("❌ Missing required dependencies")
        print("\nRun: pip install -r requirements.txt")
    print("="*60)
    
    return 0 if required_ok else 1

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