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
|
from textwrap import dedent
from pysubs2 import SSAFile, SSAEvent, make_time
def test_simple_parsing() -> None:
test_input1 = "[123][456] Line 1"
subs1 = SSAFile.from_string(test_input1)
assert len(subs1) == 1
assert subs1[0] == SSAEvent(start=make_time(ms=12300), end=make_time(ms=45600), text="Line 1")
test_input2 = "[123][456] / Line 1| Line 2/2"
subs2 = SSAFile.from_string(test_input2)
assert len(subs2) == 1
assert subs2[0] == SSAEvent(start=make_time(ms=12300), end=make_time(ms=45600), text=r"{\i1}Line 1{\i0}\NLine2/2")
test_input3 = dedent("""
[123][456] Line 1
[321][456] / Line 2| Line 3
(123)(456)This line should not be parsed
This line should also not be parsed
[789][1234] /Line 4""")
subs3 = SSAFile.from_string(test_input3)
assert len(subs3) == 3
assert subs3[0] == SSAEvent(start=make_time(ms=12300), end=make_time(ms=45600), text="Line 1")
assert subs3[1] == SSAEvent(start=make_time(ms=32100), end=make_time(ms=45600), text=r"{\i1}Line 2{\i0}\NLine 3")
assert subs3[2] == SSAEvent(start=make_time(ms=78900), end=make_time(ms=123400), text=r"{\i1}Line 4{\i0}")
def test_simple_writing() -> None:
subs = SSAFile()
subs.append(SSAEvent(start=0, end=1000, text="Hello!"))
subs.append(SSAEvent(start=1000, end=2000, text="Hello World!\\NTwo-line subtitle!"))
subs.append(SSAEvent(start=2000, end=3000, text=r"{\i1}This is an italic line {with some hidden content}"))
subs.append(SSAEvent(start=3000, end=4000, text="This is a comment line"))
subs[-1].is_comment = True
reference_output = dedent("""\
[0][10] Hello!
[10][20] Hello World!|Two-line subtitle!
[20][30] This is an italic line
""")
assert subs.to_string("mpl2").strip() == reference_output.strip()
|
