From 73562e869b8d45a46c3ac43046f6238e6d6415a7 Mon Sep 17 00:00:00 2001
From: zhenwu <zw@zilliz.com>
Date: Tue, 3 Dec 2019 19:55:51 +0800
Subject: [PATCH] fix pq case bug
---
tests/milvus_python_test/test_index.py | 6 +++++-
1 file changed, 5 insertions(+), 1 deletion(-)
diff --git a/tests/milvus_python_test/test_index.py b/tests/milvus_python_test/test_index.py
index 917bef962c..a86f65dfdd 100644
--- a/tests/milvus_python_test/test_index.py
+++ b/tests/milvus_python_test/test_index.py
@@ -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):
'''