Function main

test_gitbase_e2e.py:364–391

Run all end-to-end tests.

()

def main():
    """Run all end-to-end tests."""
    print("🚀 Starting GitBase End-to-End Tests")
    print(f"PyTorch version: {torch.__version__}")
    print(f"CUDA available: {torch.cuda.is_available()}")
    if torch.cuda.is_available():
        print(f"CUDA device: {torch.cuda.get_device_name()}")
        print(f"CUDA memory: {torch.cuda.get_device_properties(0).total_memory / 1024**3:.1f} GB")
    print()

    try:
        test_basic_functionality()
        test_gpu_optimizations()
        test_different_generation_modes()
        test_api_functions()
        test_comprehensive_image_types()
        test_memory_usage()

        print("=" * 60)
        print("🎉 ALL TESTS COMPLETED SUCCESSFULLY!")
        print("=" * 60)

    except Exception as e:
        print(f"❌ Test failed with error: {e}")
        import traceback

        traceback.print_exc()
        sys.exit(1)


if __name__ == "__main__":