diff --git a/tests/test_protobuf_serializer.py b/tests/test_protobuf_serializer.py index 5e289286f..88fd7867e 100644 --- a/tests/test_protobuf_serializer.py +++ b/tests/test_protobuf_serializer.py @@ -6,7 +6,11 @@ import unittest -from pipecat.frames.frames import AudioRawFrame, TextFrame, TranscriptionFrame +from pipecat.frames.frames import ( + OutputAudioRawFrame, + TextFrame, + TranscriptionFrame, +) from pipecat.serializers.protobuf import ProtobufFrameSerializer @@ -14,22 +18,19 @@ class TestProtobufFrameSerializer(unittest.IsolatedAsyncioTestCase): def setUp(self): self.serializer = ProtobufFrameSerializer() - @unittest.skip("FIXME: This test is failing") 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")) + self.assertEqual(text_frame, frame) 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)) 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)) - self.assertEqual(frame, audio_frame) - - -if __name__ == "__main__": - unittest.main() + 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)