crosstyan
00fcda4fe3
Move demo implementation into opengait_studio, retire Sports2D runtime integration, and align packaging with root-level monorepo dependency management.
24 lines
526 B
Python
24 lines
526 B
Python
"""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)
|