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

fix tests and more testing

Browse Source
This commit is contained in:
mmoussallam
2020-06-19 00:48:09 +02:00
parent 9510e2b5f7
commit af6dbec5cf
2 changed files with 14 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
tests/test_eval.py
View File

@@ -27,27 +27,27 @@ from spleeter.utils.configuration import load_configuration
res_4stems = { "vocals": { res_4stems = { "vocals": {
"SDR": 0.000, "SDR": 0.000,
"SAR": -15.856, "SAR": -16.212,
"SIR": -6.861, "SIR": -4.172,
"ISR": 0.000 "ISR": 0.000
}, },
"drums": { "drums": {
"SDR": -0.069, "SDR": -0.077,
"SAR": -15.769, "SAR": -15.739,
"SIR": -5.049, "SIR": -5.045,
"ISR": 0.001 "ISR": 0.001
}, },
"bass":{ "bass":{
"SDR": -0.000, "SDR": -0.000,
"SAR": -10.499, "SAR": -10.665,
"SIR": -6.988, "SIR": -5.646,
"ISR": -0.000 "ISR": -0.000
}, },
"other":{ "other":{
"SDR": -1.365, "SDR": -1.309,
"SAR": -14.591, "SAR": -14.573,
"SIR": -4.751, "SIR": -4.705,
"ISR": -0.016 "ISR": -0.014
} }
} }

9
tests/test_separator.py
View File

@@ -72,13 +72,10 @@ def test_separator_backends(test_file):
out_lib = separator_lib._separate_librosa(waveform, test_file) out_lib = separator_lib._separate_librosa(waveform, test_file)
for instrument in out_lib.keys(): for instrument in out_lib.keys():
# test that both outputs are not null # test that both outputs are close everywhere
print(np.sum(np.abs(out_tf[instrument])))
print(np.sum(np.abs(out_lib[instrument])))
assert np.sum(np.abs(out_tf[instrument])) > 1000
assert np.sum(np.abs(out_lib[instrument])) > 1000
print(np.max(out_tf[instrument]- out_lib[instrument]))
assert np.allclose(out_tf[instrument], out_lib[instrument], atol=0.025) assert np.allclose(out_tf[instrument], out_lib[instrument], atol=0.025)
# it should be even more similar outside edges zones
assert np.allclose(out_tf[instrument][4096:-4096,:], out_lib[instrument][4096:-4096,:], atol=0.002)
@pytest.mark.parametrize('test_file, configuration, backend', TEST_CONFIGURATIONS) @pytest.mark.parametrize('test_file, configuration, backend', TEST_CONFIGURATIONS)