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

1from image_utils.video_utils import write_video 

2from image_utils import Im 

3import numpy as np 

4import pytest 

5from pathlib import Path 

6 

7img_path = Path('tests/flower.jpg') 

8save_path = Path(__file__).parent / 'output' 

9 

10 

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)