diff options
| author | Tor Egge <Tor.Egge@broadpark.no> | 2019-02-25 16:52:24 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2019-02-25 16:52:24 +0100 |
| commit | 4b46918b47774d15b4882aff7db693699383ca61 (patch) | |
| tree | d7c6498e7390fb8bc2bf8b3c6290c25c5be7fcb6 /eval/src/tests/tensor/tensor_remove_operation/tensor_remove_operation_test.cpp | |
| parent | 50e5898f70d4ea1ece5065b06ea7f3a0755463b9 (diff) | |
| parent | e3ab5b19197122709f06636001955e8c84345a0f (diff) | |
Merge pull request #8604 from vespa-engine/geirst/remove-and-modify-for-mixed-tensors
Geirst/remove and modify for mixed tensors
Diffstat (limited to 'eval/src/tests/tensor/tensor_remove_operation/tensor_remove_operation_test.cpp')
| -rw-r--r-- | eval/src/tests/tensor/tensor_remove_operation/tensor_remove_operation_test.cpp | 49 |
1 files changed, 49 insertions, 0 deletions
diff --git a/eval/src/tests/tensor/tensor_remove_operation/tensor_remove_operation_test.cpp b/eval/src/tests/tensor/tensor_remove_operation/tensor_remove_operation_test.cpp index cb28019c4ee..df21b46691d 100644 --- a/eval/src/tests/tensor/tensor_remove_operation/tensor_remove_operation_test.cpp +++ b/eval/src/tests/tensor/tensor_remove_operation/tensor_remove_operation_test.cpp @@ -43,4 +43,53 @@ TEST(TensorRemoveTest, all_cells_can_be_removed_from_a_sparse_tensor) TensorSpec("tensor(x{},y{})")); } +TEST(TensorRemoveTest, cells_can_be_removed_from_a_mixed_tensor) +{ + assertRemove(TensorSpec("tensor(x{},y[2])") + .add({{"x","a"},{"y",0}}, 2) + .add({{"x","a"},{"y",1}}, 3) + .add({{"x","b"},{"y",0}}, 4) + .add({{"x","b"},{"y",1}}, 5), + TensorSpec("tensor(x{})") + .add({{"x","b"}}, 1) + .add({{"x","c"}}, 1), + TensorSpec("tensor(x{},y[2])") + .add({{"x","a"},{"y",0}}, 2) + .add({{"x","a"},{"y",1}}, 3)); + + assertRemove(TensorSpec("tensor(x{},y{},z[2])") + .add({{"x","a"},{"y","c"},{"z",0}}, 2) + .add({{"x","a"},{"y","c"},{"z",1}}, 3) + .add({{"x","b"},{"y","c"},{"z",0}}, 4) + .add({{"x","b"},{"y","c"},{"z",1}}, 5), + TensorSpec("tensor(x{},y{})") + .add({{"x","b"},{"y","c"}}, 1) + .add({{"x","c"},{"y","c"}}, 1), + TensorSpec("tensor(x{},y{},z[2])") + .add({{"x","a"},{"y","c"},{"z",0}}, 2) + .add({{"x","a"},{"y","c"},{"z",1}}, 3)); + + assertRemove(TensorSpec("tensor(x{},y[1],z[2])") + .add({{"x","a"},{"y",0},{"z",0}}, 2) + .add({{"x","a"},{"y",0},{"z",1}}, 3) + .add({{"x","b"},{"y",0},{"z",0}}, 4) + .add({{"x","b"},{"y",0},{"z",1}}, 5), + TensorSpec("tensor(x{})") + .add({{"x","b"}}, 1) + .add({{"x","c"}}, 1), + TensorSpec("tensor(x{},y[1],z[2])") + .add({{"x","a"},{"y",0},{"z",0}}, 2) + .add({{"x","a"},{"y",0},{"z",1}}, 3)); +} + +TEST(TensorRemoveTest, all_cells_can_be_removed_from_a_mixed_tensor) +{ + assertRemove(TensorSpec("tensor(x{},y[2])") + .add({{"x","a"},{"y",0}}, 2) + .add({{"x","a"},{"y",1}}, 3), + TensorSpec("tensor(x{})") + .add({{"x","a"}}, 1), + TensorSpec("tensor(x{},y[2])")); +} + GTEST_MAIN_RUN_ALL_TESTS |