remove useless prints

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

View File

@@ -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)

View File

@@ -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)