|
|
import unittest |
|
|
from transcribe.strategy import TranscriptChunk, TranscriptToken, SplitMode |
|
|
|
|
|
class TestTranscriptChunk(unittest.TestCase): |
|
|
|
|
|
def setUp(self): |
|
|
self.tokens = [ |
|
|
TranscriptToken(text="Hello", t0=0, t1=100), |
|
|
TranscriptToken(text=",", t0=100, t1=200), |
|
|
TranscriptToken(text="world", t0=200, t1=300), |
|
|
TranscriptToken(text=".", t0=300, t1=400), |
|
|
] |
|
|
self.chunk = TranscriptChunk(items=self.tokens, separator=" ") |
|
|
|
|
|
def test_split_by_punctuation(self): |
|
|
chunks = self.chunk.split_by(SplitMode.PUNCTUATION) |
|
|
self.assertEqual(len(chunks), 3) |
|
|
self.assertEqual(chunks[0].join(), "Hello ,") |
|
|
self.assertEqual(chunks[1].join(), "world .") |
|
|
self.assertEqual(chunks[2].join(), "") |
|
|
|
|
|
def test_get_split_first_rest(self): |
|
|
first, rest = self.chunk.get_split_first_rest(SplitMode.PUNCTUATION) |
|
|
self.assertEqual(first.join(), "Hello ,") |
|
|
self.assertEqual(len(rest), 2) |
|
|
self.assertEqual(rest[0].join(), "world .") |
|
|
self.assertEqual(rest[1].join(), "") |
|
|
|
|
|
def test_punctuation_numbers(self): |
|
|
self.assertEqual(self.chunk.puncation_numbers(), 2) |
|
|
|
|
|
def test_length(self): |
|
|
self.assertEqual(self.chunk.length(), 4) |
|
|
|
|
|
def test_join(self): |
|
|
self.assertEqual(self.chunk.join(), "Hello , world .") |
|
|
|
|
|
def test_compare(self): |
|
|
other_chunk = TranscriptChunk(items=[ |
|
|
TranscriptToken(text="Hello", t0=0, t1=100), |
|
|
TranscriptToken(text="!", t0=100, t1=200), |
|
|
], separator=" ") |
|
|
similarity = self.chunk.compare(other_chunk) |
|
|
self.assertTrue(0 < similarity < 1) |
|
|
|
|
|
def test_has_punctuation(self): |
|
|
self.assertTrue(self.chunk.has_punctuation()) |
|
|
|
|
|
def test_get_buffer_index(self): |
|
|
|
|
|
self.assertEqual(self.chunk.get_buffer_index(), 64000) |
|
|
|
|
|
def test_is_end_sentence(self): |
|
|
self.assertTrue(self.chunk.is_end_sentence()) |
|
|
|
|
|
if __name__ == '__main__': |
|
|
unittest.main() |
|
|
|
