fixup

laughingman7743 · May 5, 2024 · 251f012 · 251f012
1 parent d37e0a8
commit 251f012
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/pyathena/filesystem/s3.py b/pyathena/filesystem/s3.py
@@ -723,6 +723,7 @@ def get_file(self, rpath: str, lpath: str, callback=_DEFAULT_CALLBACK, outfile=N
 
         with open(lpath, "wb") as local:
             with self.open(rpath, "rb", **kwargs) as remote:
+                callback.set_size(remote.size)
                 while data := remote.read(remote.blocksize):
                     local.write(data)
                     callback.relative_update(len(data))

diff --git a/tests/pyathena/filesystem/test_s3.py b/tests/pyathena/filesystem/test_s3.py
@@ -241,7 +241,7 @@ def test_info_dir(self, fs):
         info = fs.info(dir)
         ls_info = fs.ls(dir, detail=True)[0]
 
-        assert info == ls_info
+        assert info.to_dict() == ls_info.to_dict()
         assert info.name == fs._strip_protocol(dir)
         assert info.bucket == bucket
         assert info.key == key