31 lines
1.1 KiB
Python
31 lines
1.1 KiB
Python
import unittest
|
|
|
|
from pipecat.pipeline.frames import AudioFrame, TextFrame, TranscriptionFrame
|
|
from pipecat.serializers.protobuf_serializer import ProtobufFrameSerializer
|
|
|
|
|
|
class TestProtobufFrameSerializer(unittest.IsolatedAsyncioTestCase):
|
|
def setUp(self):
|
|
self.serializer = ProtobufFrameSerializer()
|
|
|
|
async def test_roundtrip(self):
|
|
text_frame = TextFrame(text='hello world')
|
|
frame = self.serializer.deserialize(
|
|
self.serializer.serialize(text_frame))
|
|
self.assertEqual(frame, TextFrame(text='hello world'))
|
|
|
|
transcription_frame = TranscriptionFrame(
|
|
text="Hello there!", participantId="123", timestamp="2021-01-01")
|
|
frame = self.serializer.deserialize(
|
|
self.serializer.serialize(transcription_frame))
|
|
self.assertEqual(frame, transcription_frame)
|
|
|
|
audio_frame = AudioFrame(data=b'1234567890')
|
|
frame = self.serializer.deserialize(
|
|
self.serializer.serialize(audio_frame))
|
|
self.assertEqual(frame, audio_frame)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|