1 files changed, 86 insertions, 0 deletions
diff --git a/src/python/test/test_remote_datasets.py b/src/python/test/test_remote_datasets.py
new file mode 100644
index 00000000..56a273b4
--- /dev/null
+++ b/src/python/test/test_remote_datasets.py
@@ -0,0 +1,86 @@
+# This file is part of the Gudhi Library - https://gudhi.inria.fr/ - which is released under MIT.
+# See file LICENSE or go to https://gudhi.inria.fr/licensing/ for full license details.
+# Author(s):       Hind Montassif
+#
+# Copyright (C) 2021 Inria
+#
+# Modification(s):
+#   - YYYY/MM Author: Description of the modification
+
+
+from gudhi.datasets import remote
+import re
+from os.path import isfile, exists
+from os import makedirs
+import io
+import sys
+import pytest
+
+def _check_dir_file_names(path_file_dw, filename, dirname):
+    assert isfile(path_file_dw)
+
+    names_dw = re.split(r' |/|\\', path_file_dw)
+    # Case where inner directories are created in "remote_datasets/"; e.g: "remote_datasets/bunny"
+    if len(names_dw) >= 3:
+        for i in range(len(names_dw)-1):
+            assert re.split(r' |/|\\', dirname)[i] == names_dw[i]
+        assert filename == names_dw[i+1]
+    else:
+        assert dirname == names_dw[0]
+        assert filename == names_dw[1]
+
+def _check_fetch_output(url, filename, dirname = "remote_datasets", file_checksum = None):
+    if not exists(dirname):
+        makedirs(dirname)
+    path_file_dw = remote._fetch_remote(url, filename, dirname, file_checksum)
+    _check_dir_file_names(path_file_dw, filename, dirname)
+
+def _get_bunny_license_print(accept_license = False):
+    capturedOutput = io.StringIO()
+    # Redirect stdout
+    sys.stdout = capturedOutput
+
+    if not exists("remote_datasets/bunny"):
+        makedirs("remote_datasets/bunny")
+    remote._fetch_remote("https://raw.githubusercontent.com/GUDHI/gudhi-data/main/points/bunny/bunny.npy", "bunny.npy", "remote_datasets/bunny",
+                 '13f7842ebb4b45370e50641ff28c88685703efa5faab14edf0bb7d113a965e1b', accept_license)
+    # Reset redirect
+    sys.stdout = sys.__stdout__
+    return capturedOutput
+
+def test_fetch_remote_datasets():
+    # Test fetch with a wrong checksum
+    with pytest.raises(OSError):
+        _check_fetch_output("https://raw.githubusercontent.com/GUDHI/gudhi-data/main/points/spiral_2d.csv", "spiral_2d.csv", file_checksum = 'XXXXXXXXXX')
+
+    # Test files download from given urls with checksums provided
+    _check_fetch_output("https://raw.githubusercontent.com/GUDHI/gudhi-data/main/points/spiral_2d.csv", "spiral_2d.csv",
+                                file_checksum = '37530355d980d957c4ec06b18c775f90a91e446107d06c6201c9b4000b077f38')
+
+    _check_fetch_output("https://raw.githubusercontent.com/GUDHI/gudhi-data/main/points/sphere3D_pts_on_grid.off", "sphere3D_pts_on_grid.off",
+                                file_checksum = '32f96d2cafb1177f0dd5e0a019b6ff5658e14a619a7815ae55ad0fc5e8bd3f88')
+
+    # Test files download from given urls without checksums
+    _check_fetch_output("https://raw.githubusercontent.com/GUDHI/gudhi-data/main/points/spiral_2d.csv", "spiral_2d.csv")
+
+    _check_fetch_output("https://raw.githubusercontent.com/GUDHI/gudhi-data/main/points/sphere3D_pts_on_grid.off", "sphere3D_pts_on_grid.off")
+
+    # Test spiral_2d.csv wrapping function
+    spiral_2d_arr = remote.fetch_spiral_2d()
+    assert spiral_2d_arr.shape == (114562, 2)
+
+    # Test printing existing LICENSE file when fetching bunny.npy with accept_license = False (default)
+    # Fetch LICENSE file
+    if not exists("remote_datasets/bunny"):
+        makedirs("remote_datasets/bunny")
+    remote._fetch_remote("https://raw.githubusercontent.com/GUDHI/gudhi-data/main/points/bunny/LICENSE", "LICENSE", "remote_datasets/bunny",
+                 'aeb1bad319b7d74fa0b8076358182f9c6b1284c67cc07dc67cbc9bc73025d956')
+    with open("remote_datasets/bunny/LICENSE") as f:
+        assert f.read() == _get_bunny_license_print().getvalue().rstrip("\n")
+
+    # Test not printing bunny.npy LICENSE when accept_license = True
+    assert "" == _get_bunny_license_print(accept_license = True).getvalue()
+
+    # Test fetch_bunny wrapping function
+    bunny_arr = remote.fetch_bunny()
+    assert bunny_arr.shape == (35947, 3)