blackteay/ffmpeg-python
1
0
Fork 0
mirror of https://github.com/kkroening/ffmpeg-python.git synced 2025-04-06 04:15:44 +08:00
Code Issues Projects Releases Wiki Activity

Run Black code formatter

Browse Source
This commit is contained in:
Karl Kroening 2019-07-05 19:16:43 -05:00
parent c14efc9a19
commit 1c9695d2a0
1 changed files with 12 additions and 7 deletions

19
ffmpeg/tests/test_ffmpeg.py

@ -717,6 +717,7 @@ def test__probe__extra_args():
data = ffmpeg.probe(TEST_INPUT_FILE1, show_frames=None) data = ffmpeg.probe(TEST_INPUT_FILE1, show_frames=None)
assert set(data.keys()) == {'format', 'streams', 'frames'} assert set(data.keys()) == {'format', 'streams', 'frames'}
def get_filter_complex_input(flt, name): def get_filter_complex_input(flt, name):
m = re.search(r'\[([^]]+)\]{}(?=[[;]|$)'.format(name), flt) m = re.search(r'\[([^]]+)\]{}(?=[[;]|$)'.format(name), flt)
if m: if m:
@ -724,6 +725,7 @@ def get_filter_complex_input(flt, name):
else: else:
return None return None
def get_filter_complex_outputs(flt, name): def get_filter_complex_outputs(flt, name):
m = re.search(r'(^|[];]){}((\[[^]]+\])+)(?=;|$)'.format(name), flt) m = re.search(r'(^|[];]){}((\[[^]]+\])+)(?=;|$)'.format(name), flt)
if m: if m:
@ -731,6 +733,7 @@ def get_filter_complex_outputs(flt, name):
else: else:
return None return None
def test__get_filter_complex_input(): def test__get_filter_complex_input():
assert get_filter_complex_input("", "scale") is None assert get_filter_complex_input("", "scale") is None
assert get_filter_complex_input("scale", "scale") is None assert get_filter_complex_input("scale", "scale") is None
@ -739,6 +742,7 @@ def test__get_filter_complex_input():
assert get_filter_complex_input("[s2]scale;etc", "scale") == "s2" assert get_filter_complex_input("[s2]scale;etc", "scale") == "s2"
assert get_filter_complex_input("[s2]scale[s3][s4];etc", "scale") == "s2" assert get_filter_complex_input("[s2]scale[s3][s4];etc", "scale") == "s2"
def test__get_filter_complex_outputs(): def test__get_filter_complex_outputs():
assert get_filter_complex_outputs("", "scale") is None assert get_filter_complex_outputs("", "scale") is None
assert get_filter_complex_outputs("scale", "scale") is None assert get_filter_complex_outputs("scale", "scale") is None
@ -750,17 +754,18 @@ def test__get_filter_complex_outputs():
assert get_filter_complex_outputs("[s5]scale[s1];x", "scale") == ['s1'] assert get_filter_complex_outputs("[s5]scale[s1];x", "scale") == ['s1']
assert get_filter_complex_outputs("y;[s5]scale[s1];x", "scale") == ['s1'] assert get_filter_complex_outputs("y;[s5]scale[s1];x", "scale") == ['s1']
def test__multi_output_edge_label_order(): def test__multi_output_edge_label_order():
scale2ref = ffmpeg.filter_multi_output([ffmpeg.input('x'), ffmpeg.input('y')], 'scale2ref') scale2ref = ffmpeg.filter_multi_output(
out = ( [ffmpeg.input('x'), ffmpeg.input('y')], 'scale2ref'
ffmpeg.merge_outputs( )
scale2ref[1].filter('scale').output('a'), out = ffmpeg.merge_outputs(
scale2ref[10000].filter('hflip').output('b') scale2ref[1].filter('scale').output('a'),
) scale2ref[10000].filter('hflip').output('b'),
) )
args = out.get_args() args = out.get_args()
flt_cmpl = args[args.index('-filter_complex')+1] flt_cmpl = args[args.index('-filter_complex') + 1]
out1, out2 = get_filter_complex_outputs(flt_cmpl, 'scale2ref') out1, out2 = get_filter_complex_outputs(flt_cmpl, 'scale2ref')
assert out1 == get_filter_complex_input(flt_cmpl, 'scale') assert out1 == get_filter_complex_input(flt_cmpl, 'scale')
assert out2 == get_filter_complex_input(flt_cmpl, 'hflip') assert out2 == get_filter_complex_input(flt_cmpl, 'hflip')