-
Notifications
You must be signed in to change notification settings - Fork 1
/
test_annotate.py
41 lines (35 loc) · 1.23 KB
/
test_annotate.py
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
from dataclasses import dataclass, field
from pathlib import Path
import json
import shutil
from unittest.mock import patch
import app
from annotate import annotate
import common
fixtures = Path("fixtures")
@dataclass
class MockedStub:
transcriptions: dict = field(default_factory=dict)
annotate_stub = MockedStub()
@patch('annotate.stub', new=annotate_stub)
@patch('common.stub', new=annotate_stub)
def test_annotate(transcription_id = "one.wav"):
with common.tmpdir_scope() as tmp:
media_path = Path(tmp)
with patch('common.db', new=common.Store(media_path)):
from_file = fixtures / transcription_id
to_file = media_path / transcription_id
shutil.copyfile(from_file, to_file)
t = common.Transcription(
transcription_id=transcription_id,
path=to_file,
upload=common.UploadInfo(
filename="file.name",
content_type="audio/mp3",
size_bytes=15
)
)
common.db.create(t)
diarization = annotate.local(transcription_id)
assert len(diarization.turns) == 1
assert diarization.turns[0].speaker == "SPEAKER_00"