Coverage for tests/test_video_utils.py: 100%
13 statements
« prev ^ index » next coverage.py v7.2.5, created at 2023-05-19 16:12 -0700
« prev ^ index » next coverage.py v7.2.5, created at 2023-05-19 16:12 -0700
1from image_utils.video_utils import write_video
2from image_utils import Im
3import numpy as np
4import pytest
5from pathlib import Path
7img_path = Path('tests/flower.jpg')
8save_path = Path(__file__).parent / 'output'
11@pytest.mark.parametrize("fps", [1, 10])
12@pytest.mark.parametrize("frames", [1, 10])
13def test_write_video(frames, fps):
14 img = Im.open(img_path).np
15 video = np.stack([img for _ in range(frames)], axis=0)
16 write_video(video, save_path / 'test.mp4', fps=fps)