def gen_all_type_fields():
fields = []
for k, v in DataType.__members__.items():
if v != DataType.UNKNOWN:
field, _ = ApiFieldSchemaWrapper().init_field_schema(name=k.lower(), dtype=v)
fields.append(field)
return fields
def gen_float_field(name=ct.default_float_field_name,
is_primary=False,
description=ct.default_desc):
float_field, _ = ApiFieldSchemaWrapper().init_field_schema(
name=name,
dtype=DataType.FLOAT,
description=description,
is_primary=is_primary)
return float_field
def setup(self):
log.info(("*" * 35) + " setup " + ("*" * 35))
self.connection_wrap = ApiConnectionsWrapper()
self.collection_wrap = ApiCollectionWrapper()
self.partition_wrap = ApiPartitionWrapper()
self.index_wrap = ApiIndexWrapper()
self.utility_wrap = ApiUtilityWrapper()
self.collection_schema_wrap = ApiCollectionSchemaWrapper()
self.field_schema_wrap = ApiFieldSchemaWrapper()
def gen_double_field(name=ct.default_double_field_name,
is_primary=False,
description=ct.default_desc):
double_field, _ = ApiFieldSchemaWrapper().init_field_schema(
name=name,
dtype=DataType.DOUBLE,
description=description,
is_primary=is_primary)
return double_field
def setup_method(self, method):
log.info(("*" * 35) + " setup " + ("*" * 35))
log.info("[setup_method] Start setup test case %s." % method.__name__)
self.connection_wrap = ApiConnectionsWrapper()
self.utility_wrap = ApiUtilityWrapper()
self.collection_wrap = ApiCollectionWrapper()
self.partition_wrap = ApiPartitionWrapper()
self.index_wrap = ApiIndexWrapper()
self.collection_schema_wrap = ApiCollectionSchemaWrapper()
self.field_schema_wrap = ApiFieldSchemaWrapper()
def gen_binary_vec_field(name=ct.default_binary_vec_field_name,
is_primary=False,
dim=ct.default_dim,
description=ct.default_desc):
binary_vec_field, _ = ApiFieldSchemaWrapper().init_field_schema(
name=name,
dtype=DataType.BINARY_VECTOR,
description=description,
dim=dim,
is_primary=is_primary)
return binary_vec_field
def gen_int64_field(name=ct.default_int64_field_name,
description=ct.default_desc,
is_primary=False,
**kwargs):
int64_field, _ = ApiFieldSchemaWrapper().init_field_schema(
name=name,
dtype=DataType.INT64,
description=description,
is_primary=is_primary,
**kwargs)
return int64_field
def gen_bool_field(name=ct.default_bool_field_name,
description=ct.default_desc,
is_primary=False,
**kwargs):
bool_field, _ = ApiFieldSchemaWrapper().init_field_schema(
name=name,
dtype=DataType.BOOL,
description=description,
is_primary=is_primary,
**kwargs)
return bool_field
def gen_string_field(name=ct.default_string_field_name,
description=ct.default_desc,
is_primary=False,
max_length=ct.default_length,
**kwargs):
string_field, _ = ApiFieldSchemaWrapper().init_field_schema(
name=name,
dtype=DataType.VARCHAR,
description=description,
max_length=max_length,
is_primary=is_primary,
**kwargs)
return string_field
