Merge branch 'copilot/add-seekable-option-in-kwargs' into copilot/update-seekable-test-cases

Author: Copilot

fsspec/implementations/arrow.py

@@ -205,11 +205,11 @@ def modified(self, path):
         return self.fs.get_file_info(path).mtime
 
     def cat_file(self, path, start=None, end=None, **kwargs):
-        kwargs["seekable"] = start not in [None, 0]
+        kwargs.setdefault("seekable", start not in [None, 0])
         return super().cat_file(path, start=None, end=None, **kwargs)
 
     def get_file(self, rpath, lpath, **kwargs):
-        kwargs["seekable"] = False
+        kwargs.setdefault("seekable", False)
         super().get_file(rpath, lpath, **kwargs)
 
 

fsspec/implementations/tests/test_arrow.py

@@ -268,3 +268,53 @@ def test_get_kwargs_from_urls_hadoop_fs():
     assert kwargs["host"] == "localhost"
     assert kwargs["port"] == 8020
     assert "replication" not in kwargs
+
+
+def test_get_file_seekable_default(fs, remote_dir, tmp_path):
+    """Test that get_file defaults to seekable=False but allows override."""
+    data = b"test data for seekable"
+
+    # Create a test file
+    with fs.open(remote_dir + "/test_file.txt", "wb") as f:
+        f.write(data)
+
+    # Test default behavior (seekable=False)
+    local_file = tmp_path / "test_default.txt"
+    fs.get_file(remote_dir + "/test_file.txt", str(local_file))
+    with open(local_file, "rb") as f:
+        assert f.read() == data
+
+    # Test with explicit seekable=True
+    local_file_seekable = tmp_path / "test_seekable.txt"
+    fs.get_file(remote_dir + "/test_file.txt", str(local_file_seekable), seekable=True)
+    with open(local_file_seekable, "rb") as f:
+        assert f.read() == data
+
+    # Test with explicit seekable=False
+    local_file_not_seekable = tmp_path / "test_not_seekable.txt"
+    fs.get_file(
+        remote_dir + "/test_file.txt", str(local_file_not_seekable), seekable=False
+    )
+    with open(local_file_not_seekable, "rb") as f:
+        assert f.read() == data
+
+
+def test_cat_file_seekable_override(fs, remote_dir):
+    """Test that cat_file allows seekable to be overridden."""
+    data = b"test data for cat_file seekable"
+
+    # Create a test file
+    with fs.open(remote_dir + "/test_cat.txt", "wb") as f:
+        f.write(data)
+
+    # Test default behavior - when start is None, seekable should default to False
+    result = fs.cat_file(remote_dir + "/test_cat.txt")
+    assert result == data
+
+    # Test with explicit seekable=True even when start is None
+    result = fs.cat_file(remote_dir + "/test_cat.txt", seekable=True)
+    assert result == data
+
+    # Test with explicit seekable=False
+    result = fs.cat_file(remote_dir + "/test_cat.txt", seekable=False)
+    assert result == data