mirror of
https://github.com/YuzuZensai/spleeter.git
synced 2026-01-06 04:32:43 +00:00
test: add error test
This commit is contained in:
@@ -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:
|
||||
|
||||
Reference in New Issue
Block a user