| 227 | } |
| 228 | |
| 229 | void createCsrRelTable(main::Connection& connection, const std::string& tableName = "csr_knows", |
| 230 | const char* weightName = "weight") { |
| 231 | std::vector<ArrowArrayWrapper> indices; |
| 232 | indices.push_back(makeCsrEdgeBatch( |
| 233 | {{1, 10, "ab", DATE_2020_01_01, {1}}, {2, 20, "ac", DATE_2021_01_01, {1, 2}}, |
| 234 | {2, 30, "bc", DATE_2022_01_01, {1}}, {3, 40, "gd", DATE_2023_01_01, {2}}})); |
| 235 | std::vector<ArrowArrayWrapper> indptr; |
| 236 | indptr.push_back(createStructArray(6, |
| 237 | {[](ArrowArray* array) { createUint64Array(array, {0, 2, 3, 4, 4, 4}); }})); |
| 238 | auto result = ArrowTableSupport::createRelTableFromArrowCSR(connection, tableName, "csr_node", |
| 239 | "csr_node", makeCsrIndexSchema(weightName), std::move(indices), makeIndptrSchema(), |
| 240 | std::move(indptr)); |
| 241 | ASSERT_TRUE(result.queryResult->isSuccess()) << result.queryResult->getErrorMessage(); |
| 242 | } |
| 243 | |
| 244 | } // namespace |
| 245 |
no test coverage detected