added speaker_id conversions to Amazon and Google
This commit is contained in:
38
tests/test_amazon.py
Normal file
38
tests/test_amazon.py
Normal file
@@ -0,0 +1,38 @@
|
||||
import json
|
||||
import os
|
||||
|
||||
import pytest
|
||||
|
||||
from transcript_processing.converters.amazon import AmazonConverter
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def transcript_data():
|
||||
with open(os.getenv('AMAZON_TRANSCRIPT_TEST_FILE'), 'r') as fin:
|
||||
return json.load(fin)
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def converter(transcript_data):
|
||||
return AmazonConverter(transcript_data)
|
||||
|
||||
|
||||
def test_get_word_objects(converter):
|
||||
word_objects = converter.get_word_objects(converter.json_data)
|
||||
assert word_objects
|
||||
|
||||
|
||||
def test_get_speaker_segments(converter):
|
||||
speaker_segments = converter.get_speaker_segments()
|
||||
assert speaker_segments
|
||||
|
||||
|
||||
def test_get_speaker_id(converter):
|
||||
speaker_segments = converter.get_speaker_segments()
|
||||
assert speaker_segments[54.58] == 0
|
||||
assert speaker_segments[32.36] == 1
|
||||
|
||||
|
||||
def test_convert(converter):
|
||||
converter.convert()
|
||||
print(converter.converted_words)
|
||||
Reference in New Issue
Block a user