Unverified Commit 98e0aa51 authored by Jin Hai's avatar Jin Hai Committed by GitHub
Browse files

Merge pull request #672 from del-zhenwu/0.6.0 

fix pq case bug on CPU-version
parents 56e3b8fe 73562e86

tests/milvus_python_test/test_index.py

+5 −1
Original line number Diff line number Diff line
@@ -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):

        '''