"""Test fixtures for demo package."""

import numpy as np
import pytest
import torch


@pytest.fixture
def mock_frame_tensor():
    """Return a mock video frame tensor (C, H, W)."""
    return torch.randn(3, 224, 224)


@pytest.fixture
def mock_frame_array():
    """Return a mock video frame as numpy array (H, W, C)."""
    return np.random.randn(224, 224, 3).astype(np.float32)


@pytest.fixture
def mock_video_sequence():
    """Return a mock video sequence tensor (T, C, H, W)."""
    return torch.randn(16, 3, 224, 224)