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
|
from typing import cast
import av
from av.subtitles.subtitle import AssSubtitle, BitmapSubtitle
from .common import fate_suite
class TestSubtitle:
def test_movtext(self) -> None:
path = fate_suite("sub/MovText_capability_tester.mp4")
subs: list[AssSubtitle] = []
with av.open(path) as container:
for packet in container.demux():
subs.extend(cast(list[AssSubtitle], packet.decode()))
assert len(subs) == 3
sub = subs[0]
assert isinstance(sub, AssSubtitle)
assert sub.type == b"ass"
assert sub.text == b""
assert sub.ass == b"0,0,Default,,0,0,0,,- Test 1.\\N- Test 2."
assert sub.dialogue == b"- Test 1.\n- Test 2."
def test_subset(self) -> None:
path = fate_suite("sub/MovText_capability_tester.mp4")
with av.open(path) as container:
subs = container.streams.subtitles[0]
for packet in container.demux(subs):
subset = subs.decode2(packet)
if subset is not None:
assert not isinstance(subset, av.subtitles.subtitle.Subtitle)
assert isinstance(subset, av.subtitles.subtitle.SubtitleSet)
assert subset.format == 1
assert hasattr(subset, "pts")
assert subset.start_display_time == 0
assert hasattr(subset, "end_display_time")
def test_vobsub(self) -> None:
path = fate_suite("sub/vobsub.sub")
subs: list[BitmapSubtitle] = []
with av.open(path) as container:
for packet in container.demux():
subs.extend(cast(list[BitmapSubtitle], packet.decode()))
assert len(subs) == 43
sub = subs[0]
assert isinstance(sub, BitmapSubtitle)
assert sub.type == b"bitmap"
assert sub.x == 259
assert sub.y == 379
assert sub.width == 200
assert sub.height == 24
assert sub.nb_colors == 4
bms = sub.planes
assert len(bms) == 1
assert len(memoryview(bms[0])) == 4800 # type: ignore
def test_subtitle_flush(self) -> None:
path = fate_suite("sub/MovText_capability_tester.mp4")
subs: list[object] = []
with av.open(path) as container:
stream = container.streams.subtitles[0]
for packet in container.demux(stream):
subs.extend(stream.decode(packet))
subs.extend(stream.decode())
assert len(subs) == 3
|
