1
1
import json
2
2
import os
3
- import platform
4
3
import tempfile
5
4
6
5
import numpy as np
@@ -13,7 +12,6 @@ class TestCmd:
13
12
14
13
# TODO:
15
14
# - To check transcription out of the final video
16
- # - Check speed in macOS
17
15
def _get_transcription (self , filename ):
18
16
model = WhisperModel ("medium" )
19
17
segments , info = model .transcribe (filename , language = "ca" , temperature = [0 ])
@@ -43,7 +41,6 @@ def _get_utterances(self, directory):
43
41
return utterances
44
42
45
43
def _assert_dubbing_action (self , directory ):
46
- # operating = platform.system().lower()
47
44
utterances = self ._get_utterances (directory )
48
45
text_array = [entry ["translated_text" ] for entry in utterances ]
49
46
@@ -56,8 +53,6 @@ def _assert_dubbing_action(self, directory):
56
53
ends = [entry ["end" ] for entry in utterances ]
57
54
speeds = [entry ["speed" ] for entry in utterances ]
58
55
59
- print (f"starts: { starts } " )
60
-
61
56
assert np .allclose (
62
57
[
63
58
1.26284375 ,
@@ -68,7 +63,6 @@ def _assert_dubbing_action(self, directory):
68
63
atol = 0.5 ,
69
64
), "Utterance start check failed"
70
65
71
- print (f"ends: { ends } " )
72
66
assert np .allclose (
73
67
[
74
68
3.94596875 ,
@@ -79,7 +73,6 @@ def _assert_dubbing_action(self, directory):
79
73
atol = 0.5 ,
80
74
), "Utterance end check failed"
81
75
82
- print (f"speeds: { speeds } " )
83
76
assert np .allclose (
84
77
[1.0 , 1.0 , 1.3 ], speeds , atol = 2
85
78
), "Utterance speed check failed"
@@ -89,7 +82,6 @@ def _assert_dubbing_action(self, directory):
89
82
assert "I m'encanta aquesta ciutat." == text_array [2 ], "translated text 2"
90
83
91
84
def _assert_update_action (self , directory ):
92
- operating = platform .system ().lower ()
93
85
utterances = self ._get_utterances (directory )
94
86
text_array = [entry ["translated_text" ] for entry in utterances ]
95
87
0 commit comments