"""Tests for preprocessing pipeline."""

import numpy as np

from sups_yolo.preprocessing.pipeline import PreprocessingPipeline


def test_resize() -> None:
    pipeline = PreprocessingPipeline({"resize": [640, 480]})
    image = np.zeros((1080, 1920, 3), dtype=np.uint8)
    result = pipeline.run(image)
    assert result.shape == (480, 640, 3)


def test_rotation_preserves_shape() -> None:
    pipeline = PreprocessingPipeline({"rotation": 5})
    image = np.zeros((1080, 1920, 3), dtype=np.uint8)
    result = pipeline.run(image)
    assert result.shape == (1080, 1920, 3)