Function test_cast_array_to_features_array_xd

tests/test_table.py:1192–1199

()

def test_cast_array_to_features_array_xd():
    # same storage type
    arr = pa.array([[[0, 1], [2, 3]], [[4, 5], [6, 7]]], pa.list_(pa.list_(pa.int32(), 2), 2))
    casted_array = cast_array_to_feature(arr, Array2D(shape=(2, 2), dtype="int32"))
    assert casted_array.type == Array2DExtensionType(shape=(2, 2), dtype="int32")
    # different storage type
    casted_array = cast_array_to_feature(arr, Array2D(shape=(2, 2), dtype="float32"))
    assert casted_array.type == Array2DExtensionType(shape=(2, 2), dtype="float32")


def test_cast_array_to_features_sequence_classlabel():