chore: import upstream snapshot with attribution

tests/python_client/base/index_wrapper.py

@@ -0,0 +1,53 @@
+import sys
+from pymilvus import Index
+
+sys.path.append("..")
+from check.func_check import ResponseChecker
+from utils.api_request import api_request
+
+TIMEOUT = 20
+INDEX_NAME = ""
+
+
+class ApiIndexWrapper:
+    index = None
+
+    def init_index(self, collection, field_name, index_params, index_name=None, check_task=None, check_items=None,
+                   **kwargs):
+        disktimeout = 600
+        timeout = kwargs.get("timeout", disktimeout * 2)
+        index_name = INDEX_NAME if index_name is None else index_name
+        index_name = kwargs.get("index_name", index_name)
+        kwargs.update({"timeout": timeout, "index_name": index_name})
+
+        """ In order to distinguish the same name of index """
+        func_name = sys._getframe().f_code.co_name
+        res, is_succ = api_request([Index, collection, field_name, index_params], **kwargs)
+        self.index = res if is_succ is True else None
+        check_result = ResponseChecker(res, func_name, check_task, check_items, is_succ,
+                                       collection=collection, field_name=field_name,
+                                       index_params=index_params, **kwargs).run()
+        return res, check_result
+
+    def drop(self, index_name=None, check_task=None, check_items=None, **kwargs):
+        timeout = kwargs.get("timeout", TIMEOUT)
+        index_name = INDEX_NAME if index_name is None else index_name
+        index_name = kwargs.get("index_name", index_name)
+        kwargs.update({"timeout": timeout, "index_name": index_name})
+
+        func_name = sys._getframe().f_code.co_name
+        res, is_succ = api_request([self.index.drop], **kwargs)
+        check_result = ResponseChecker(res, func_name, check_task, check_items, is_succ, **kwargs).run()
+        return res, check_result
+
+    @property
+    def params(self):
+        return self.index.params
+
+    @property
+    def collection_name(self):
+        return self.index.collection_name
+
+    @property
+    def field_name(self):
+        return self.index.field_name