Skip to content

Commit 2d5361f

Browse files
committed
test: add tests for executing UUIDs to pyarrow
1 parent 4707c44 commit 2d5361f

File tree

1 file changed

+28
-0
lines changed

1 file changed

+28
-0
lines changed

ibis/backends/tests/test_uuid.py

+28
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@
33
import contextlib
44
import uuid
55

6+
import pyarrow as pa
67
import pytest
78

89
import ibis
@@ -63,3 +64,30 @@ def test_uuid_unique_each_row(con):
6364
con.tables.functional_alltypes.mutate(uuid=ibis.uuid()).limit(2).uuid.nunique()
6465
)
6566
assert expr.execute() == 2
67+
68+
69+
@pytest.mark.notimpl(
70+
["druid", "exasol", "mysql", "oracle", "polars", "pyspark", "risingwave"],
71+
raises=com.OperationNotDefinedError,
72+
)
73+
@pytest.mark.broken(
74+
"impala",
75+
reason="Expected bytes, got a 'UUID' object. https://github.com/ibis-project/ibis/issues/8902",
76+
raises=pa.ArrowInvalid,
77+
)
78+
@pytest.mark.broken(
79+
[
80+
"clickhouse",
81+
"druid",
82+
"flink",
83+
"mysql",
84+
"postgres",
85+
"sqlite",
86+
"trino",
87+
],
88+
reason="Expected bytes, got a 'UUID' object. https://github.com/ibis-project/ibis/issues/8902",
89+
raises=pa.lib.ArrowTypeError,
90+
)
91+
@pytest.mark.notimpl(["pandas", "dask"], raises=com.OperationNotDefinedError)
92+
def test_uuid_to_pyarrow(con):
93+
con.to_pyarrow(ibis.uuid())

0 commit comments

Comments
 (0)