From 4b40685f9dbb9edd1baa65e0e9d3a243069d4119 Mon Sep 17 00:00:00 2001 From: mmoussallam Date: Thu, 18 Jun 2020 18:09:34 +0200 Subject: [PATCH] remove useless prints --- tests/test_eval.py | 4 +--- tests/test_separator.py | 2 -- 2 files changed, 1 insertion(+), 5 deletions(-) diff --git a/tests/test_eval.py b/tests/test_eval.py index e9bf762..fcd37a6 100644 --- a/tests/test_eval.py +++ b/tests/test_eval.py @@ -106,7 +106,5 @@ def test_evaluate(path="FAKE_MUSDB_DIR"): params = load_configuration(arguments.configuration) metrics = evaluate.entrypoint(arguments, params) for instrument, metric in metrics.items(): - print(instrument), print(metric) 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) \ No newline at end of file diff --git a/tests/test_separator.py b/tests/test_separator.py index 06f5676..f2a3f6d 100644 --- a/tests/test_separator.py +++ b/tests/test_separator.py @@ -73,8 +73,6 @@ def test_separator_backends(test_file): # test that both outputs are not null assert np.sum(np.abs(out_tf[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) @pytest.mark.parametrize('test_file, configuration, backend', TEST_CONFIGURATIONS)