diff options
Diffstat (limited to 'searchlib/src/tests/tensor/tensor_buffer_store')
-rw-r--r-- | searchlib/src/tests/tensor/tensor_buffer_store/tensor_buffer_store_test.cpp | 13 |
1 files changed, 5 insertions, 8 deletions
diff --git a/searchlib/src/tests/tensor/tensor_buffer_store/tensor_buffer_store_test.cpp b/searchlib/src/tests/tensor/tensor_buffer_store/tensor_buffer_store_test.cpp index 05e40200167..ec7fc2334c4 100644 --- a/searchlib/src/tests/tensor/tensor_buffer_store/tensor_buffer_store_test.cpp +++ b/searchlib/src/tests/tensor/tensor_buffer_store/tensor_buffer_store_test.cpp @@ -162,22 +162,19 @@ TEST_F(TensorBufferStoreTest, stored_tensor_can_be_encoded_and_stored_as_encoded } } -TEST_F(TensorBufferStoreTest, get_typed_cells) +TEST_F(TensorBufferStoreTest, get_vectors) { auto ref = store_tensor(tensor_specs.back()); std::vector<double> values; + auto vectors = _store.get_vectors(ref); + EXPECT_EQ(4, vectors.subspaces()); for (uint32_t subspace = 0; subspace < 4; ++subspace) { - auto cells = _store.get_typed_cells(ref, subspace).typify<double>(); + auto cells = vectors.cells(subspace).typify<double>(); EXPECT_EQ(1, cells.size()); values.emplace_back(cells[0]); } EXPECT_EQ((std::vector<double>{4.5, 5.5, 6.5, 7.5}), values); - for (auto tref : { ref, EntryRef() }) { - auto subspace = tref.valid() ? 4 : 0; - auto cells = _store.get_typed_cells(tref, subspace).typify<double>(); - EXPECT_EQ(1, cells.size()); - EXPECT_EQ(0.0, cells[0]); - } + EXPECT_EQ(0, _store.get_vectors(EntryRef()).subspaces()); } GTEST_MAIN_RUN_ALL_TESTS() |