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

rebase

Browse Source
This commit is contained in:
mmoussallam
2020-07-24 15:39:43 +02:00
parent bba1341bdf
commit f98f16ed70
1 changed files with 11 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
tests/test_eval.py
View File

@@ -99,14 +99,14 @@ def generate_fake_eval_dataset(path):
aa.save(filename, data, fs) aa.save(filename, data, fs)
def test_evaluate(path="FAKE_MUSDB_DIR"): @pytest.mark.parametrize('backend', TEST_CONFIGURATIONS)
generate_fake_eval_dataset(path) def test_evaluate(backend):
p = create_argument_parser() with TemporaryDirectory() as directory:
arguments = p.parse_args(["evaluate", "-p", "spleeter:4stems", "--mus_dir", path]) generate_fake_eval_dataset(directory)
params = load_configuration(arguments.configuration) p = create_argument_parser()
metrics = evaluate.entrypoint(arguments, params) arguments = p.parse_args(["evaluate", "-p", "spleeter:4stems", "--mus_dir", directory, "-B", backend])
for instrument, metric in metrics.items(): params = load_configuration(arguments.configuration)
print(instrument), print(metric) metrics = evaluate.entrypoint(arguments, params)
for m, value in metric.items(): for instrument, metric in metrics.items():
print(np.median(value)), print(res_4stems[instrument][m]) for m, value in metric.items():
assert np.allclose(np.median(value), res_4stems[instrument][m], atol=1e-3) assert np.allclose(np.median(value), res_4stems[backend][instrument][m], atol=1e-3)