Skip to content

Commit c86eb8e

Browse files
fix: stop calling deprecated PuffinFile.to_vector in Spark DV test (#3804)
1 parent 03ad0d6 commit c86eb8e

1 file changed

Lines changed: 5 additions & 4 deletions

File tree

tests/integration/test_deletes.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,7 @@
2929
from pyiceberg.partitioning import PartitionField, PartitionSpec
3030
from pyiceberg.schema import Schema
3131
from pyiceberg.table import Table
32+
from pyiceberg.table.deletion_vector import deletion_vectors_from_puffin_file
3233
from pyiceberg.table.puffin import PuffinFile
3334
from pyiceberg.table.snapshots import Operation, Summary
3435
from pyiceberg.transforms import IdentityTransform
@@ -1081,10 +1082,10 @@ def test_read_spark_written_puffin_dv(spark: SparkSession, session_catalog: Rest
10811082
assert "referenced-data-file" in blob.properties
10821083
assert blob.properties["cardinality"] == "4"
10831084

1084-
dv_dict = puffin.to_vector()
1085-
assert len(dv_dict) == 1, "Expected one data file's deletions"
1085+
dvs = deletion_vectors_from_puffin_file(puffin)
1086+
assert len(dvs) == 1, "Expected one data file's deletions"
10861087

1087-
for _data_file_path, chunked_array in dv_dict.items():
1088-
positions = chunked_array.to_pylist()
1088+
for dv in dvs:
1089+
positions = dv.to_vector().to_pylist()
10891090
assert len(positions) == 4, f"Expected 4 deleted positions, got {len(positions)}"
10901091
assert sorted(positions) == [9, 19, 29, 39], f"Unexpected positions: {positions}"

0 commit comments

Comments
 (0)