|
|
import unittest |
|
|
from transcribe.strategy import TranscriptBuffer |
|
|
|
|
|
class TestTranscriptBuffer(unittest.TestCase): |
|
|
|
|
|
def setUp(self): |
|
|
self.buffer = TranscriptBuffer() |
|
|
|
|
|
def test_update_pending_text(self): |
|
|
self.buffer.update_pending_text("Hello") |
|
|
self.assertEqual(self.buffer.pending_text, "Hello") |
|
|
|
|
|
def test_commit_line(self): |
|
|
self.buffer.update_pending_text("This is a line.") |
|
|
self.buffer.commit_line() |
|
|
self.assertEqual(self.buffer.paragraph, "This is a line.") |
|
|
self.assertEqual(self.buffer.pending_text, "") |
|
|
|
|
|
def test_commit_paragraph(self): |
|
|
self.buffer.update_pending_text("Sentence 1.") |
|
|
self.buffer.commit_line() |
|
|
self.buffer.update_pending_text("Sentence 2.") |
|
|
self.buffer.commit_line() |
|
|
self.buffer.commit_paragraph(end_of_sentence=True) |
|
|
|
|
|
self.assertEqual(self.buffer.get_seg_id(), 1) |
|
|
self.assertEqual(self.buffer.latest_paragraph, "Sentence 1.Sentence 2.") |
|
|
self.assertEqual(self.buffer.paragraph, "") |
|
|
self.assertEqual(self.buffer.pending_text, "") |
|
|
|
|
|
def test_commit_paragraph_without_end(self): |
|
|
self.buffer.update_pending_text("Incomplete sentence.") |
|
|
self.buffer.commit_line() |
|
|
self.buffer.commit_paragraph(end_of_sentence=False) |
|
|
|
|
|
|
|
|
self.assertEqual(self.buffer.get_seg_id(), 0) |
|
|
self.assertEqual(self.buffer.paragraph, "Incomplete sentence.") |
|
|
|
|
|
def test_update_and_commit_end_sentence(self): |
|
|
self.buffer.update_and_commit("Stable.", "Remaining", is_end_sentence=True) |
|
|
|
|
|
self.assertEqual(self.buffer.latest_paragraph, "Stable.") |
|
|
self.assertEqual(self.buffer.pending_text, "Remaining") |
|
|
self.assertEqual(self.buffer.paragraph, "") |
|
|
|
|
|
def test_update_and_commit_partial_sentence(self): |
|
|
self.buffer.update_and_commit("Partial", "New Buffer", is_end_sentence=False) |
|
|
|
|
|
self.assertEqual(self.buffer.paragraph, "Partial") |
|
|
self.assertEqual(self.buffer.pending_text, "New Buffer") |
|
|
self.assertEqual(self.buffer.get_seg_id(), 0) |
|
|
|
|
|
def test_current_not_commit_text(self): |
|
|
self.buffer.update_and_commit("Part 1.", "Live text", is_end_sentence=False) |
|
|
self.assertEqual(self.buffer.current_not_commit_text, "Part 1.Live text") |
|
|
|
|
|
if __name__ == "__main__": |
|
|
unittest.main() |
|
|
|
