def test_is_nested_or_struct(): struct_ex = pa.struct([pa.field('a', pa.int32()), pa.field('b', pa.int8()), pa.field('c', pa.string())]) assert types.is_struct(struct_ex) assert not types.is_struct(pa.list_(pa.int32())) assert types.is_nested(struct_ex) assert types.is_nested(pa.list_(pa.int32())) assert not types.is_nested(pa.int32())
def is_complex(arrow_type: DataType, /) -> bool: """Check if data type is complex.""" return is_dictionary(arrow_type) or is_nested(arrow_type)
def is_complex(arrow_type): return is_dictionary(arrow_type) or is_nested(arrow_type)