1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
|
"""Video file clip tests meant to be run with pytest."""
import copy
import os
import pytest
from moviepy.video.compositing.CompositeVideoClip import clips_array
from moviepy.video.io.VideoFileClip import VideoFileClip
from moviepy.video.VideoClip import ColorClip
def test_setup(util):
"""Test VideoFileClip setup."""
filename = os.path.join(util.TMP_DIR, "test.mp4")
red = ColorClip((256, 200), color=(255, 0, 0))
green = ColorClip((256, 200), color=(0, 255, 0))
blue = ColorClip((256, 200), color=(0, 0, 255))
red.fps = green.fps = blue.fps = 10
with clips_array([[red, green, blue]]).with_duration(5) as video:
video.write_videofile(filename, logger=None)
assert os.path.exists(filename)
clip = VideoFileClip(filename)
assert clip.duration == 5
assert clip.fps == 10
assert clip.size == [256 * 3, 200]
assert clip.reader.bitrate == 2
def test_ffmpeg_resizing():
"""Test FFmpeg resizing, to include downscaling."""
video_file = "media/big_buck_bunny_432_433.webm"
target_resolutions = [(128, 128), (128, None), (None, 128), (None, 256)]
for target_resolution in target_resolutions:
video = VideoFileClip(video_file, target_resolution=target_resolution)
frame = video.get_frame(0)
for target, observed in zip(target_resolution[::-1], frame.shape):
if target is not None:
assert target == observed
video.close()
def test_copied_videofileclip_write_videofile(util):
"""Check that a copied ``VideoFileClip`` can be renderizable using
``write_videofile``, opened from that render and the new video shares
the same data that the original clip.
"""
input_video_filepath = "media/big_buck_bunny_432_433.webm"
output_video_filepath = os.path.join(util.TMP_DIR, "copied_videofileclip.mp4")
clip = VideoFileClip(input_video_filepath).subclipped(0, 1)
copied_clip = clip.copy()
copied_clip.write_videofile(output_video_filepath)
assert os.path.exists(output_video_filepath)
copied_clip_from_file = VideoFileClip(output_video_filepath)
assert copied_clip.fps == copied_clip_from_file.fps
assert list(copied_clip.size) == copied_clip_from_file.size
assert isinstance(copied_clip.reader, type(copied_clip_from_file.reader))
def test_videofileclip_safe_deepcopy(monkeypatch):
"""Attempts to do a deepcopy of a VideoFileClip will do a mixed copy,
being redirected to ``__copy__`` method of ``VideoClip``, see the
documentation of ``VideoFileClip.__deepcopy__`` for more information
about this.
"""
clip = VideoFileClip("media/chaplin.mp4")
# patch __copy__ in the clip
def fake__copy__():
return "foo"
monkeypatch.setattr(clip, "__copy__", fake__copy__)
# this should not raise any exception (see `VideoFileClip.__deepcopy__`)
assert copy.deepcopy(clip) == "foo"
def test_ffmpeg_transparency_mask(util):
"""Test VideoFileClip and FFMPEG reading of video with transparency."""
video_file = "media/transparent.webm"
video = VideoFileClip(video_file, has_mask=True)
assert video.mask is not None
mask_frame = video.mask.get_frame(0)
assert mask_frame[100, 100] == 1.0
assert mask_frame[10, 10] == 0
video.close()
if __name__ == "__main__":
pytest.main()
|
