wingkitlee0 commented on code in PR #47350:
URL: https://github.com/apache/arrow/pull/47350#discussion_r2296255684


##########
python/pyarrow/tests/test_fs.py:
##########
@@ -1201,14 +1201,17 @@ def test_s3_options(pickle_module):
     assert isinstance(fs, S3FileSystem)
     assert pickle_module.loads(pickle_module.dumps(fs)) == fs
 
-    # Note that the retry strategy won't survive pickling for now
     fs = S3FileSystem(
         retry_strategy=AwsStandardS3RetryStrategy(max_attempts=5))
     assert isinstance(fs, S3FileSystem)
+    assert pickle_module.loads(pickle_module.dumps(fs)) == fs
+    assert fs.retry_strategy.max_attempts == 5
 
     fs = S3FileSystem(
         retry_strategy=AwsDefaultS3RetryStrategy(max_attempts=5))
     assert isinstance(fs, S3FileSystem)
+    assert pickle_module.loads(pickle_module.dumps(fs)) == fs

Review Comment:
   Thanks for the suggestion. I will look into it. It seems like Cython will 
call cpp side of things, e.g., `Equals` of FileSystem. I think I will need to 
add AwsRetryStrategy::Equals as well.



-- 
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

To unsubscribe, e-mail: github-unsubscr...@arrow.apache.org

For queries about this service, please contact Infrastructure at:
us...@infra.apache.org

Reply via email to