Merge pull request #571 from del-zhenwu/0.6.0

Add regex cases to test search partition tags in 0.6.0

tests/milvus_python_test/test_search_vectors.py

@@ -266,6 +266,37 @@ class TestSearchBase:
         assert check_result(result[1], new_ids[0])
         assert result[1][0].distance <= epsilon
 
+    def test_search_l2_index_params_partition_F(self, connect, table, get_simple_index_params):
+        '''
+        target: test basic search fuction, all the search params is corrent, test all index params, and build
+        method: search table with the given vectors and tags with "re" expr, check the result
+        expected: search status ok, and the length of the result is top_k
+        '''
+        tag = "atag"
+        new_tag = "new_tag"
+        index_params = get_simple_index_params
+        logging.getLogger().info(index_params)
+        partition_name = gen_unique_str()
+        new_partition_name = gen_unique_str()
+        status = connect.create_partition(table, partition_name, tag)
+        status = connect.create_partition(table, new_partition_name, new_tag)
+        vectors, ids = self.init_data(connect, partition_name)
+        new_vectors, new_ids = self.init_data(connect, new_partition_name, nb=1000)
+        status = connect.create_index(table, index_params)
+        query_vec = [vectors[0], new_vectors[0]]
+        top_k = 10
+        nprobe = 1
+        status, result = connect.search_vectors(table, top_k, nprobe, query_vec, partition_tags=["new(.*)"])
+        logging.getLogger().info(result)
+        assert status.OK()
+        assert result[0][0].distance > epsilon
+        assert result[1][0].distance <= epsilon
+        status, result = connect.search_vectors(table, top_k, nprobe, query_vec, partition_tags=["(.*)tag"])
+        logging.getLogger().info(result)
+        assert status.OK()
+        assert result[0][0].distance <= epsilon
+        assert result[1][0].distance <= epsilon
+
     def test_search_ip_index_params(self, connect, ip_table, get_index_params):
         '''
         target: test basic search fuction, all the search params is corrent, test all index params, and build