from src.services.image_service import ImageService
import pytest
import numpy as np
@pytest.fixture
def image_service():
    """Create an ImageService instance for testing."""
    return ImageService()

@pytest.fixture
def sample_images():
    """Create sample images of different types for testing."""
    # RGB image (3 channels)
    rgb = np.zeros((100, 100, 3), dtype=np.uint8)
    rgb[20:80, 20:80] = [255, 128, 64]  # Orange square

    # Grayscale image (1 channel)
    gray = np.zeros((100, 100), dtype=np.uint8)
    gray[20:80, 20:80] = 255  # White square

    # RGBA image (4 channels)
    rgba = np.zeros((100, 100, 4), dtype=np.uint8)
    rgba[20:80, 20:80] = [255, 128, 64, 255]  # Orange square with full opacity

    return {
        'rgb': rgb,
        'gray': gray,
        'rgba': rgba
    }
