@@ -271,11 +271,6 @@ def test_summaries(spark: SparkSession, session_catalog: Catalog, arrow_table_wi
271271
272272@pytest .mark .integration
273273def test_summaries_partial_overwrite (spark : SparkSession , session_catalog : Catalog ) -> None :
274- import pyarrow
275- from packaging import version
276-
277- under_20_arrow = version .parse (pyarrow .__version__ ) < version .parse ("20.0.0" )
278-
279274 identifier = "default.test_summaries_partial_overwrite"
280275 TEST_DATA = {
281276 "id" : [1 , 2 , 3 , 1 , 1 ],
@@ -314,15 +309,17 @@ def test_summaries_partial_overwrite(spark: SparkSession, session_catalog: Catal
314309 assert file_size > 0
315310
316311 # APPEND
312+ assert "added-files-size" in summaries [0 ]
313+ assert "total-files-size" in summaries [0 ]
317314 assert summaries [0 ] == {
318315 "added-data-files" : "3" ,
319- "added-files-size" : "2570" if under_20_arrow else "2618" ,
316+ "added-files-size" : summaries [ 0 ][ "added-files-size" ] ,
320317 "added-records" : "5" ,
321318 "changed-partition-count" : "3" ,
322319 "total-data-files" : "3" ,
323320 "total-delete-files" : "0" ,
324321 "total-equality-deletes" : "0" ,
325- "total-files-size" : "2570" if under_20_arrow else "2618" ,
322+ "total-files-size" : summaries [ 0 ][ "total-files-size" ] ,
326323 "total-position-deletes" : "0" ,
327324 "total-records" : "5" ,
328325 }
@@ -349,18 +346,21 @@ def test_summaries_partial_overwrite(spark: SparkSession, session_catalog: Catal
349346 # }
350347 files = tbl .inspect .data_files ()
351348 assert len (files ) == 3
349+ assert "added-files-size" in summaries [1 ]
350+ assert "removed-files-size" in summaries [1 ]
351+ assert "total-files-size" in summaries [1 ]
352352 assert summaries [1 ] == {
353353 "added-data-files" : "1" ,
354- "added-files-size" : "859" if under_20_arrow else "875" ,
354+ "added-files-size" : summaries [ 1 ][ "added-files-size" ] ,
355355 "added-records" : "2" ,
356356 "changed-partition-count" : "1" ,
357357 "deleted-data-files" : "1" ,
358358 "deleted-records" : "3" ,
359- "removed-files-size" : "866" if under_20_arrow else "882" ,
359+ "removed-files-size" : summaries [ 1 ][ "removed-files-size" ] ,
360360 "total-data-files" : "3" ,
361361 "total-delete-files" : "0" ,
362362 "total-equality-deletes" : "0" ,
363- "total-files-size" : "2563" if under_20_arrow else "2611" ,
363+ "total-files-size" : summaries [ 1 ][ "total-files-size" ] ,
364364 "total-position-deletes" : "0" ,
365365 "total-records" : "4" ,
366366 }
0 commit comments