@@ -129,8 +129,7 @@ def _inspect_files_asserts(df: pa.Table, spark_df: DataFrame) -> None:
129129 "record_count" ,
130130 "file_size_in_bytes" ,
131131 "split_offsets" ,
132- # Fixed in https://github.com/apache/iceberg-rust/pull/1705
133- # "equality_ids",
132+ "equality_ids" ,
134133 "sort_order_id" ,
135134 ]
136135 ]
@@ -143,19 +142,14 @@ def _inspect_files_asserts(df: pa.Table, spark_df: DataFrame) -> None:
143142 "record_count" ,
144143 "file_size_in_bytes" ,
145144 "split_offsets" ,
146- # Fixed in https://github.com/apache/iceberg-rust/pull/1705
147- # "equality_ids",
145+ "equality_ids" ,
148146 "sort_order_id" ,
149147 ]
150148 ]
151149
152150 assert_frame_equal (lhs_subset , rhs_subset , check_dtype = False , check_categorical = False )
153151
154152 for column in df .column_names :
155- if column == "equality_ids" :
156- # Fixed in https://github.com/apache/iceberg-rust/pull/1705
157- continue
158-
159153 if column == "partition" :
160154 # Spark leaves out the partition if the table is unpartitioned
161155 continue
@@ -363,10 +357,6 @@ def check_pyiceberg_df_equals_spark_df(df: pa.Table, spark_df: DataFrame) -> Non
363357 # Arrow turns dicts into lists of tuple
364358 df_lhs = dict (df_lhs )
365359
366- if "equality_ids" == df_column :
367- # Fixed in https://github.com/apache/iceberg-rust/pull/1705
368- continue
369-
370360 assert df_lhs == df_rhs , f"Difference in data_file column { df_column } : { df_lhs } != { df_rhs } "
371361 elif column == "readable_metrics" :
372362 assert list (left .keys ()) == [
@@ -887,7 +877,7 @@ def test_inspect_history(spark: SparkSession, session_catalog: Catalog, format_v
887877 if isinstance (left , float ) and math .isnan (left ) and isinstance (right , float ) and math .isnan (right ):
888878 # NaN != NaN in Python
889879 continue
890- # assert left == right, f"Difference in column {column}: {left} != {right}"
880+ assert left == right , f"Difference in column { column } : { left } != { right } "
891881
892882
893883@pytest .mark .integration
@@ -1094,7 +1084,6 @@ def test_inspect_all_files(
10941084
10951085
10961086@pytest .mark .integration
1097- @pytest .mark .skip ("Fixed in https://github.com/apache/iceberg-rust/pull/1682/" )
10981087def test_inspect_files_format_version_3 (spark : SparkSession , session_catalog : Catalog , arrow_table_with_null : pa .Table ) -> None :
10991088 identifier = "default.table_metadata_files"
11001089
@@ -1140,9 +1129,7 @@ def test_inspect_files_format_version_3(spark: SparkSession, session_catalog: Ca
11401129
11411130
11421131@pytest .mark .integration
1143- # @pytest.mark.parametrize("format_version", [1, 2, 3])
1144- # V3 support in https://github.com/apache/iceberg-rust/pull/1682/
1145- @pytest .mark .parametrize ("format_version" , [1 , 2 ])
1132+ @pytest .mark .parametrize ("format_version" , [1 , 2 , 3 ])
11461133def test_inspect_files_partitioned (spark : SparkSession , session_catalog : Catalog , format_version : int ) -> None :
11471134 from pandas .testing import assert_frame_equal
11481135
0 commit comments