test_subsample_simple.py

#!/usr/bin/env python3
"""
Quick test to verify the subsampling script works without errors.
"""

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

def test_runtime_info():
    """Test that get_runtime_info() works without errors."""
    print("Testing get_runtime_info()...")
    try:
        from subsample_dataset import get_runtime_info
        info = get_runtime_info()
        print("SUCCESS: 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"FAILED: get_runtime_info() failed: {e}")
        return False

def test_imports():
    """Test that all required packages are available."""
    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("SUCCESS: 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"FAILED: Import failed: {e}")
        return False

if __name__ == "__main__":
    print("=== Subsampling Infrastructure Test ===")
    
    # Test imports first
    imports_ok = test_imports()
    if not imports_ok:
        print("Cannot proceed without required packages")
        sys.exit(1)
    
    # Test runtime info
    runtime_ok = test_runtime_info()
    
    print(f"\n=== Test Summary ===")
    if runtime_ok:
        print("SUCCESS: All tests passed! The subsampling script should work correctly.")
        sys.exit(0)
    else:
        print("FAILED: Runtime info test failed. Please check the errors above.")
        sys.exit(1)