From bfb02250835a2c7bbd8010686b32482fa58dedb7 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?F=C3=A9lix=20Voituret?= <fvoituret@deezer.com>
Date: Wed, 20 Nov 2019 14:36:41 +0100
Subject: [PATCH] test: add error test

---
 tests/test_ffmpeg_adapter.py | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/tests/test_ffmpeg_adapter.py b/tests/test_ffmpeg_adapter.py
index cd503c2..195b737 100644
--- a/tests/test_ffmpeg_adapter.py
+++ b/tests/test_ffmpeg_adapter.py
@@ -11,12 +11,13 @@ from os.path import join
 from tempfile import TemporaryDirectory
 
 # pylint: disable=import-error
-from pytest import fixture
+from pytest import fixture, raises
 
 import numpy as np
 import ffmpeg
 # pylint: enable=import-error
 
+from spleeter import SpleeterError
 from spleeter.audio.adapter import AudioAdapter
 from spleeter.audio.adapter import get_default_audio_adapter
 from spleeter.audio.adapter import get_audio_adapter
@@ -61,6 +62,16 @@ def test_load(audio_data):
     assert waveform.shape[1] == 2
 
 
+def test_load_error(adapter):
+    """ Test load ffprobe exception """
+    with raises(SpleeterError):
+        adapter.load(
+            'Paris City Jazz',
+            TEST_OFFSET,
+            TEST_DURATION,
+            TEST_SAMPLE_RATE)
+
+
 def test_save(adapter, audio_data):
     """ Test audio saving. """
     with TemporaryDirectory() as directory: