Skip to content

Commit e576ed9

Browse files
committed
chore: add generate stream test
1 parent 6d4f625 commit e576ed9

File tree

1 file changed

+24
-10
lines changed

1 file changed

+24
-10
lines changed

tests/test_tts.py

Lines changed: 24 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -7,16 +7,6 @@
77
import base64
88

99

10-
def test_tts_convert() -> None:
11-
"""Test basic text-to-speech generation."""
12-
client = ElevenLabs()
13-
audio_generator = client.text_to_speech.convert(text=DEFAULT_TEXT, voice_id=DEFAULT_VOICE, model_id=DEFAULT_MODEL)
14-
audio = b"".join(audio_generator)
15-
assert isinstance(audio, bytes), "TTS should return bytes"
16-
if not IN_GITHUB:
17-
play(audio)
18-
19-
2010
def test_tts_generate() -> None:
2111
"""Test basic text-to-speech generation w/ custom generate."""
2212
client = ElevenLabs()
@@ -44,6 +34,30 @@ def test_tts_generate_with_voice_settings() -> None:
4434
play(audio)
4535

4636

37+
def test_tts_generate_stream() -> None:
38+
"""Test streaming text-to-speech generation."""
39+
client = ElevenLabs()
40+
audio_generator = client.generate(
41+
stream=True,
42+
text=DEFAULT_TEXT,
43+
model=DEFAULT_MODEL,
44+
)
45+
audio = b"".join(audio_generator)
46+
assert isinstance(audio, bytes), "TTS should return bytes"
47+
if not IN_GITHUB:
48+
play(audio)
49+
50+
51+
def test_tts_convert() -> None:
52+
"""Test basic text-to-speech generation."""
53+
client = ElevenLabs()
54+
audio_generator = client.text_to_speech.convert(text=DEFAULT_TEXT, voice_id=DEFAULT_VOICE, model_id=DEFAULT_MODEL)
55+
audio = b"".join(audio_generator)
56+
assert isinstance(audio, bytes), "TTS should return bytes"
57+
if not IN_GITHUB:
58+
play(audio)
59+
60+
4761
def test_tts_convert_with_voice_settings() -> None:
4862
"""Test TTS with custom voice settings."""
4963
client = ElevenLabs()

0 commit comments

Comments
 (0)