diff --git a/tests/milvus_python_test/test_index.py b/tests/milvus_python_test/test_index.py index 917bef962cbd24dedb461b336695e39534102171..a86f65dfdd6d3b4137892afd1dbfec4b3ae4d749 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): '''