wehub-resource-sync
54 lines
2.4 KiB
Python
54 lines
2.4 KiB
Python
from openai.types.realtime.realtime_audio_formats import AudioPCM, AudioPCMA, AudioPCMU
|
|
|
|
from agents.realtime.audio_formats import to_realtime_audio_format
|
|
|
|
|
|
def test_to_realtime_audio_format_from_strings():
|
|
assert to_realtime_audio_format("pcm").type == "audio/pcm" # type: ignore[union-attr]
|
|
assert to_realtime_audio_format("pcm16").type == "audio/pcm" # type: ignore[union-attr]
|
|
assert to_realtime_audio_format("audio/pcm").type == "audio/pcm" # type: ignore[union-attr]
|
|
assert to_realtime_audio_format("pcmu").type == "audio/pcmu" # type: ignore[union-attr]
|
|
assert to_realtime_audio_format("audio/pcmu").type == "audio/pcmu" # type: ignore[union-attr]
|
|
assert to_realtime_audio_format("g711_ulaw").type == "audio/pcmu" # type: ignore[union-attr]
|
|
assert to_realtime_audio_format("pcma").type == "audio/pcma" # type: ignore[union-attr]
|
|
assert to_realtime_audio_format("audio/pcma").type == "audio/pcma" # type: ignore[union-attr]
|
|
assert to_realtime_audio_format("g711_alaw").type == "audio/pcma" # type: ignore[union-attr]
|
|
|
|
|
|
def test_to_realtime_audio_format_passthrough_and_unknown_logs():
|
|
fmt = AudioPCM(type="audio/pcm", rate=24000)
|
|
# Passing a RealtimeAudioFormats should return the same instance
|
|
assert to_realtime_audio_format(fmt) is fmt
|
|
|
|
# Unknown string returns None (and logs at debug level internally)
|
|
assert to_realtime_audio_format("something_else") is None
|
|
|
|
|
|
def test_to_realtime_audio_format_none():
|
|
assert to_realtime_audio_format(None) is None
|
|
|
|
|
|
def test_to_realtime_audio_format_from_mapping():
|
|
pcm_exact_rate = to_realtime_audio_format({"type": "audio/pcm", "rate": 24000})
|
|
assert isinstance(pcm_exact_rate, AudioPCM)
|
|
assert pcm_exact_rate.rate == 24000
|
|
|
|
pcm = to_realtime_audio_format({"type": "audio/pcm", "rate": 16000})
|
|
assert isinstance(pcm, AudioPCM)
|
|
assert pcm.type == "audio/pcm"
|
|
assert pcm.rate == 24000
|
|
|
|
pcm_default_rate = to_realtime_audio_format({"type": "audio/pcm"})
|
|
assert isinstance(pcm_default_rate, AudioPCM)
|
|
assert pcm_default_rate.rate == 24000
|
|
|
|
ulaw = to_realtime_audio_format({"type": "audio/pcmu"})
|
|
assert isinstance(ulaw, AudioPCMU)
|
|
assert ulaw.type == "audio/pcmu"
|
|
|
|
alaw = to_realtime_audio_format({"type": "audio/pcma"})
|
|
assert isinstance(alaw, AudioPCMA)
|
|
assert alaw.type == "audio/pcma"
|
|
|
|
assert to_realtime_audio_format({"type": "audio/unknown", "rate": 8000}) is None
|