wx44wx/pipecat
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix protobuf serializer test to compare attributes instead of frame objects

Browse Source
This commit is contained in:
Luke Payyapilli
2026-01-08 17:00:40 -05:00
parent 1874269a48
commit ccd795445f
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/test_protobuf_serializer.py
View File

@@ -21,7 +21,7 @@ class TestProtobufFrameSerializer(unittest.IsolatedAsyncioTestCase):
async def test_roundtrip(self):
text_frame = TextFrame(text="hello world")
frame = await self.serializer.deserialize(await self.serializer.serialize(text_frame))
self.assertEqual(text_frame, frame)
self.assertEqual(frame.text, text_frame.text)
transcription_frame = TranscriptionFrame(
text="Hello there!", user_id="123", timestamp="2021-01-01"
@@ -29,7 +29,9 @@ class TestProtobufFrameSerializer(unittest.IsolatedAsyncioTestCase):
frame = await self.serializer.deserialize(
await self.serializer.serialize(transcription_frame)
)
self.assertEqual(frame, transcription_frame)
self.assertEqual(frame.text, transcription_frame.text)
self.assertEqual(frame.user_id, transcription_frame.user_id)
self.assertEqual(frame.timestamp, transcription_frame.timestamp)
audio_frame = OutputAudioRawFrame(audio=b"1234567890", sample_rate=16000, num_channels=1)
frame = await self.serializer.deserialize(await self.serializer.serialize(audio_frame))