Function TEST

test/api/arrow_test.cpp:15–55

}

TEST(ArrowQueryResultTest, exportsCSRMetadataAsZeroCopyArrowArrays) {
    ArrowQueryResult::CSRMetadata metadata;
    metadata.indptr = {0, 2, 3};
    metadata.indices = {4, 5, 6};
    metadata.edgeIDs = {10, 11, 12};
    metadata.hasEdgeIDs = true;
    std::vector<ArrowQueryResult::CSRMetadata> csrChunks;
    csrChunks.push_back(std::move(metadata));

    ArrowQueryResult result{{}, 8, std::move(csrChunks)};
    const auto& storedMetadata = result.getCSRMetadata();
    auto csrArrays = result.getCSRArrowArrays();

    ASSERT_STREQ(csrArrays.indptr.schema.format, "l");
    ASSERT_STREQ(csrArrays.indices.schema.format, "l");
    ASSERT_TRUE(csrArrays.edgeIDs.has_value());
    ASSERT_STREQ(csrArrays.edgeIDs->schema.format, "l");
    ASSERT_EQ(csrArrays.indptr.array.length, static_cast<int64_t>(storedMetadata.indptr.size()));
    ASSERT_EQ(csrArrays.indices.array.length, static_cast<int64_t>(storedMetadata.indices.size()));
    ASSERT_EQ(csrArrays.edgeIDs->array.length, static_cast<int64_t>(storedMetadata.edgeIDs.size()));
    // No validity bitmap (the array is non-nullable).
    ASSERT_EQ(csrArrays.indptr.array.buffers[0], nullptr);
    ASSERT_EQ(csrArrays.indices.array.buffers[0], nullptr);
    ASSERT_EQ(csrArrays.edgeIDs->array.buffers[0], nullptr);
    // Value-equivalence: the exported Arrow data buffers hold the merged
    // CSR values. We check the contents rather than the backing pointer,
    // so the test does not break if the merge path changes whether it moves
    // or copies the per-batch vectors.
    const auto* indptrData = static_cast<const int64_t*>(csrArrays.indptr.array.buffers[1]);
    const auto* indicesData = static_cast<const int64_t*>(csrArrays.indices.array.buffers[1]);
    const auto* edgeIDsData = static_cast<const int64_t*>(csrArrays.edgeIDs->array.buffers[1]);
    ASSERT_NE(indptrData, nullptr);
    ASSERT_NE(indicesData, nullptr);
    ASSERT_NE(edgeIDsData, nullptr);
    ASSERT_EQ(std::vector<int64_t>(indptrData, indptrData + storedMetadata.indptr.size()),
        (std::vector<int64_t>{0, 2, 3}));
    ASSERT_EQ(std::vector<int64_t>(indicesData, indicesData + storedMetadata.indices.size()),
        (std::vector<int64_t>{4, 5, 6}));
    ASSERT_EQ(std::vector<int64_t>(edgeIDsData, edgeIDsData + storedMetadata.edgeIDs.size()),
        (std::vector<int64_t>{10, 11, 12}));
}

TEST_F(ArrowTest, resultToArrow) {
    auto query = "MATCH (a:person) WHERE a.fName = 'Bob' RETURN a.fName";