Fix CI mock filesystem fixtures

tests/fixtures/fsspec.py

@@ -2,9 +2,9 @@
 from pathlib import Path
 from unittest.mock import patch
 
-import fsspec
 import pytest
 from fsspec.implementations.local import AbstractFileSystem, LocalFileSystem, stringify_path
+from fsspec.registry import _registry as _fsspec_registry
 
 
 class MockFileSystem(AbstractFileSystem):
@@ -92,11 +92,11 @@ def _strip_protocol(cls, path):
 
 @pytest.fixture
 def mock_fsspec():
-    original_registry = fsspec.registry.copy()
-    fsspec.register_implementation("mock", MockFileSystem, clobber=True)
-    fsspec.register_implementation("tmp", TmpDirFileSystem, clobber=True)
+    _fsspec_registry["mock"] = MockFileSystem
+    _fsspec_registry["tmp"] = TmpDirFileSystem
     yield
-    fsspec.registry = original_registry
+    del _fsspec_registry["mock"]
+    del _fsspec_registry["tmp"]
 
 
 @pytest.fixture

tests/test_filesystem.py

@@ -2,12 +2,23 @@
 
 import fsspec
 import pytest
+from fsspec.registry import _registry as _fsspec_registry
 
 from datasets.filesystems import COMPRESSION_FILESYSTEMS, HfFileSystem, extract_path_from_uri, is_remote_filesystem
 
 from .utils import require_lz4, require_zstandard
 
 
+def test_mockfs(mockfs):
+    assert "mock" in _fsspec_registry
+    assert "bz2" in _fsspec_registry
+
+
+def test_non_mockfs():
+    assert "mock" not in _fsspec_registry
+    assert "bz2" in _fsspec_registry
+
+
 def test_extract_path_from_uri():
     mock_bucket = "mock-s3-bucket"
     dataset_path = f"s3://{mock_bucket}"

tests/test_streaming_download_manager.py

@@ -3,8 +3,8 @@
 import re
 from pathlib import Path
 
-import fsspec
 import pytest
+from fsspec.registry import _registry as _fsspec_registry
 from fsspec.spec import AbstractBufferedFile, AbstractFileSystem
 
 from datasets.download.streaming_download_manager import (
@@ -130,10 +130,9 @@ def _open(
 
 @pytest.fixture
 def mock_fsspec():
-    original_registry = fsspec.registry.copy()
-    fsspec.register_implementation("mock", DummyTestFS, clobber=True)
+    _fsspec_registry["mock"] = DummyTestFS
     yield
-    fsspec.registry = original_registry
+    del _fsspec_registry["mock"]
 
 
 def _readd_double_slash_removed_by_path(path_as_posix: str) -> str: