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
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
|
"""Define general test helper attributes and utilities."""
import ast
import contextlib
import functools
import http.server
import importlib
import inspect
import io
import pkgutil
import socketserver
import tempfile
import threading
import numpy as np
import pytest
from moviepy.video.io.VideoFileClip import VideoFileClip
TMP_DIR = tempfile.gettempdir() # because tempfile.tempdir is sometimes None
# Arbitrary font used in caption testing.
FONT = "media/doc_medias/example.ttf"
# Dir for doc examples medias
DOC_EXAMPLES_MEDIAS_DIR = "media/doc_medias"
@functools.lru_cache(maxsize=None)
def get_video(start_time=0, end_time=1):
return VideoFileClip("media/big_buck_bunny_432_433.webm").subclipped(
start_time, end_time
)
@functools.lru_cache(maxsize=None)
def get_stereo_wave(left_freq=440, right_freq=220):
def make_stereo_frame(t):
return np.array(
[np.sin(left_freq * 2 * np.pi * t), np.sin(right_freq * 2 * np.pi * t)]
).T.copy(order="C")
return make_stereo_frame
@functools.lru_cache(maxsize=None)
def get_mono_wave(freq=440):
def make_mono_frame(t):
return np.sin(freq * 2 * np.pi * t)
return make_mono_frame
@contextlib.contextmanager
def get_static_files_server(port=8000):
my_server = socketserver.TCPServer(("", port), http.server.SimpleHTTPRequestHandler)
thread = threading.Thread(target=my_server.serve_forever, daemon=True)
thread.start()
yield thread
@functools.lru_cache(maxsize=None)
def get_moviepy_modules():
"""Get all moviepy module names and if each one is a package."""
response = []
with contextlib.redirect_stdout(io.StringIO()):
moviepy_module = importlib.import_module("moviepy")
modules = pkgutil.walk_packages(
path=moviepy_module.__path__,
prefix=moviepy_module.__name__ + ".",
)
for importer, modname, ispkg in modules:
response.append((modname, ispkg))
return response
def get_functions_with_decorator_defined(code, decorator_name):
"""Get all functions in a code object which have a decorator defined,
along with the arguments of the function and the decorator.
Parameters
----------
code : object
Module or class object from which to retrieve the functions.
decorator_name : str
Name of the decorator defined in the functions to search.
"""
class FunctionsWithDefinedDecoratorExtractor(ast.NodeVisitor):
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.functions_with_decorator = []
def generic_visit(self, node):
if isinstance(node, ast.FunctionDef) and node.decorator_list:
for dec in node.decorator_list:
if not isinstance(dec, ast.Call) or dec.func.id != decorator_name:
continue
decorator_argument_names = []
if isinstance(dec.args, ast.List):
for args in dec.args:
decorator_argument_names.extend(
[e.value for e in args.elts]
)
else:
for args in dec.args:
if isinstance(args, (ast.List, ast.Tuple)):
decorator_argument_names.extend(
[e.value for e in args.elts]
)
else:
decorator_argument_names.append(args.value)
function_argument_names = [arg.arg for arg in node.args.args]
for arg in node.args.kwonlyargs:
function_argument_names.append(arg.arg)
self.functions_with_decorator.append(
{
"function_name": node.name,
"function_arguments": function_argument_names,
"decorator_arguments": decorator_argument_names,
}
)
ast.NodeVisitor.generic_visit(self, node)
modtree = ast.parse(inspect.getsource(code))
visitor = FunctionsWithDefinedDecoratorExtractor()
visitor.visit(modtree)
return visitor.functions_with_decorator
@pytest.fixture
def util():
class MoviepyTestUtils:
FONT = FONT
TMP_DIR = TMP_DIR
DOC_EXAMPLES_MEDIAS_DIR = DOC_EXAMPLES_MEDIAS_DIR
return MoviepyTestUtils
@pytest.fixture
def video():
return get_video
@pytest.fixture
def stereo_wave():
return get_stereo_wave
@pytest.fixture
def mono_wave():
return get_mono_wave
@pytest.fixture
def static_files_server():
return get_static_files_server
@pytest.fixture
def moviepy_modules():
return get_moviepy_modules
@pytest.fixture
def functions_with_decorator_defined():
return get_functions_with_decorator_defined
|
