fix bug

mir-dataset-loaders · Nov 30, 2020 · 2572e70 · 2572e70
1 parent ee9948d
commit 2572e70
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 6 deletions.
diff --git a/...sticbrainz_genre_dataset_little_test.json → ...sticbrainz_genre_dataset_little_test.json b/...sticbrainz_genre_dataset_little_test.json → ...sticbrainz_genre_dataset_little_test.json
diff --git 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):