| |
| """Test script to verify HunyuanWorldMirror inference-only module is properly set up. |
| |
| This will check if all necessary files and imports are in place. |
| """ |
|
|
| import sys |
| import os |
|
|
|
|
| def test_imports(): |
| """Test if core modules can be imported.""" |
| print("Testing HunyuanWorldMirror inference module setup...") |
| print("-" * 60) |
|
|
| |
| try: |
| from HunyuanWorldMirror import WorldMirror |
|
|
| print("✓ WorldMirror class imported successfully") |
| return True |
| except ImportError as e: |
| print(f"✗ Failed to import WorldMirror: {e}") |
| print("\nMissing dependencies detected. Please install:") |
| print(" pip install torch einops gsplat huggingface_hub") |
| return False |
|
|
|
|
| def test_structure(): |
| """Test if file structure is correct.""" |
| print("\nChecking file structure...") |
| print("-" * 60) |
|
|
| base_path = os.path.dirname(os.path.abspath(__file__)) |
|
|
| required_files = [ |
| "src/models/models/worldmirror.py", |
| "src/models/models/visual_transformer.py", |
| "src/models/models/rasterization.py", |
| "src/models/heads/camera_head.py", |
| "src/models/heads/dense_head.py", |
| "src/utils/inference_utils.py", |
| "infer.py", |
| ] |
|
|
| all_exist = True |
| for filepath in required_files: |
| full_path = os.path.join(base_path, filepath) |
| if os.path.exists(full_path): |
| print(f"✓ {filepath}") |
| else: |
| print(f"✗ {filepath} - MISSING") |
| all_exist = False |
|
|
| return all_exist |
|
|
|
|
| def test_removed_files(): |
| """Test if training/demo files were removed.""" |
| print("\nChecking removed files...") |
| print("-" * 60) |
|
|
| base_path = os.path.dirname(os.path.abspath(__file__)) |
|
|
| should_not_exist = [ |
| "training", |
| "demo", |
| "examples", |
| "app.py", |
| "submodules", |
| ] |
|
|
| all_removed = True |
| for item in should_not_exist: |
| full_path = os.path.join(base_path, item) |
| if not os.path.exists(full_path): |
| print(f"✓ {item} removed") |
| else: |
| print(f"✗ {item} still exists") |
| all_removed = False |
|
|
| return all_removed |
|
|
|
|
| if __name__ == "__main__": |
| print("=" * 60) |
| print("HunyuanWorldMirror Inference Module Test") |
| print("=" * 60) |
|
|
| structure_ok = test_structure() |
| removed_ok = test_removed_files() |
| import_ok = test_imports() |
|
|
| print("\n" + "=" * 60) |
| if structure_ok and removed_ok: |
| print("✓ File structure is correct") |
| if import_ok: |
| print("✓ All tests passed!") |
| sys.exit(0) |
| else: |
| print("⚠ Structure OK, but missing Python dependencies") |
| print( |
| " Install with: pip install torch einops gsplat huggingface_hub" |
| ) |
| sys.exit(1) |
| else: |
| print("✗ Some tests failed") |
| sys.exit(1) |
|
|
