Function create_test_images

test_gitbase_e2e.py:22–60

Create a variety of test images for comprehensive testing.

()

def create_test_images():
    """Create a variety of test images for comprehensive testing."""
    test_images = {}

    # Simple colored squares
    test_images["red_square"] = Image.new("RGB", (224, 224), color=(255, 0, 0))
    test_images["blue_square"] = Image.new("RGB", (224, 224), color=(0, 0, 255))
    test_images["green_square"] = Image.new("RGB", (224, 224), color=(0, 255, 0))

    # Gradient image
    gradient = Image.new("RGB", (224, 224))
    for x in range(224):
        for y in range(224):
            gradient.putpixel((x, y), (x, y, (x + y) // 2))
    test_images["gradient"] = gradient

    # Simple pattern
    pattern = Image.new("RGB", (224, 224), color=(255, 255, 255))
    for x in range(0, 224, 20):
        for y in range(0, 224, 20):
            for i in range(10):
                for j in range(10):
                    if x + i < 224 and y + j < 224:
                        pattern.putpixel((x + i, y + j), (0, 0, 0))
    test_images["checkerboard"] = pattern

    # Load real image if available
    real_image_paths = ["static/img/me.jpg", "static/img/android-chrome-512x512.png", "static/images/logo.png"]

    for path in real_image_paths:
        if os.path.exists(path):
            try:
                test_images[f"real_{os.path.basename(path)}"] = Image.open(path).convert("RGB")
                logger.info(f"Loaded real test image: {path}")
                break
            except Exception as e:
                logger.warning(f"Failed to load {path}: {e}")

    return test_images


def test_basic_functionality():