Function test_api_functions

test_gitbase_e2e.py:231–273

Test module-level API functions.

()

def test_api_functions():
    """Test module-level API functions."""
    print("=" * 60)
    print("TESTING API FUNCTIONS")
    print("=" * 60)

    test_images = create_test_images()
    sample_image = list(test_images.values())[0]

    # Test get_gitbase_captioner caching
    print("\n--- Testing Cached Captioner ---")
    captioner1 = get_gitbase_captioner(dtype=torch.float16)
    captioner2 = get_gitbase_captioner(dtype=torch.float16)

    print(f"Same instance returned: {captioner1 is captioner2}")
    assert captioner1 is captioner2, "Caching not working properly"

    # Test caption_image_bytes
    print("\n--- Testing Bytes API ---")
    from io import BytesIO

    # Convert image to bytes
    buffer = BytesIO()
    sample_image.save(buffer, format="JPEG")
    image_bytes = buffer.getvalue()

    # Test fast mode
    start_time = time.time()
    caption_fast = caption_image_bytes(image_bytes, fast_mode=True)
    fast_time = time.time() - start_time

    # Test quality mode
    start_time = time.time()
    caption_quality = caption_image_bytes(image_bytes, fast_mode=False)
    quality_time = time.time() - start_time

    print(f"Fast caption: {caption_fast} (time: {fast_time:.3f}s)")
    print(f"Quality caption: {caption_quality} (time: {quality_time:.3f}s)")

    assert isinstance(caption_fast, str)
    assert isinstance(caption_quality, str)
    assert len(caption_fast) > 0
    assert len(caption_quality) > 0


def test_comprehensive_image_types():