summaryrefslogtreecommitdiffstats
path: root/eval/src/tests/tensor/dense_tensor_builder/dense_tensor_builder_test.cpp
diff options
context:
space:
mode:
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.cpp261
1 files changed, 0 insertions, 261 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
deleted file mode 100644
index a0aeb6b63c9..00000000000
--- a/eval/src/tests/tensor/dense_tensor_builder/dense_tensor_builder_test.cpp
+++ /dev/null
@@ -1,261 +0,0 @@
-// Copyright 2017 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
-
-#include <vespa/vespalib/test/insertion_operators.h>
-#include <vespa/vespalib/testkit/test_kit.h>
-#include <vespa/eval/tensor/dense/dense_tensor_builder.h>
-#include <vespa/vespalib/util/exceptions.h>
-
-using namespace vespalib::tensor;
-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,
- const DenseTensor::Cells &expCells,
- const Tensor &tensor)
-{
- const DenseTensor &realTensor = dynamic_cast<const DenseTensor &>(tensor);
- EXPECT_EQUAL(ValueType::tensor_type(expDims), realTensor.type());
- EXPECT_EQUAL(expCells, make_vector(realTensor.cellsRef()));
-}
-
-void
-assertTensorSpec(const TensorSpec &expSpec, const Tensor &tensor)
-{
- TensorSpec actSpec = tensor.toSpec();
- EXPECT_EQUAL(expSpec, actSpec);
-}
-
-struct Fixture
-{
- Builder builder;
- Fixture() : builder() {}
-};
-
-Tensor::UP
-build1DTensor(Builder &builder)
-{
- Builder::Dimension dimX = builder.defineDimension("x", 3);
- builder.addLabel(dimX, 0).addCell(10).
- addLabel(dimX, 1).addCell(11).
- addLabel(dimX, 2).addCell(12);
- return builder.build();
-}
-
-TEST_F("require that 1d tensor can be constructed", Fixture)
-{
- assertTensor({{"x",3}}, {10,11,12}, *build1DTensor(f.builder));
-}
-
-TEST_F("require that 1d tensor can be converted to tensor spec", Fixture)
-{
- assertTensorSpec(TensorSpec("tensor(x[3])").
- add({{"x", 0}}, 10).
- add({{"x", 1}}, 11).
- add({{"x", 2}}, 12),
- *build1DTensor(f.builder));
-}
-
-Tensor::UP
-build2DTensor(Builder &builder)
-{
- Builder::Dimension dimX = builder.defineDimension("x", 3);
- Builder::Dimension dimY = builder.defineDimension("y", 2);
- builder.addLabel(dimX, 0).addLabel(dimY, 0).addCell(10).
- addLabel(dimX, 0).addLabel(dimY, 1).addCell(11).
- addLabel(dimX, 1).addLabel(dimY, 0).addCell(12).
- addLabel(dimX, 1).addLabel(dimY, 1).addCell(13).
- addLabel(dimX, 2).addLabel(dimY, 0).addCell(14).
- addLabel(dimX, 2).addLabel(dimY, 1).addCell(15);
- return builder.build();
-}
-
-TEST_F("require that 2d tensor can be constructed", Fixture)
-{
- assertTensor({{"x",3},{"y",2}}, {10,11,12,13,14,15}, *build2DTensor(f.builder));
-}
-
-TEST_F("require that 2d tensor can be converted to tensor spec", Fixture)
-{
- assertTensorSpec(TensorSpec("tensor(x[3],y[2])").
- add({{"x", 0},{"y", 0}}, 10).
- add({{"x", 0},{"y", 1}}, 11).
- add({{"x", 1},{"y", 0}}, 12).
- add({{"x", 1},{"y", 1}}, 13).
- add({{"x", 2},{"y", 0}}, 14).
- add({{"x", 2},{"y", 1}}, 15),
- *build2DTensor(f.builder));
-}
-
-TEST_F("require that 3d tensor can be constructed", Fixture)
-{
- Builder::Dimension dimX = f.builder.defineDimension("x", 3);
- Builder::Dimension dimY = f.builder.defineDimension("y", 2);
- Builder::Dimension dimZ = f.builder.defineDimension("z", 2);
- f.builder.addLabel(dimX, 0).addLabel(dimY, 0).addLabel(dimZ, 0).addCell(10).
- addLabel(dimX, 0).addLabel(dimY, 0).addLabel(dimZ, 1).addCell(11).
- addLabel(dimX, 0).addLabel(dimY, 1).addLabel(dimZ, 0).addCell(12).
- addLabel(dimX, 0).addLabel(dimY, 1).addLabel(dimZ, 1).addCell(13).
- addLabel(dimX, 1).addLabel(dimY, 0).addLabel(dimZ, 0).addCell(14).
- addLabel(dimX, 1).addLabel(dimY, 0).addLabel(dimZ, 1).addCell(15).
- addLabel(dimX, 1).addLabel(dimY, 1).addLabel(dimZ, 0).addCell(16).
- addLabel(dimX, 1).addLabel(dimY, 1).addLabel(dimZ, 1).addCell(17).
- addLabel(dimX, 2).addLabel(dimY, 0).addLabel(dimZ, 0).addCell(18).
- addLabel(dimX, 2).addLabel(dimY, 0).addLabel(dimZ, 1).addCell(19).
- addLabel(dimX, 2).addLabel(dimY, 1).addLabel(dimZ, 0).addCell(20).
- addLabel(dimX, 2).addLabel(dimY, 1).addLabel(dimZ, 1).addCell(21);
- assertTensor({{"x",3},{"y",2},{"z",2}},
- {10,11,12,13,14,15,16,17,18,19,20,21},
- *f.builder.build());
-}
-
-TEST_F("require that cells get default value 0 if not specified", Fixture)
-{
- Builder::Dimension dimX = f.builder.defineDimension("x", 3);
- f.builder.addLabel(dimX, 1).addCell(11);
- assertTensor({{"x",3}}, {0,11,0},
- *f.builder.build());
-}
-
-TEST_F("require that labels can be added in arbitrarily order", Fixture)
-{
- Builder::Dimension dimX = f.builder.defineDimension("x", 2);
- Builder::Dimension dimY = f.builder.defineDimension("y", 3);
- f.builder.addLabel(dimY, 0).addLabel(dimX, 1).addCell(10);
- assertTensor({{"x",2},{"y",3}}, {0,0,0,10,0,0},
- *f.builder.build());
-}
-
-TEST_F("require that builder can be re-used", Fixture)
-{
- {
- Builder::Dimension dimX = f.builder.defineDimension("x", 2);
- f.builder.addLabel(dimX, 0).addCell(10).
- addLabel(dimX, 1).addCell(11);
- assertTensor({{"x",2}}, {10,11},
- *f.builder.build());
- }
- {
- Builder::Dimension dimY = f.builder.defineDimension("y", 3);
- f.builder.addLabel(dimY, 0).addCell(20).
- addLabel(dimY, 1).addCell(21).
- addLabel(dimY, 2).addCell(22);
- assertTensor({{"y",3}}, {20,21,22},
- *f.builder.build());
- }
-}
-
-void
-assertTensorCell(const DenseTensor::Address &expAddress,
- double expCell,
- const DenseTensor::CellsIterator &itr)
-{
- EXPECT_TRUE(itr.valid());
- EXPECT_EQUAL(expAddress, itr.address());
- EXPECT_EQUAL(expCell, itr.cell());
-}
-
-TEST_F("require that dense tensor cells iterator works for 1d tensor", Fixture)
-{
- Tensor::UP tensor;
- {
- Builder::Dimension dimX = f.builder.defineDimension("x", 2);
- f.builder.addLabel(dimX, 0).addCell(2).
- addLabel(dimX, 1).addCell(3);
- tensor = f.builder.build();
- }
-
- const DenseTensor &denseTensor = dynamic_cast<const DenseTensor &>(*tensor);
- DenseTensor::CellsIterator itr = denseTensor.cellsIterator();
-
- assertTensorCell({0}, 2, itr);
- itr.next();
- assertTensorCell({1}, 3, itr);
- itr.next();
- EXPECT_FALSE(itr.valid());
-}
-
-TEST_F("require that dense tensor cells iterator works for 2d tensor", Fixture)
-{
- Tensor::UP tensor;
- {
- Builder::Dimension dimX = f.builder.defineDimension("x", 2);
- Builder::Dimension dimY = f.builder.defineDimension("y", 2);
- f.builder.addLabel(dimX, 0).addLabel(dimY, 0).addCell(2).
- addLabel(dimX, 0).addLabel(dimY, 1).addCell(3).
- addLabel(dimX, 1).addLabel(dimY, 0).addCell(5).
- addLabel(dimX, 1).addLabel(dimY, 1).addCell(7);
- tensor = f.builder.build();
- }
-
- const DenseTensor &denseTensor = dynamic_cast<const DenseTensor &>(*tensor);
- DenseTensor::CellsIterator itr = denseTensor.cellsIterator();
-
- assertTensorCell({0,0}, 2, itr);
- itr.next();
- assertTensorCell({0,1}, 3, itr);
- itr.next();
- assertTensorCell({1,0}, 5, itr);
- itr.next();
- assertTensorCell({1,1}, 7, itr);
- itr.next();
- EXPECT_FALSE(itr.valid());
-}
-
-TEST_F("require that undefined label for a dimension throws exception", Fixture)
-{
- Builder::Dimension dimX = f.builder.defineDimension("x", 2);
- f.builder.defineDimension("y", 3);
- EXPECT_EXCEPTION(f.builder.addLabel(dimX, 0).addCell(10),
- IllegalArgumentException,
- "Label for dimension 'y' is undefined. Expected a value in the range [0, 3>");
-}
-
-TEST_F("require that label outside range throws exception", Fixture)
-{
- Builder::Dimension dimX = f.builder.defineDimension("x", 2);
- EXPECT_EXCEPTION(f.builder.addLabel(dimX, 2).addCell(10),
- IllegalArgumentException,
- "Label '2' for dimension 'x' is outside range [0, 2>");
-}
-
-TEST_F("require that already specified label throws exception", Fixture)
-{
- Builder::Dimension dimX = f.builder.defineDimension("x", 2);
- EXPECT_EXCEPTION(f.builder.addLabel(dimX, 0).addLabel(dimX, 1).addCell(10),
- IllegalArgumentException,
- "Label for dimension 'x' is already specified with value '0'");
-}
-
-TEST_F("require that dimensions are sorted", Fixture)
-{
- Builder::Dimension dimY = f.builder.defineDimension("y", 3);
- Builder::Dimension dimX = f.builder.defineDimension("x", 5);
- f.builder.addLabel(dimX, 0).addLabel(dimY, 0).addCell(10);
- f.builder.addLabel(dimX, 0).addLabel(dimY, 1).addCell(11);
- f.builder.addLabel(dimX, 1).addLabel(dimY, 0).addCell(12);
- std::unique_ptr<Tensor> tensor = f.builder.build();
- const DenseTensor &denseTensor = dynamic_cast<const DenseTensor &>(*tensor);
- assertTensor({{"x", 5}, {"y", 3}},
- {10, 11, 0, 12, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0},
- denseTensor);
- EXPECT_EQUAL("tensor(x[5],y[3])", denseTensor.type().to_spec());
-}
-
-
-
-
-
-
-TEST_MAIN() { TEST_RUN_ALL(); }