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
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
|
"""FFmpeg tools tests for moviepy."""
import os
import shutil
import pytest
from moviepy.video.io.ffmpeg_tools import (
ffmpeg_extract_subclip,
ffmpeg_resize,
ffmpeg_stabilize_video,
)
from moviepy.video.io.VideoFileClip import VideoFileClip
def test_ffmpeg_extract_subclip(util):
extract_subclip_tempdir = os.path.join(
util.TMP_DIR, "moviepy_ffmpeg_extract_subclip"
)
if os.path.isdir(extract_subclip_tempdir):
shutil.rmtree(extract_subclip_tempdir)
os.mkdir(extract_subclip_tempdir)
inputfile = os.path.join(extract_subclip_tempdir, "fire2.mp4")
shutil.copyfile("media/fire2.mp4", inputfile)
# default name
expected_outputfile = os.path.join(extract_subclip_tempdir, "fire2SUB300_500.mp4")
ffmpeg_extract_subclip(inputfile, 0.3, "00:00:00,5", logger=None)
assert os.path.isfile(expected_outputfile)
# custom name
expected_outputfile = os.path.join(extract_subclip_tempdir, "foo.mp4")
ffmpeg_extract_subclip(
inputfile, 0.3, "00:00:00,5", outputfile=expected_outputfile, logger=None
)
assert os.path.isfile(expected_outputfile)
# assert subclip duration
clip = VideoFileClip(expected_outputfile)
assert 0.18 <= clip.duration <= 0.22 # not accurate
if os.path.isdir(extract_subclip_tempdir):
try:
shutil.rmtree(extract_subclip_tempdir)
except PermissionError:
pass
def test_ffmpeg_resize(util):
outputfile = os.path.join(util.TMP_DIR, "moviepy_ffmpeg_resize.mp4")
if os.path.isfile(outputfile):
os.remove(outputfile)
expected_size = (30, 30)
ffmpeg_resize("media/bitmap.mp4", outputfile, expected_size, logger=None)
assert os.path.isfile(outputfile)
# overwrite file
with pytest.raises(OSError):
ffmpeg_resize("media/bitmap.mp4", outputfile, expected_size, logger=None)
clip = VideoFileClip(outputfile)
assert clip.size[0] == expected_size[0]
assert clip.size[1] == expected_size[1]
if os.path.isfile(outputfile):
try:
os.remove(outputfile)
except PermissionError:
pass
def test_ffmpeg_stabilize_video(util):
stabilize_video_tempdir = os.path.join(util.TMP_DIR, "moviepy_ffmpeg_stabilize")
if os.path.isdir(stabilize_video_tempdir):
shutil.rmtree(stabilize_video_tempdir)
os.mkdir(stabilize_video_tempdir)
# no output file
ffmpeg_stabilize_video(
"media/bitmap.mp4",
output_dir=stabilize_video_tempdir,
logger=None,
)
expected_filepath = os.path.join(stabilize_video_tempdir, "bitmap_stabilized.mp4")
assert os.path.isfile(expected_filepath)
# with output file
ffmpeg_stabilize_video(
"media/bitmap.mp4",
output_dir=stabilize_video_tempdir,
outputfile="foo.mp4",
logger=None,
)
expected_filepath = os.path.join(stabilize_video_tempdir, "foo.mp4")
assert os.path.isfile(expected_filepath)
# don't overwrite file
with pytest.raises(OSError):
ffmpeg_stabilize_video(
"media/bitmap.mp4",
output_dir=stabilize_video_tempdir,
outputfile="foo.mp4",
overwrite_file=False,
logger=None,
)
if os.path.isdir(stabilize_video_tempdir):
try:
shutil.rmtree(stabilize_video_tempdir)
except PermissionError:
pass
if __name__ == "__main__":
pytest.main()
|
