yuzu/spleeter
1
0
Fork 0
mirror of https://github.com/YuzuZensai/spleeter.git synced 2026-01-30 20:24:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

test: add error test

Browse Source
This commit is contained in:
Félix Voituret
2019-11-20 14:36:41 +01:00
parent 7b5222a64a
commit bfb0225083
1 changed files with 12 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
tests/test_ffmpeg_adapter.py
View File

@@ -11,12 +11,13 @@ from os.path import join
from tempfile import TemporaryDirectory from tempfile import TemporaryDirectory
# pylint: disable=import-error # pylint: disable=import-error
from pytest import fixture from pytest import fixture, raises
import numpy as np import numpy as np
import ffmpeg import ffmpeg
# pylint: enable=import-error # pylint: enable=import-error
from spleeter import SpleeterError
from spleeter.audio.adapter import AudioAdapter from spleeter.audio.adapter import AudioAdapter
from spleeter.audio.adapter import get_default_audio_adapter from spleeter.audio.adapter import get_default_audio_adapter
from spleeter.audio.adapter import get_audio_adapter from spleeter.audio.adapter import get_audio_adapter
@@ -61,6 +62,16 @@ def test_load(audio_data):
assert waveform.shape[1] == 2 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): def test_save(adapter, audio_data):
""" Test audio saving. """ """ Test audio saving. """
with TemporaryDirectory() as directory: with TemporaryDirectory() as directory: