admin/milvus
1
0
Fork 0
mirror of https://gitee.com/milvus-io/milvus.git synced 2025-12-30 15:35:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'branch-0.3.0' into 'branch-0.3.0'

Browse Source 
add more unittest

See merge request megasearch/vecwise_engine!136

Former-commit-id: 6eaff772efc7dc5abdf185c286c5724ce0452b50
This commit is contained in:
jinhai 2019-06-30 15:21:32 +08:00
commit 64f01e2c83
2 changed files with 39 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
cpp/unittest/db/meta_tests.cpp
View File

@ -264,4 +264,9 @@ TEST_F(MetaTest, TABLE_FILES_TEST) {
ASSERT_TRUE(status.ok());
ASSERT_EQ(dated_files[table_file.date_].size(),
to_index_files_cnt+raw_files_cnt+index_files_cnt);
status = impl_->FilesToSearch(table_id, meta::DatesT(), dated_files);
ASSERT_TRUE(status.ok());
ASSERT_EQ(dated_files[table_file.date_].size(),
to_index_files_cnt+raw_files_cnt+index_files_cnt);
}

34
cpp/unittest/db/misc_test.cpp
View File

@ -13,11 +13,19 @@
#include "db/Status.h"
#include "db/Options.h"
#include "db/DBMetaImpl.h"
#include "db/EngineFactory.h"
#include <vector>
using namespace zilliz::milvus;
namespace {
void CopyStatus(engine::Status& st1, engine::Status& st2) {
st1 = st2;
}
}
TEST(DBMiscTest, ENGINE_API_TEST) {
//engine api AddWithIdArray
const uint16_t dim = 512;
@ -34,6 +42,15 @@ TEST(DBMiscTest, ENGINE_API_TEST) {
auto status = engine.AddWithIdArray(vectors, ids);
ASSERT_TRUE(status.ok());
auto engine_ptr = engine::EngineFactory::Build(128, "/tmp", engine::EngineType::INVALID);
ASSERT_EQ(engine_ptr, nullptr);
engine_ptr = engine::EngineFactory::Build(128, "/tmp", engine::EngineType::FAISS_IVFFLAT);
ASSERT_NE(engine_ptr, nullptr);
engine_ptr = engine::EngineFactory::Build(128, "/tmp", engine::EngineType::FAISS_IDMAP);
ASSERT_NE(engine_ptr, nullptr);
}
TEST(DBMiscTest, EXCEPTION_TEST) {
@ -65,6 +82,10 @@ TEST(DBMiscTest, STATUS_TEST) {
ASSERT_TRUE(status.IsDBTransactionError());
str = status.ToString();
ASSERT_FALSE(str.empty());
engine::Status status_copy = engine::Status::OK();
CopyStatus(status_copy, status);
ASSERT_TRUE(status.IsDBTransactionError());
}
TEST(DBMiscTest, OPTIONS_TEST) {
@ -88,6 +109,19 @@ TEST(DBMiscTest, OPTIONS_TEST) {
engine::ArchiveConf archive("delete", "1:2:3");
ASSERT_TRUE(archive.GetCriterias().empty());
}
{
engine::ArchiveConf archive("delete");
engine::ArchiveConf::CriteriaT criterial = {
{"disk", 1024},
{"days", 100}
};
archive.SetCriterias(criterial);
auto crit = archive.GetCriterias();
ASSERT_EQ(criterial["disk"], 1024);
ASSERT_EQ(criterial["days"], 100);
}
}
TEST(DBMiscTest, META_TEST) {