feat: add e2e tests

Author: louisjoecodes

.gitignore

@@ -3,3 +3,4 @@ dist/
 __pycache__/
 poetry.toml
 .ruff_cache/
+.DS_Store

tests/fixtures/voice_sample.mp3

@@ -0,0 +1,32 @@
+from elevenlabs import play
+from elevenlabs.client import ElevenLabs
+
+from .utils import IN_GITHUB, DEFAULT_VOICE_FILE
+
+
+def test_audio_isolation() -> None:
+    """Test basic audio isolation."""
+    client = ElevenLabs()
+    audio_file = open(DEFAULT_VOICE_FILE, "rb")
+    try:
+        audio_stream = client.audio_isolation.audio_isolation(audio=audio_file)
+        audio = b"".join(chunk for chunk in audio_stream)
+        assert isinstance(audio, bytes), "Combined audio should be bytes"
+        if not IN_GITHUB:
+            play(audio)
+    finally:
+        audio_file.close()
+
+
+def test_audio_isolation_as_stream():
+    """Test audio isolation with streaming."""
+    client = ElevenLabs()
+    audio_file = open(DEFAULT_VOICE_FILE, "rb")
+    try:
+        audio_stream = client.audio_isolation.audio_isolation_stream(audio=audio_file)
+        audio = b"".join(chunk for chunk in audio_stream)
+        assert isinstance(audio, bytes), "Combined audio should be bytes"
+        if not IN_GITHUB:
+            play(audio)
+    finally:
+        audio_file.close()

tests/test_audio_isolation.py

@@ -2,7 +2,7 @@
 from elevenlabs.client import ElevenLabs
 
 
-def test_models():
+def test_models_get_all():
     client = ElevenLabs()
     models = client.models.get_all()
     assert len(models) > 0

tests/test_models.py

@@ -0,0 +1,31 @@
+from elevenlabs import play
+from elevenlabs.client import ElevenLabs
+
+from .utils import IN_GITHUB, DEFAULT_VOICE, DEFAULT_VOICE_FILE
+
+
+def test_sts() -> None:
+    """Test basic speech-to-speech generation."""
+    client = ElevenLabs()
+    audio_file = open(DEFAULT_VOICE_FILE, "rb")
+    try:
+        audio_stream = client.speech_to_speech.convert(voice_id=DEFAULT_VOICE, audio=audio_file)
+        audio = b"".join(chunk for chunk in audio_stream)
+        assert isinstance(audio, bytes), "Combined audio should be bytes"
+        if not IN_GITHUB:
+            play(audio)
+    finally:
+        audio_file.close()
+
+
+def test_sts_as_stream():
+    client = ElevenLabs()
+    audio_file = open(DEFAULT_VOICE_FILE, "rb")
+    try:
+        audio_stream = client.speech_to_speech.convert_as_stream(voice_id=DEFAULT_VOICE, audio=audio_file)
+        audio = b"".join(chunk for chunk in audio_stream)
+        assert isinstance(audio, bytes), "Combined audio should be bytes"
+        if not IN_GITHUB:
+            play(audio)
+    finally:
+        audio_file.close()

tests/test_sts.py

@@ -0,0 +1,17 @@
+from elevenlabs import play
+from elevenlabs.client import ElevenLabs
+
+from .utils import IN_GITHUB
+
+
+def test_text_to_sound_effects_convert() -> None:
+    """Test basic sound-effect generation."""
+    client = ElevenLabs()
+    audio_generator = client.text_to_sound_effects.convert(
+        text="Hypnotic throbbing sound effect. Increases in imtensity.",
+        duration_seconds=2,
+    )
+    audio = b"".join(audio_generator)
+    assert isinstance(audio, bytes), "TTS should return bytes"
+    if not IN_GITHUB:
+        play(audio)

tests/test_ttsfx.py

@@ -0,0 +1,17 @@
+from elevenlabs.client import ElevenLabs
+
+
+def test_voice_preview_generation():
+    """Test generating voice previews from description."""
+    client = ElevenLabs()
+
+    # Test parameters
+    description = "A warm and friendly female voice with a slight British accent, speaking clearly and professionally"
+    sample_text = "This is a test message that needs to be at least one hundred characters long to meet the API requirements. Here it is."
+
+    previews = client.text_to_voice.create_previews(voice_description=description, text=sample_text)
+
+    assert hasattr(previews, "previews"), "Response should have 'previews' attribute"
+    assert len(previews.previews) > 0, "Should receive at least one preview"
+    assert hasattr(previews.previews[0], "generated_voice_id"), "Preview should contain generated_voice_id"
+    assert hasattr(previews.previews[0], "audio_base_64"), "Preview should contain audio_base_64"

tests/test_ttv.py

@@ -9,6 +9,7 @@
 DEFAULT_VOICE = "21m00Tcm4TlvDq8ikWAM"
 DEFAULT_TEXT = "Hello"
 DEFAULT_MODEL = "eleven_multilingual_v2"
+DEFAULT_VOICE_FILE = "tests/fixtures/voice_sample.mp3"
 
 
 def as_local_files(urls: Sequence[str]) -> Generator[str, None, None]: