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 iterations

Browse Source
This commit is contained in:
Félix Voituret
2019-11-08 19:40:45 -05:00
parent ebd03c8565
commit 7322b3bdea
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/test_separator.py
View File

@@ -25,7 +25,7 @@ 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:
for configuration, instruments in TEST_CONFIGURATIONS.items():
separator = Separator(configuration)
prediction = separator.separate(waveform)
assert len(prediction) == 2
@@ -35,7 +35,7 @@ def test_separate():
def test_separate_to_file():
""" Test file based separation. """
for configuration, instruments in TEST_CONFIGURATIONS:
for configuration, instruments in TEST_CONFIGURATIONS.items():
separator = Separator(configuration)
with TemporaryDirectory() as directory:
separator.separate_to_file(