admin/milvus
1
0
Fork 0
mirror of https://gitee.com/milvus-io/milvus.git synced 2026-01-07 19:31:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #672 from del-zhenwu/0.6.0

Browse Source 
fix pq case bug on CPU-version
This commit is contained in:
Jin Hai 2019-12-04 10:11:36 +08:00 committed by GitHub
commit 98e0aa513d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tests/milvus_python_test/test_index.py
View File

@ -587,6 +587,8 @@ class TestIndexIP:
'''
index_params = get_simple_index_params
logging.getLogger().info(index_params)
if index_params["index_type"] == IndexType.IVF_PQ:
pytest.skip("Skip some PQ cases")
status, ids = connect.add_vectors(ip_table, vectors)
status = connect.create_index(ip_table, index_params)
assert status.OK()
@ -812,11 +814,13 @@ class TestIndexIP:
status, result = connect.describe_index(ip_table)
logging.getLogger().info(result)
assert result._table_name == ip_table
if index_params["index_type"] == IndexType.IVF_PQ:
status, mode = connect._cmd("mode")
if str(mode) == "GPU" and index_params["index_type"] == IndexType.IVF_PQ:
assert result._index_type == IndexType.FLAT
assert result._nlist == 16384
else:
assert result._index_type == index_params["index_type"]
assert result._nlist == index_params["nlist"]
def test_describe_index_partition(self, connect, ip_table, get_simple_index_params):
'''