From 2572e7045b86614ebda726ec9d29ccf8ad694a47 Mon Sep 17 00:00:00 2001 From: pedroramonedafranco Date: Mon, 30 Nov 2020 20:18:31 +0100 Subject: [PATCH] fix bug --- ...cousticbrainz_genre_dataset_little_test.json | 0 tests/test_utils.py | 17 +++++++++++------ 2 files changed, 11 insertions(+), 6 deletions(-) rename tests/{indexes => resources/download}/acousticbrainz_genre_dataset_little_test.json (100%) diff --git a/tests/indexes/acousticbrainz_genre_dataset_little_test.json b/tests/resources/download/acousticbrainz_genre_dataset_little_test.json similarity index 100% rename from tests/indexes/acousticbrainz_genre_dataset_little_test.json rename to tests/resources/download/acousticbrainz_genre_dataset_little_test.json diff --git a/tests/test_utils.py b/tests/test_utils.py index 11570dbba..9a150c642 100644 --- a/tests/test_utils.py +++ b/tests/test_utils.py @@ -85,19 +85,24 @@ def mock_check_index(mocker): return mocker.patch.object(utils, "check_index") -def test_remote_index(): +def test_remote_index(httpserver): + httpserver.serve_content( + open("tests/resources/download/acousticbrainz_genre_dataset_little_test.json", "rb").read() + ) REMOTE_INDEX = { "remote_index": download_utils.RemoteFileMetadata( filename="acousticbrainz_genre_dataset_little_test.json", - url="https://zenodo.org/record/4274551/files/acousticbrainz_genre_dataset_little_test.json?download=1", + url=httpserver.url, checksum="7f256c49438022ab493c88f5a1b43e88", # the md5 checksum destination_dir=".", # relative path for where to unzip the data, or None - ), + ) } DATA = LargeData("acousticbrainz_genre_dataset_little_test.json", remote_index=REMOTE_INDEX) - with open("tests/indexes/acousticbrainz_genre_dataset_little_test.json") as f: - little_index = json.load(f) - assert DATA.index == little_index['tracks'] + ind = DATA.index + assert len(ind["tracks"]) == 16 + os.remove("mirdata/datasets/indexes/acousticbrainz_genre_dataset_little_test.json") + + def test_md5(mocker):