diff options
author | Tor Egge <Tor.Egge@online.no> | 2022-11-22 11:20:36 +0100 |
---|---|---|
committer | Tor Egge <Tor.Egge@online.no> | 2022-11-22 11:20:36 +0100 |
commit | 25d902a8d776ff6e67ae15b188cbc96ca57c88e4 (patch) | |
tree | 5c1e4e52ff97658afa797a1571eef378dccd6d13 /searchlib/src/tests/tensor/hnsw_index/hnsw_index_test.cpp | |
parent | 3e6deec22291d2e4304be1e17547e9e147084340 (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.cpp | 38 |
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 { |