tests: fix test_protobuf_serializer.py
This commit is contained in:
@@ -6,7 +6,11 @@
|
|||||||
|
|
||||||
import unittest
|
import unittest
|
||||||
|
|
||||||
from pipecat.frames.frames import AudioRawFrame, TextFrame, TranscriptionFrame
|
from pipecat.frames.frames import (
|
||||||
|
OutputAudioRawFrame,
|
||||||
|
TextFrame,
|
||||||
|
TranscriptionFrame,
|
||||||
|
)
|
||||||
from pipecat.serializers.protobuf import ProtobufFrameSerializer
|
from pipecat.serializers.protobuf import ProtobufFrameSerializer
|
||||||
|
|
||||||
|
|
||||||
@@ -14,22 +18,19 @@ class TestProtobufFrameSerializer(unittest.IsolatedAsyncioTestCase):
|
|||||||
def setUp(self):
|
def setUp(self):
|
||||||
self.serializer = ProtobufFrameSerializer()
|
self.serializer = ProtobufFrameSerializer()
|
||||||
|
|
||||||
@unittest.skip("FIXME: This test is failing")
|
|
||||||
async def test_roundtrip(self):
|
async def test_roundtrip(self):
|
||||||
text_frame = TextFrame(text="hello world")
|
text_frame = TextFrame(text="hello world")
|
||||||
frame = self.serializer.deserialize(self.serializer.serialize(text_frame))
|
frame = self.serializer.deserialize(self.serializer.serialize(text_frame))
|
||||||
self.assertEqual(frame, TextFrame(text="hello world"))
|
self.assertEqual(text_frame, frame)
|
||||||
|
|
||||||
transcription_frame = TranscriptionFrame(
|
transcription_frame = TranscriptionFrame(
|
||||||
text="Hello there!", participantId="123", timestamp="2021-01-01"
|
text="Hello there!", user_id="123", timestamp="2021-01-01"
|
||||||
)
|
)
|
||||||
frame = self.serializer.deserialize(self.serializer.serialize(transcription_frame))
|
frame = self.serializer.deserialize(self.serializer.serialize(transcription_frame))
|
||||||
self.assertEqual(frame, transcription_frame)
|
self.assertEqual(frame, transcription_frame)
|
||||||
|
|
||||||
audio_frame = AudioRawFrame(data=b"1234567890")
|
audio_frame = OutputAudioRawFrame(audio=b"1234567890", sample_rate=16000, num_channels=1)
|
||||||
frame = self.serializer.deserialize(self.serializer.serialize(audio_frame))
|
frame = self.serializer.deserialize(self.serializer.serialize(audio_frame))
|
||||||
self.assertEqual(frame, audio_frame)
|
self.assertEqual(frame.audio, audio_frame.audio)
|
||||||
|
self.assertEqual(frame.sample_rate, audio_frame.sample_rate)
|
||||||
|
self.assertEqual(frame.num_channels, audio_frame.num_channels)
|
||||||
if __name__ == "__main__":
|
|
||||||
unittest.main()
|
|
||||||
|
|||||||
Reference in New Issue
Block a user