Tweak old huggingface hub tests

EducationalTestingService · Feb 12, 2024 · 3bdd3e7 · 3bdd3e7
1 parent 85c9f9c
commit 3bdd3e7
Showing 1 changed file with 3 additions and 9 deletions.
diff --git a/tests/common/file_utils_test.py b/tests/common/file_utils_test.py
@@ -505,17 +505,11 @@ def test_snapshot_download(self):
     def test_cached_download_no_user_or_org(self):
         path = cached_path("hf://t5-small/config.json", cache_dir=self.TEST_DIR)
         assert os.path.isfile(path)
-        assert pathlib.Path(os.path.dirname(path)) == self.TEST_DIR
-        assert os.path.isfile(path + ".json")
-        meta = _Meta.from_path(path + ".json")
-        assert meta.etag is not None
-        assert meta.resource == "hf://t5-small/config.json"
+        assert self.TEST_DIR in pathlib.Path(path).parents
 
     def test_snapshot_download_no_user_or_org(self):
         # This is the smallest snapshot I could find that is not associated with a user / org.
         model_name = "distilbert-base-german-cased"
-        path = cached_path(f"hf://{model_name}")
+        path = cached_path(f"hf://{model_name}", cache_dir=self.TEST_DIR)
         assert os.path.isdir(path)
-        assert os.path.isfile(path + ".json")
-        meta = _Meta.from_path(path + ".json")
-        assert meta.resource == f"hf://{model_name}"
+        assert self.TEST_DIR in pathlib.Path(path).parents