Skip to content

Commit 3f211f9

Browse files
committed
Fix moar tests
1 parent ecf029d commit 3f211f9

File tree

2 files changed

+15
-17
lines changed

2 files changed

+15
-17
lines changed

tests/integration/test_deletes.py

Lines changed: 9 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -467,21 +467,19 @@ def test_partitioned_table_positional_deletes_sequence_number(spark: SparkSessio
467467
assert snapshots[2].summary == Summary(
468468
Operation.OVERWRITE,
469469
**{
470-
"added-files-size": snapshots[2].summary["total-files-size"],
471470
"added-data-files": "1",
471+
"added-files-size": snapshots[2].summary["added-files-size"],
472472
"added-records": "2",
473473
"changed-partition-count": "1",
474-
"total-files-size": snapshots[2].summary["total-files-size"],
475-
"total-delete-files": "0",
476-
"total-data-files": "1",
477-
"total-position-deletes": "0",
478-
"total-records": "2",
479-
"total-equality-deletes": "0",
480-
"deleted-data-files": "2",
481-
"removed-delete-files": "1",
482-
"deleted-records": "5",
474+
"deleted-data-files": "1",
475+
"deleted-records": "3",
483476
"removed-files-size": snapshots[2].summary["removed-files-size"],
484-
"removed-position-deletes": "1",
477+
"total-data-files": "2",
478+
"total-delete-files": "1",
479+
"total-equality-deletes": "0",
480+
"total-files-size": snapshots[2].summary["total-files-size"],
481+
"total-position-deletes": "1",
482+
"total-records": "4",
485483
},
486484
)
487485

tests/integration/test_writes/test_writes.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -271,8 +271,8 @@ def test_summaries_partial_overwrite(spark: SparkSession, session_catalog: Catal
271271
}
272272
pa_schema = pa.schema(
273273
[
274-
pa.field("id", pa.int32(), pa.int32()),
275-
pa.field("name", pa.int32(), pa.string()),
274+
pa.field("id", pa.int32()),
275+
pa.field("name", pa.string()),
276276
]
277277
)
278278
arrow_table = pa.Table.from_pydict(TEST_DATA, schema=pa_schema)
@@ -305,13 +305,13 @@ def test_summaries_partial_overwrite(spark: SparkSession, session_catalog: Catal
305305
# APPEND
306306
assert summaries[0] == {
307307
"added-data-files": "3",
308-
"added-files-size": "2848",
308+
"added-files-size": "2570",
309309
"added-records": "5",
310310
"changed-partition-count": "3",
311311
"total-data-files": "3",
312312
"total-delete-files": "0",
313313
"total-equality-deletes": "0",
314-
"total-files-size": "2848",
314+
"total-files-size": "2570",
315315
"total-position-deletes": "0",
316316
"total-records": "5",
317317
}
@@ -345,11 +345,11 @@ def test_summaries_partial_overwrite(spark: SparkSession, session_catalog: Catal
345345
"changed-partition-count": "1",
346346
"deleted-data-files": "1",
347347
"deleted-records": "3",
348-
"removed-files-size": "950",
348+
"removed-files-size": "866",
349349
"total-data-files": "3",
350350
"total-delete-files": "0",
351351
"total-equality-deletes": "0",
352-
"total-files-size": "2757",
352+
"total-files-size": "2563",
353353
"total-position-deletes": "0",
354354
"total-records": "4",
355355
}

0 commit comments

Comments
 (0)