diff options
Diffstat (limited to 'eval/src/tests/tensor/dense_tensor_builder/dense_tensor_builder_test.cpp')
-rw-r--r-- | eval/src/tests/tensor/dense_tensor_builder/dense_tensor_builder_test.cpp | 11 |
1 files changed, 10 insertions, 1 deletions
diff --git a/eval/src/tests/tensor/dense_tensor_builder/dense_tensor_builder_test.cpp b/eval/src/tests/tensor/dense_tensor_builder/dense_tensor_builder_test.cpp index 61efdbe6d22..ae6166f9d24 100644 --- a/eval/src/tests/tensor/dense_tensor_builder/dense_tensor_builder_test.cpp +++ b/eval/src/tests/tensor/dense_tensor_builder/dense_tensor_builder_test.cpp @@ -10,6 +10,15 @@ using vespalib::IllegalArgumentException; using Builder = DenseTensorBuilder; using vespalib::eval::TensorSpec; using vespalib::eval::ValueType; +using vespalib::ConstArrayRef; + +template <typename T> std::vector<T> make_vector(const ConstArrayRef<T> &ref) { + std::vector<T> vec; + for (const T &t: ref) { + vec.push_back(t); + } + return vec; +} void assertTensor(const std::vector<ValueType::Dimension> &expDims, @@ -18,7 +27,7 @@ assertTensor(const std::vector<ValueType::Dimension> &expDims, { const DenseTensor &realTensor = dynamic_cast<const DenseTensor &>(tensor); EXPECT_EQUAL(ValueType::tensor_type(expDims), realTensor.type()); - EXPECT_EQUAL(expCells, realTensor.cells()); + EXPECT_EQUAL(expCells, make_vector(realTensor.cellsRef())); } void |