test_subsample.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Quick test to verify the subsampling script works without errors.
"""

import sys
import os
sys.path.insert(0, '.')

from subsample_dataset import get_runtime_info, setup_logging

def test_runtime_info():
    """Test that get_runtime_info() works without errors."""
    print("Testing get_runtime_info()...")
    try:
        info = get_runtime_info()
        print("✓ get_runtime_info() completed successfully")
        print(f"  Platform: {info['platform']}")
        print(f"  Python version: {info['python_version']}")
        print(f"  Hostname: {info['hostname']}")
        print(f"  CPU count: {info['cpu_count']}")
        print(f"  Memory info keys: {len(info['memory_info'])} items")
        print(f"  Environment variables: {info['environment_variables']}")
        return True
    except Exception as e:
        print(f"✗ get_runtime_info() failed: {e}")
        return False

def test_logging():
    """Test that logging setup works."""
    print("\nTesting logging setup...")
    try:
        import tempfile
        with tempfile.NamedTemporaryFile(suffix='.log', delete=False) as f:
            log_file = f.name
        
        logger = setup_logging(log_file, "INFO")
        logger.info("Test log message")
        print(f"✓ Logging setup successful, log file: {log_file}")
        
        # Clean up
        os.unlink(log_file)
        return True
    except Exception as e:
        print(f"✗ Logging setup failed: {e}")
        return False

def test_imports():
    """Test that all required imports work."""
    print("\nTesting imports...")
    try:
        import anndata as ad
        import numpy as np
        import pandas as pd
        import scanpy as sc
        import scipy.sparse as sp
        from tqdm.auto import tqdm
        print("✓ All required packages imported successfully")
        print(f"  anndata: {ad.__version__}")
        print(f"  numpy: {np.__version__}")
        print(f"  pandas: {pd.__version__}")
        print(f"  scanpy: {sc.__version__}")
        return True
    except ImportError as e:
        print(f"✗ Import failed: {e}")
        return False

if __name__ == "__main__":
    print("=== Subsampling Infrastructure Test ===")
    
    tests = [
        test_imports,
        test_runtime_info,
        test_logging
    ]
    
    results = []
    for test in tests:
        results.append(test())
    
    print(f"\n=== Test Summary ===")
    passed = sum(results)
    total = len(results)
    print(f"Passed: {passed}/{total}")
    
    if passed == total:
        print("✓ All tests passed! The subsampling script should work correctly.")
        sys.exit(0)
    else:
        print("✗ Some tests failed. Please check the errors above.")
        sys.exit(1)