diff options
author | Hind-M <hind.montassif@gmail.com> | 2022-02-07 13:03:45 +0100 |
---|---|---|
committer | Hind-M <hind.montassif@gmail.com> | 2022-02-07 13:03:45 +0100 |
commit | b5d7d6c2857d305ba2828065310c11edefb37c4e (patch) | |
tree | fd3162f4b82de4428c7d873dbb41d9595d89c2cb /src/python/test | |
parent | b0071de9ee7b6b4feb2eb9f19ceb759de21c997f (diff) |
Test get_data_home and clear_data_home on a separate folder
Diffstat (limited to 'src/python/test')
-rw-r--r-- | src/python/test/test_remote_datasets.py | 21 |
1 files changed, 10 insertions, 11 deletions
diff --git a/src/python/test/test_remote_datasets.py b/src/python/test/test_remote_datasets.py index 9532b4ec..c160f270 100644 --- a/src/python/test/test_remote_datasets.py +++ b/src/python/test/test_remote_datasets.py @@ -87,21 +87,20 @@ def test_fetch_remote_datasets(): assert bunny_arr.shape == (35947, 3) # Check that default dir was created - assert isdir(expanduser("~/remote_datasets")) == True - - # Test clear_data_home - remote.clear_data_home() - assert isdir(expanduser("~/remote_datasets")) == False + assert isdir(expanduser("~/remote_datasets")) # Test fetch_spiral_2d and fetch_bunny wrapping functions with data directory different from default - spiral_2d_arr = remote.fetch_spiral_2d(dirname = "~/test") + spiral_2d_arr = remote.fetch_spiral_2d(dirname = "~/another_fetch_folder") assert spiral_2d_arr.shape == (114562, 2) - bunny_arr = remote.fetch_bunny(dirname = "~/test") + bunny_arr = remote.fetch_bunny(dirname = "~/another_fetch_folder") assert bunny_arr.shape == (35947, 3) - assert isdir(expanduser("~/test")) == True + assert isdir(expanduser("~/another_fetch_folder")) + + # Test get_data_home and clear_data_home on new empty folder + empty_data_home = remote.get_data_home(data_home="empty_folder") + assert isdir(empty_data_home) - # Test clear_data_home with data directory different from default - remote.clear_data_home("~/test") - assert isdir(expanduser("~/test")) == False + remote.clear_data_home(data_home=empty_data_home) + assert not isdir(empty_data_home) |