MCPcopy Create free account
hub / github.com/apache/arrow / DoCheckTakeAAA

Function DoCheckTakeAAA

cpp/src/arrow/compute/kernels/vector_selection_test.cc:1202–1221  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

1200}
1201
1202void DoCheckTakeAAA(const std::shared_ptr<Array>& values,
1203 const std::shared_ptr<Array>& indices,
1204 const std::shared_ptr<Array>& expected) {
1205 DoAssertTakeAAA(values, indices, expected);
1206
1207 // Check sliced values
1208 ASSERT_OK_AND_ASSIGN(auto values_filler, MakeArrayOfNull(values->type(), 2));
1209 ASSERT_OK_AND_ASSIGN(auto values_sliced,
1210 Concatenate({values_filler, values, values_filler}));
1211 values_sliced = values_sliced->Slice(2, values->length());
1212 DoAssertTakeAAA(values_sliced, indices, expected);
1213
1214 // Check sliced indices
1215 ASSERT_OK_AND_ASSIGN(auto zero, MakeScalar(indices->type(), int8_t{0}));
1216 ASSERT_OK_AND_ASSIGN(auto indices_filler, MakeArrayFromScalar(*zero, 3));
1217 ASSERT_OK_AND_ASSIGN(auto indices_sliced,
1218 Concatenate({indices_filler, indices, indices_filler}));
1219 indices_sliced = indices_sliced->Slice(3, indices->length());
1220 DoAssertTakeAAA(values, indices_sliced, expected);
1221}
1222
1223void DoCheckTakeCACWithArrays(const std::shared_ptr<Array>& values,
1224 const std::shared_ptr<Array>& indices,

Callers 1

DoCheckTakeXAFunction · 0.85

Calls 3

DoAssertTakeAAAFunction · 0.85
SliceMethod · 0.45
lengthMethod · 0.45

Tested by

no test coverage detected