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

fix tests and more testing

Browse Source
This commit is contained in:
mmoussallam
2020-06-19 00:48:09 +02:00
parent 97458816c9
commit bba1341bdf
2 changed files with 6 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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)
for instrument in out_lib.keys():
# test that both outputs are not null
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]))
# test that both outputs are close everywhere
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)