Add test for multi-file download

nsidc · Dec 4, 2023 · 11bb43b · 11bb43b
1 parent 4de8af6
commit 11bb43b
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/tests/integration/test_cloud_download.py b/tests/integration/test_cloud_download.py
@@ -157,3 +157,13 @@ def test_earthaccess_can_download_cloud_collection_granules(daac):
                 f"Warning: {concept_id} downloaded size {total_mb_downloaded}MB is "
                 f"different from the size reported by CMR: {total_size_cmr}MB"
             )
+
+
+def test_multi_file_granule(tmp_path):
+    # Ensure granules that contain multiple files are handled correctly
+    granules = earthaccess.search_data(short_name="HLSL30", count=1)
+    assert len(granules) == 1
+    urls = granules[0].data_links()
+    assert len(urls) > 1
+    files = earthaccess.download(granules, str(tmp_path))
+    assert set(map(os.path.basename, urls)) == set(map(os.path.basename, files))