Files
iGavroche-rocm-ninodes/tests/test_integration.py
T
Gavroche 1466b7f75c ✅ COMPLETE: ROCM Performance Recovery & Testing Infrastructure
- Restored v1.0.13 with all 9 working nodes (ROCMOptimizedCheckpointLoader, ROCMOptimizedKSampler, etc.)
- Implemented comprehensive debug flag system (ROCM_NINODES_DEBUG=1)
- Created complete test suite with performance and correctness tests
- Added data capture infrastructure for real workflow testing
- Built standalone test runner with mock data support
- Created ComfyUI integration tests for Flux and WAN workflows
- Added comprehensive documentation (ARCHITECTURE.md, RULES.md, TESTING_GUIDE.md)
- Updated README with testing section and performance targets
- All tests passing with proper error handling and data validation

Performance Targets:
- Flux: 78% improvement (≤110s total)
- WAN: 5.6% improvement (≤93s sampling)
- VAE Decode: <10s for any resolution
- Checkpoint Load: <30s for Flux models

Ready for production use with comprehensive testing infrastructure!
2025-10-11 09:56:20 -04:00

41 lines
1.1 KiB
Python

#!/usr/bin/env python3
"""
Main integration test runner
Combines Flux and WAN integration tests
"""
import pytest
import sys
import os
def main():
"""Run all integration tests"""
print("ROCM Ninodes Integration Tests")
print("=============================")
# Run Flux integration tests
print("\n1. Flux Integration Tests")
print("-" * 30)
flux_result = pytest.main(["-v", "test_integration_flux.py"])
# Run WAN integration tests
print("\n2. WAN Integration Tests")
print("-" * 30)
wan_result = pytest.main(["-v", "test_integration_wan.py"])
# Summary
print("\nIntegration Test Summary")
print("========================")
print(f"Flux tests: {'PASSED' if flux_result == 0 else 'FAILED'}")
print(f"WAN tests: {'PASSED' if wan_result == 0 else 'FAILED'}")
if flux_result == 0 and wan_result == 0:
print("\n✓ All integration tests passed!")
return 0
else:
print("\n✗ Some integration tests failed!")
return 1
if __name__ == "__main__":
sys.exit(main())