From aada3aa5db4143639c777c11adf87604e3c64253 Mon Sep 17 00:00:00 2001 From: XuanYang-cn Date: Wed, 24 Feb 2021 17:03:42 +0800 Subject: [PATCH] Adjust some ci tests Signed-off-by: XuanYang-cn --- tests/python/test_describe_collection.py | 31 ++++++++++++++++++++++++ 1 file changed, 31 insertions(+) create mode 100644 tests/python/test_describe_collection.py diff --git a/tests/python/test_describe_collection.py b/tests/python/test_describe_collection.py new file mode 100644 index 0000000000..c4315691c9 --- /dev/null +++ b/tests/python/test_describe_collection.py @@ -0,0 +1,31 @@ +from .utils import * +from .constants import * + +uid = "describe_collection" + + +class TestDescribeCollection: + """ + ****************************************************************** + The following cases are used to test `describe_collection` function + ****************************************************************** + """ + def test_describe_collection(self, connect): + ''' + target: test describe collection + method: create collection then describe the same collection + expected: returned value is the same + ''' + collection_name = gen_unique_str(uid) + df = default_fields + df["fields"].append({"name": "int32", "type": DataType.INT32}) + + connect.create_collection(collection_name, df) + info = connect.describe_collection(collection_name) + assert info.get("collection_name") == collection_name + assert len(info.get("fields")) == 4 + + for field in info.get("fields"): + assert field.get("name") in ["int32", "int64", "float", "float_vector"] + if field.get("name") == "float_vector": + assert field.get("params").get("dim") == str(default_dim)