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

Improve timestamp test to check frame pts difference

Browse Source
This commit is contained in:
James Hush
2026-02-05 10:12:45 +08:00
parent 63083cd021
commit 5d3aadadf1
1 changed files with 7 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
tests/test_smallwebrtc_transport.py
View File

@@ -115,15 +115,16 @@ class TestRawAudioTrack:
num_chunks = 4 # 40ms num_chunks = 4 # 40ms
track = RawAudioTrack(sample_rate=sample_rate, num_10ms_chunks=num_chunks) track = RawAudioTrack(sample_rate=sample_rate, num_10ms_chunks=num_chunks)
# Initial timestamp should be 0 # Receive first frame and check its timestamp
assert track._timestamp == 0 frame1 = await track.recv()
first_pts = frame1.pts
# Receive one frame (silence is fine) # Receive second frame
await track.recv() frame2 = await track.recv()
# Timestamp should advance by samples_per_chunk # Timestamp should advance by samples_per_chunk between frames
expected_samples = int(sample_rate * 40 / 1000) # 640 samples expected_samples = int(sample_rate * 40 / 1000) # 640 samples
assert track._timestamp == expected_samples assert frame2.pts - first_pts == expected_samples
def test_different_sample_rates(self): def test_different_sample_rates(self):
"""Test chunk size calculation at different sample rates.""" """Test chunk size calculation at different sample rates."""