test: add error test

This commit is contained in:
Félix Voituret
2019-11-20 14:36:41 +01:00
parent 7b5222a64a
commit bfb0225083

View File

@@ -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: