aboutsummaryrefslogtreecommitdiffstats
path: root/searchlib/src/tests/tensor/hnsw_index/hnsw_index_test.cpp
diff options
context:
space:
mode:
authorTor Egge <Tor.Egge@online.no>2022-11-22 11:20:36 +0100
committerTor Egge <Tor.Egge@online.no>2022-11-22 11:20:36 +0100
commit25d902a8d776ff6e67ae15b188cbc96ca57c88e4 (patch)
tree5c1e4e52ff97658afa797a1571eef378dccd6d13 /searchlib/src/tests/tensor/hnsw_index/hnsw_index_test.cpp
parent3e6deec22291d2e4304be1e17547e9e147084340 (diff)
Move shared test code to test library.
Diffstat (limited to 'searchlib/src/tests/tensor/hnsw_index/hnsw_index_test.cpp')
-rw-r--r--searchlib/src/tests/tensor/hnsw_index/hnsw_index_test.cpp38
1 files changed, 4 insertions, 34 deletions
diff --git a/searchlib/src/tests/tensor/hnsw_index/hnsw_index_test.cpp b/searchlib/src/tests/tensor/hnsw_index/hnsw_index_test.cpp
index b86913caa16..13caad8b6d6 100644
--- a/searchlib/src/tests/tensor/hnsw_index/hnsw_index_test.cpp
+++ b/searchlib/src/tests/tensor/hnsw_index/hnsw_index_test.cpp
@@ -2,6 +2,8 @@
#include <vespa/eval/eval/value_type.h>
#include <vespa/searchlib/common/bitvector.h>
+#include <vespa/searchlib/test/vector_buffer_reader.h>
+#include <vespa/searchlib/test/vector_buffer_writer.h>
#include <vespa/searchlib/tensor/distance_functions.h>
#include <vespa/searchlib/tensor/doc_vector_access.h>
#include <vespa/searchlib/tensor/hnsw_index.h>
@@ -11,7 +13,6 @@
#include <vespa/searchlib/tensor/inv_log_level_generator.h>
#include <vespa/searchlib/tensor/subspace_type.h>
#include <vespa/searchlib/tensor/vector_bundle.h>
-#include <vespa/searchlib/util/bufferwriter.h>
#include <vespa/searchlib/queryeval/global_filter.h>
#include <vespa/vespalib/datastore/compaction_spec.h>
#include <vespa/vespalib/datastore/compaction_strategy.h>
@@ -36,39 +37,8 @@ using vespalib::eval::ValueType;
using vespalib::datastore::CompactionSpec;
using vespalib::datastore::CompactionStrategy;
using search::queryeval::GlobalFilter;
-
-class VectorBufferWriter : public BufferWriter {
-private:
- char tmp[1024];
-public:
- std::vector<char> output;
- VectorBufferWriter() {
- setup(tmp, 1024);
- }
- ~VectorBufferWriter() {}
- void flush() override {
- for (size_t i = 0; i < usedLen(); ++i) {
- output.push_back(tmp[i]);
- }
- rewind();
- }
-};
-
-class VectorBufferReader {
-private:
- const std::vector<char>& _data;
- size_t _pos;
-
-public:
- VectorBufferReader(const std::vector<char>& data) : _data(data), _pos(0) {}
- uint32_t readHostOrder() {
- uint32_t result = 0;
- assert(_pos + sizeof(uint32_t) <= _data.size());
- std::memcpy(&result, _data.data() + _pos, sizeof(uint32_t));
- _pos += sizeof(uint32_t);
- return result;
- }
-};
+using search::test::VectorBufferReader;
+using search::test::VectorBufferWriter;
template <typename FloatType>
class MyDocVectorAccess : public DocVectorAccess {