update

Author: deependujha

tests/streaming/test_fs_provider.py

@@ -1,5 +1,8 @@
+from unittest.mock import Mock
+
 import pytest
 
+from litdata.streaming import fs_provider as fs_provider_module
 from litdata.streaming.fs_provider import GCPFsProvider, S3FsProvider, _get_fs_provider, get_bucket_and_path
 
 
@@ -17,7 +20,11 @@ def test_get_bucket_and_path():
     assert path == "path/to/file.txt"
 
 
-def test_get_fs_provider():
+def test_get_fs_provider(monkeypatch, google_mock):
+    google_mock.cloud.storage.Client = Mock()
+    monkeypatch.setattr(fs_provider_module, "_GOOGLE_STORAGE_AVAILABLE", True)
+    monkeypatch.setattr(fs_provider_module, "S3Client", Mock())
+
     fs_provider = _get_fs_provider("s3://bucket/path/to/file.txt")
     assert isinstance(fs_provider, S3FsProvider)