update tests

Author: lhoestq

tests/test_info_utils.py

@@ -7,7 +7,7 @@
 @pytest.fixture(params=[None, 0, 100 * 2 ** 20, 900 * 2 ** 20])
 def env_max_in_memory_dataset_size(request, monkeypatch):
     if request.param:
-        monkeypatch.setenv("MAX_IN_MEMORY_DATASET_SIZE_IN_BYTES", request.param)
+        monkeypatch.setenv("HF_MAX_IN_MEMORY_DATASET_SIZE_IN_BYTES", request.param)
 
 
 @pytest.mark.parametrize("dataset_size", [None, 400 * 2 ** 20, 600 * 2 ** 20])
@@ -16,9 +16,9 @@ def test_is_small_dataset(
     dataset_size, config_max_in_memory_dataset_size, env_max_in_memory_dataset_size, monkeypatch
 ):
     if config_max_in_memory_dataset_size != "default":
-        monkeypatch.setattr(datasets.config, "MAX_IN_MEMORY_DATASET_SIZE_IN_BYTES", config_max_in_memory_dataset_size)
+        monkeypatch.setattr(datasets.config, "HF_MAX_IN_MEMORY_DATASET_SIZE_IN_BYTES", config_max_in_memory_dataset_size)
 
-    max_in_memory_dataset_size = datasets.config.MAX_IN_MEMORY_DATASET_SIZE_IN_BYTES
+    max_in_memory_dataset_size = datasets.config.HF_MAX_IN_MEMORY_DATASET_SIZE_IN_BYTES
     if config_max_in_memory_dataset_size == "default":
         if env_max_in_memory_dataset_size:
             assert max_in_memory_dataset_size == env_max_in_memory_dataset_size

tests/test_load.py

@@ -233,9 +233,9 @@ def test_load_dataset_local_with_default_in_memory(
     current_dataset_size = 148
     if max_in_memory_dataset_size == "default":
         # default = 250 * 2 ** 20
-        max_in_memory_dataset_size = datasets.config.MAX_IN_MEMORY_DATASET_SIZE_IN_BYTES
+        max_in_memory_dataset_size = datasets.config.HF_MAX_IN_MEMORY_DATASET_SIZE_IN_BYTES
     else:
-        monkeypatch.setattr(datasets.config, "MAX_IN_MEMORY_DATASET_SIZE_IN_BYTES", max_in_memory_dataset_size)
+        monkeypatch.setattr(datasets.config, "HF_MAX_IN_MEMORY_DATASET_SIZE_IN_BYTES", max_in_memory_dataset_size)
     if max_in_memory_dataset_size:
         expected_in_memory = current_dataset_size < max_in_memory_dataset_size
     else:
@@ -253,9 +253,9 @@ def test_load_from_disk_with_default_in_memory(
     current_dataset_size = 512  # arrow file size = 512, in-memory dataset size = 148
     if max_in_memory_dataset_size == "default":
         # default = 250 * 2 ** 20
-        max_in_memory_dataset_size = datasets.config.MAX_IN_MEMORY_DATASET_SIZE_IN_BYTES
+        max_in_memory_dataset_size = datasets.config.HF_MAX_IN_MEMORY_DATASET_SIZE_IN_BYTES
     else:
-        monkeypatch.setattr(datasets.config, "MAX_IN_MEMORY_DATASET_SIZE_IN_BYTES", max_in_memory_dataset_size)
+        monkeypatch.setattr(datasets.config, "HF_MAX_IN_MEMORY_DATASET_SIZE_IN_BYTES", max_in_memory_dataset_size)
     if max_in_memory_dataset_size:
         expected_in_memory = current_dataset_size < max_in_memory_dataset_size
     else: