Function test_basic_functionality

test_gitbase_e2e.py:63–95

Test basic GitBase functionality.

()

def test_basic_functionality():
    """Test basic GitBase functionality."""
    print("=" * 60)
    print("TESTING BASIC GITBASE FUNCTIONALITY")
    print("=" * 60)

    test_images = create_test_images()

    # Test with CPU first for compatibility
    print("\n--- Testing CPU Implementation ---")
    captioner_cpu = GitBaseCaptioner(dtype=torch.float32, device="cpu", use_compile=False, use_channels_last=False)

    for name, image in list(test_images.items())[:2]:  # Test first 2 images
        try:
            start_time = time.time()
            caption = captioner_cpu.caption_image(image)
            elapsed = time.time() - start_time

            print(f"Image: {name}")
            print(f"Caption: {caption}")
            print(f"Time: {elapsed:.3f}s")
            print()

            assert isinstance(caption, str)
            assert len(caption) > 0

        except Exception as e:
            print(f"ERROR with {name}: {e}")
            import traceback

            traceback.print_exc()

    print("✅ Basic CPU functionality test passed!")


def test_gpu_optimizations():