subtitle-tools/test_trans_func.py at develop · mayf09/subtitle-tools · GitHub

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
from datetime import timedelta

import srt

from draft_srt import DraftSrt
from utils.trans_func import trans


def test_trans():

    subs = [
        DraftSrt(
            index=1,
            start=timedelta(0, 10, 100),
            end=timedelta(0, 20, 200),
            content='\n'.join(['000,100 100,200 200,300 300,400 400,500', 'one two three| four five'])
        ),
        DraftSrt(
            index=2,
            start=timedelta(0, 30, 100),
            end=timedelta(0, 40, 200),
            content='\n'.join(['000,100 100,200 200,300', 'six| seven| eight|'])
        ),
        DraftSrt(
            index=3,
            start=timedelta(0, 50, 100),
            end=timedelta(0, 60, 200),
            content='\n'.join(['000,100 100,200 200,300', 'nine| ten| eleven'])
        ),
        DraftSrt(
            index=4,
            start=timedelta(0, 70, 100),
            end=timedelta(0, 80, 200),
            content='\n'.join(['000,100 100,200', 'twelve thirteen'])
        ),
        DraftSrt(
            index=5,
            start=timedelta(0, 90, 100),
            end=timedelta(0, 100, 200),
            content='\n'.join(['000,100 200,300', 'fourteen| fifteen'])
        ),
        DraftSrt(
            index=6,
            start=timedelta(0, 110, 100),
            end=timedelta(0, 120, 200),
            content='\n'.join(['000,100', 'sixteen'])
        )
    ]

    trans(subs, 33, lambda l: [str(len(_.split())) for _ in l])

    trans_subs = [
        DraftSrt(
            index=1,
            start=timedelta(0, 10, 100),
            end=timedelta(0, 20, 200),
            content='\n'.join([
                '000,100 100,200 200,300 300,400 400,500',
                'one two three| four five',
                '3|3'
            ])
        ),
        DraftSrt(
            index=2,
            start=timedelta(0, 30, 100),
            end=timedelta(0, 40, 200),
            content='\n'.join(['000,100 100,200 200,300', 'six| seven| eight|', '|1|1|'])
        ),
        DraftSrt(
            index=3,
            start=timedelta(0, 50, 100),
            end=timedelta(0, 60, 200),
            content='\n'.join(['000,100 100,200 200,300', 'nine| ten| eleven', '1|1|4'])
        ),
        DraftSrt(
            index=4,
            start=timedelta(0, 70, 100),
            end=timedelta(0, 80, 200),
            content='\n'.join(['000,100 100,200', 'twelve thirteen'])
        ),
        DraftSrt(
            index=5,
            start=timedelta(0, 90, 100),
            end=timedelta(0, 100, 200),
            content='\n'.join(['000,100 200,300', 'fourteen| fifteen', '|2'])
        ),
        DraftSrt(
            index=6,
            start=timedelta(0, 110, 100),
            end=timedelta(0, 120, 200),
            content='\n'.join(['000,100', 'sixteen'])
        )
    ]

    for i, sub in enumerate(subs):
        assert(sub == trans_subs[i])