milvus-io
/
milvus
mirror of https://github.com/milvus-io/milvus.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix pq case bug

pull/672/head
zhenwu 2019-12-03 19:55:51 +08:00
parent 56e3b8feb1
commit 73562e869b
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):
'''