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

remove useless prints

Browse Source
This commit is contained in:
mmoussallam
2020-06-18 18:09:34 +02:00
parent 27f52b71d5
commit 4b40685f9d
2 changed files with 1 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/test_eval.py
View File

@@ -106,7 +106,5 @@ def test_evaluate(path="FAKE_MUSDB_DIR"):
params = load_configuration(arguments.configuration) params = load_configuration(arguments.configuration)
metrics = evaluate.entrypoint(arguments, params) metrics = evaluate.entrypoint(arguments, params)
for instrument, metric in metrics.items(): for instrument, metric in metrics.items():
print(instrument), print(metric)
for m, value in metric.items(): for m, value in metric.items():
print(np.median(value)), print(res_4stems[instrument][m]) assert np.allclose(np.median(value), res_4stems[instrument][m], atol=1e-3)
assert np.allclose(np.median(value), res_4stems[instrument][m], atol=1e-3)

2
tests/test_separator.py
View File

@@ -73,8 +73,6 @@ def test_separator_backends(test_file):
# test that both outputs are not null # test that both outputs are not null
assert np.sum(np.abs(out_tf[instrument])) > 1000 assert np.sum(np.abs(out_tf[instrument])) > 1000
assert np.sum(np.abs(out_lib[instrument])) > 1000 assert np.sum(np.abs(out_lib[instrument])) > 1000
max_diff = np.max(np.abs(out_tf[instrument] - out_lib[instrument]))
print(f"Max diff on {instrument} is {max_diff}")
assert np.allclose(out_tf[instrument], out_lib[instrument], atol=0.1) assert np.allclose(out_tf[instrument], out_lib[instrument], atol=0.1)
@pytest.mark.parametrize('test_file, configuration, backend', TEST_CONFIGURATIONS) @pytest.mark.parametrize('test_file, configuration, backend', TEST_CONFIGURATIONS)