Nice

Author: Fokko

tests/test_transforms.py

@@ -1654,38 +1654,6 @@ def test_bucket_pyarrow_transforms(
     assert expected == transform.pyarrow_transform(source_type)(input_arr)
 
 
-# pyiceberg_core currently does not support bucket transform on timestamp_ns and timestamptz_ns
-# https://github.com/apache/iceberg-rust/issues/1110
-@pytest.mark.parametrize(
-    "source_type, input_arr, num_buckets",
-    [
-        (
-            TimestampNanoType(),
-            pa.array([datetime(1970, 1, 1, 0, 0, 0), datetime(2025, 2, 26, 1, 2, 3)], type=pa.timestamp(unit="ns")),
-            10,
-        ),
-        (
-            TimestamptzNanoType(),
-            pa.array(
-                [datetime(1970, 1, 1, 0, 0, 0), datetime(2025, 2, 26, 1, 2, 3)],
-                type=pa.timestamp(unit="ns", tz=pytz.timezone("Etc/GMT+10")),
-            ),
-            10,
-        ),
-    ],
-)
-def test_unsupported_bucket_pyarrow_transform(
-    source_type: PrimitiveType,
-    input_arr: Union[pa.Array, pa.ChunkedArray],
-    num_buckets: int,
-) -> None:
-    transform: Transform[Any, Any] = BucketTransform(num_buckets=num_buckets)
-    with pytest.raises(ValueError) as exc_info:
-        transform.pyarrow_transform(source_type)(input_arr)
-
-    assert "FeatureUnsupported => Unsupported data type for bucket transform" in str(exc_info.value)
-
-
 @pytest.mark.parametrize(
     "source_type, input_arr, expected, width",
     [