From 69a088cc12f63d0f5b97d94e6a5ef836afbc251c Mon Sep 17 00:00:00 2001 From: del-zhenwu <56623710+del-zhenwu@users.noreply.github.com> Date: Fri, 15 May 2020 19:31:26 +0800 Subject: [PATCH] Tear down in test_collection_add_rows_count_multi_threading (#2356) * Update test_collection_add_rows_count_multi_threading Signed-off-by: zw * remove repeat Signed-off-by: zw --- tests/milvus_python_test/test_add_vectors.py | 9 ++------- 1 file changed, 2 insertions(+), 7 deletions(-) diff --git a/tests/milvus_python_test/test_add_vectors.py b/tests/milvus_python_test/test_add_vectors.py index 72266494..e0da4b0a 100644 --- a/tests/milvus_python_test/test_add_vectors.py +++ b/tests/milvus_python_test/test_add_vectors.py @@ -582,7 +582,8 @@ class TestAddBase: @pytest.mark.level(2) @pytest.mark.timeout(30) - def test_collection_add_rows_count_multi_threading(self, args): + @pytest.mark.repeat(2) + def test_collection_add_rows_count_multi_threading(self, args, collection): ''' target: test collection rows_count is correct or not with multi threading method: create collection and add vectors in it(idmap), @@ -593,13 +594,7 @@ class TestAddBase: pytest.skip("Skip test in http mode") thread_num = 8 threads = [] - collection = gen_unique_str() - param = {'collection_name': collection, - 'dimension': dim, - 'index_file_size': index_file_size, - 'metric_type': MetricType.L2} milvus = get_milvus(host=args["ip"], port=args["port"], handler=args["handler"]) - milvus.create_collection(param) vectors = gen_vectors(nb, dim) def add(thread_i): logging.getLogger().info("In thread-%d" % thread_i) -- GitLab