yuzu/spleeter
1
0
Fork 0
mirror of https://github.com/YuzuZensai/spleeter.git synced 2026-01-06 04:32:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: test parameters

Browse Source
This commit is contained in:
Félix Voituret
2019-11-08 20:03:16 -05:00
parent 05571be661
commit e9f1ba9450
1 changed files with 23 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

44
tests/test_separator.py
View File

@@ -10,37 +10,39 @@ __license__ = 'MIT License'
from os.path import exists, join
from tempfile import TemporaryDirectory
import pytest
from spleeter.audio.adapter import get_default_audio_adapter
from spleeter.separator import Separator
TEST_AUDIO_DESCRIPTOR = 'audio_example.mp3'
TEST_CONFIGURATIONS = {
'spleeter:2stems': ('vocals', 'accompaniment'),
'spleeter:4stems': ('vocals', 'drums', 'bass', 'other'),
'spleeter:5stems': ('vocals', 'drums', 'bass', 'piano', 'other')
}
TEST_CONFIGURATIONS = [
('spleeter:2stems', ('vocals', 'accompaniment')),
('spleeter:4stems', ('vocals', 'drums', 'bass', 'other')),
('spleeter:5stems', ('vocals', 'drums', 'bass', 'piano', 'other'))
]
@pytest.mark.parametrize('configuration, instruments', TEST_CONFIGURATIONS)
def test_separate():
""" Test separation from raw data. """
adapter = get_default_audio_adapter()
waveform, _ = adapter.load(TEST_AUDIO_DESCRIPTOR)
for configuration, instruments in TEST_CONFIGURATIONS.items():
separator = Separator(configuration)
prediction = separator.separate(waveform)
assert len(prediction) == 2
for instrument in instruments:
assert instrument in prediction
separator = Separator(configuration)
prediction = separator.separate(waveform)
assert len(prediction) == 2
for instrument in instruments:
assert instrument in prediction
def test_separate_to_file():
@pytest.mark.parametrize('configuration, instruments', TEST_CONFIGURATIONS)
def test_separate_to_file(configuration, instruments):
""" Test file based separation. """
for configuration, instruments in TEST_CONFIGURATIONS.items():
separator = Separator(configuration)
with TemporaryDirectory() as directory:
separator.separate_to_file(
TEST_AUDIO_DESCRIPTOR,
directory)
for instrument in instruments:
assert exists(join(directory, '{}.wav'.format(instrument)))
# TODO: Consider testing generated file as well.
separator = Separator(configuration)
with TemporaryDirectory() as directory:
separator.separate_to_file(
TEST_AUDIO_DESCRIPTOR,
directory)
for instrument in instruments:
assert exists(join(directory, '{}.wav'.format(instrument)))
# TODO: Consider testing generated file as well.