Function TEST_F

test/api/arrow_table_function_test.cpp:17–66

};

TEST_F(ArrowTableFunctionTest, CreateArrowTable) {
    // Create simple test data
    std::vector<int32_t> intData = {1, 2, 3};

    // Create Arrow schema
    ArrowSchemaWrapper schema;
    createStructSchema(&schema, 1);
    createSchema<int32_t>(schema.children[0], "id");

    // Create Arrow array
    ArrowArray array;
    array.length = intData.size();
    array.null_count = 0;
    array.offset = 0;
    array.n_buffers = 1;
    array.n_children = 1;
    array.buffers = static_cast<const void**>(malloc(sizeof(void*)));
    array.buffers[0] = nullptr;
    array.children = static_cast<ArrowArray**>(malloc(sizeof(ArrowArray*)));
    array.children[0] = static_cast<ArrowArray*>(malloc(sizeof(ArrowArray)));
    createInt32Array(array.children[0], intData);
    array.dictionary = nullptr;
    array.release = [](ArrowArray* arr) {
        if (arr->children) {
            for (int64_t i = 0; i < arr->n_children; i++) {
                if (arr->children[i]->release) {
                    arr->children[i]->release(arr->children[i]);
                }
                free(arr->children[i]);
            }
            free(arr->children);
        }
        if (arr->buffers) {
            free(const_cast<void**>(arr->buffers));
        }
        arr->release = nullptr;
    };
    array.private_data = nullptr;

    // Test that we can create arrays with Arrow C Data Interface
    EXPECT_EQ(array.length, 3);
    EXPECT_EQ(array.n_children, 1);
    EXPECT_STREQ(schema.children[0]->name, "id");

    // Cleanup
    if (schema.release)
        schema.release(&schema);
    if (array.release)
        array.release(&array);
}

TEST_F(ArrowTableFunctionTest, CreateArrowTableMultipleColumns) {
    // Create test data with multiple columns