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: