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

Merge pull request #652 from del-zhenwu/0.6.0 

assert failed if create index with pq on ip_table on gpu, assert pass on cpu
parents 51e8be01 52b58366

tests/milvus_python_test/test_index.py

+3 −1
Original line number Diff line number Diff line
@@ -972,9 +972,11 @@ class TestIndexIP: 
        expected: return code 0, and default index param

        '''

        index_params = get_simple_index_params

        status, mode = connect._cmd("mode")

        assert status.OK()

        # status, ids = connect.add_vectors(ip_table, vectors)

        status = connect.create_index(ip_table, index_params)

        if index_params["index_type"] == IndexType.IVF_PQ:

        if str(mode) == "GPU" and index_params["index_type"] == IndexType.IVF_PQ:

            assert not status.OK()

        else:

            assert status.OK()