diff --git a/tests/python/test_describe_collection.py b/tests/python/test_describe_collection.py new file mode 100644 index 0000000000000000000000000000000000000000..c4315691c9d451d2336e55a5c30aebf103f9ee59 --- /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)