# This is a combination of 2 commits.

# This is the 1st commit message:

feat: add e2e tests

# This is the commit message elevenlabs#2:

feat: update tests

Author: louisjoecodes

tests/test_async_generation.py

@@ -1,36 +1,8 @@
-import time
-from random import randint
-
-from elevenlabs import GetSpeechHistoryResponse, \
-    play
-
-from .utils import IN_GITHUB, client
+from elevenlabs import GetSpeechHistoryResponse, ElevenLabs
 
 
 def test_history():
+    client = ElevenLabs()
     page_size = 5
     history = client.history.get_all(page_size=page_size)
     assert isinstance(history, GetSpeechHistoryResponse)
-
-
-def test_history_item_delete():
-    text = f"Test {randint(0, 1000)}"
-    audio = client.generate(text=text)
-    if not IN_GITHUB:
-        play(audio)  # type: ignore
-
-    time.sleep(1)
-
-    history = client.history.get_all().history
-    print(history)
-    history_item = history[0]
-
-    assert history_item.text != None
-
-    # Check that item matches
-    # assert history_item.text == text
-    # client.history.delete(history_item.history_item_id)
-
-    # Test that the history item was deleted
-    # history = client.history.get_all(page_size=1).history
-    # assert len(history) == 0 or history[0].text != text

tests/test_client.py

@@ -0,0 +1,9 @@
+from elevenlabs import Model
+from elevenlabs.client import ElevenLabs
+
+
+def test_models():
+    client = ElevenLabs()
+    models = client.models.get_all()
+    assert len(models) > 0
+    assert isinstance(models[0], Model)

tests/test_history.py

@@ -0,0 +1,83 @@
+import asyncio
+
+from elevenlabs import VoiceSettings, play
+from elevenlabs.client import AsyncElevenLabs, ElevenLabs
+
+from .utils import IN_GITHUB, DEFAULT_TEXT, DEFAULT_VOICE, DEFAULT_MODEL
+import base64
+
+
+def test_tts_convert() -> None:
+    """Test basic text-to-speech generation."""
+    client = ElevenLabs()
+    audio_generator = client.text_to_speech.convert(text=DEFAULT_TEXT, voice_id=DEFAULT_VOICE, model_id=DEFAULT_MODEL)
+    audio = b"".join(audio_generator)
+    assert isinstance(audio, bytes), "TTS should return bytes"
+    if not IN_GITHUB:
+        play(audio)
+
+
+def test_tts_convert_with_voice_settings() -> None:
+    """Test TTS with custom voice settings."""
+    client = ElevenLabs()
+    audio_generator = client.text_to_speech.convert(
+        text=DEFAULT_TEXT,
+        voice_id=DEFAULT_VOICE,
+        model_id=DEFAULT_MODEL,
+        voice_settings=VoiceSettings(stability=0.71, similarity_boost=0.5, style=0.0, use_speaker_boost=True),
+    )
+    audio = b"".join(audio_generator)
+    assert isinstance(audio, bytes), "TTS with voice settings should return bytes"
+    if not IN_GITHUB:
+        play(audio)
+
+
+def test_tts_convert_as_stream():
+    async def main():
+        async_client = AsyncElevenLabs()
+        results = async_client.text_to_speech.convert_as_stream(
+            text=DEFAULT_TEXT, voice_id=DEFAULT_VOICE, model_id=DEFAULT_MODEL
+        )
+        out = b""
+        async for value in results:
+            assert isinstance(value, bytes), "Stream chunks should be bytes"
+            out += value
+        if not IN_GITHUB:
+            play(out)
+
+    asyncio.run(main())
+
+
+def test_tts_convert_with_timestamps() -> None:
+    """Test TTS generation with timestamps."""
+    client = ElevenLabs()
+    result = client.text_to_speech.convert_with_timestamps(
+        text=DEFAULT_TEXT, voice_id=DEFAULT_VOICE, model_id=DEFAULT_MODEL
+    )
+
+    assert "alignment" in result  # type: ignore
+    assert "characters" in result["alignment"]  # type: ignore
+
+    if not IN_GITHUB:
+        audio_bytes = base64.b64decode(result["audio_base64"])  # type: ignore
+        play(audio_bytes)
+
+
+def test_tts_stream_with_timestamps():
+    async def main():
+        async_client = AsyncElevenLabs()
+        audio_data = b""
+        async_stream = async_client.text_to_speech.stream_with_timestamps(
+            voice_id=DEFAULT_VOICE,
+            text=DEFAULT_TEXT,
+            model_id=DEFAULT_MODEL,
+        )
+        async for chunk in async_stream:
+            if hasattr(chunk, "audio_base_64"):
+                audio_bytes = base64.b64decode(chunk.audio_base_64)
+                audio_data += audio_bytes
+
+        if not IN_GITHUB:
+            play(audio_data)
+
+    asyncio.run(main())

tests/test_model.py

@@ -0,0 +1,24 @@
+from elevenlabs import Voice, VoiceSettings, ElevenLabs
+from .utils import DEFAULT_VOICE
+
+
+def test_get_voice():
+    client = ElevenLabs()
+    voice_id = DEFAULT_VOICE
+
+    voice = client.voices.get(voice_id)
+    assert isinstance(voice, Voice)
+
+    assert voice.voice_id == voice_id
+    if voice.settings is not None:
+        assert isinstance(voice.settings, VoiceSettings)
+
+
+def test_get_voices():
+    client = ElevenLabs()
+    response = client.voices.get_all()
+
+    assert len(response.voices) > 0
+
+    for voice in response.voices:
+        assert isinstance(voice, Voice)

tests/test_models.py

@@ -3,14 +3,12 @@
 import httpx
 
 from typing import Sequence, Generator
-from elevenlabs.client import ElevenLabs, \
-    AsyncElevenLabs
 
 IN_GITHUB = "GITHUB_ACTIONS" in os.environ
 
-client = ElevenLabs()
-
-async_client = AsyncElevenLabs()
+DEFAULT_VOICE = "21m00Tcm4TlvDq8ikWAM"
+DEFAULT_TEXT = "Hello"
+DEFAULT_MODEL = "eleven_multilingual_v2"
 
 
 def as_local_files(urls: Sequence[str]) -> Generator[str, None, None]:
@@ -25,4 +23,4 @@ def as_local_files(urls: Sequence[str]) -> Generator[str, None, None]:
         yield temp_file.name
     # Remove the files
     for temp_file in temp_files:
-        temp_file.close()
+        temp_file.close()