run_tests.py

# __define-ocg__ Test Runner Script
"""
Simple test runner for the ML Challenge project.
"""

import subprocess
import sys
import os
from pathlib import Path

def run_tests():
    """Run the test suite"""
    print("🧪 Running ML Challenge Test Suite")
    print("=" * 50)
    
    # Check if pytest is available
    try:
        import pytest
    except ImportError:
        print("❌ pytest not found. Installing...")
        subprocess.run([sys.executable, "-m", "pip", "install", "pytest", "pytest-cov"])
    
    # Run tests
    print("📁 Running tests from tests/ directory...")
    
    # Change to project root if needed
    project_root = Path(__file__).parent
    os.chdir(project_root)
    
    # Run pytest with coverage
    result = subprocess.run([
        sys.executable, "-m", "pytest", 
        "tests/", 
        "-v", 
        "--cov=src",
        "--cov-report=term-missing",
        "--cov-report=html:htmlcov"
    ])
    
    if result.returncode == 0:
        print("\n✅ All tests passed!")
        print("📊 Coverage report generated in htmlcov/ directory")
    else:
        print("\n❌ Some tests failed!")
        sys.exit(1)

if __name__ == "__main__":
    run_tests()