def test_cmp(self, raw_dict_binary_vector): import copy field1 = FieldSchema.construct_from_dict(raw_dict_binary_vector) field2 = FieldSchema.construct_from_dict(raw_dict_binary_vector) assert field1 == field2 dict1 = copy.deepcopy(raw_dict_binary_vector) dict1["name"] = dict1["name"] + "_" field3 = FieldSchema.construct_from_dict(dict1) assert field1 != field3
def test_to_dict(self, raw_dict_norm, raw_dict_float_vector, raw_dict_binary_vector): fields = [] dicts = [raw_dict_norm, raw_dict_float_vector, raw_dict_binary_vector] fields.append(FieldSchema.construct_from_dict(raw_dict_norm)) fields.append(FieldSchema.construct_from_dict(raw_dict_float_vector)) fields.append(FieldSchema.construct_from_dict(raw_dict_binary_vector)) for i, f in enumerate(fields): target = f.to_dict() assert target == dicts[i] assert target is not dicts[i]
def test_constructor_from_float_dict(self, raw_dict_float_vector): field = FieldSchema.construct_from_dict(raw_dict_float_vector) assert field.dtype == DataType.FLOAT_VECTOR assert field.description == raw_dict_float_vector['description'] assert field.is_primary == False assert field.name == raw_dict_float_vector['name'] assert field.dim == raw_dict_float_vector['params']['dim']
def test_constructor_from_norm_dict(self, raw_dict_norm): field = FieldSchema.construct_from_dict(raw_dict_norm) assert field.dtype == DataType.INT64 assert field.description == raw_dict_norm['description'] assert field.is_primary == False assert field.name == raw_dict_norm['name'] assert field.dim is None assert field.dummy is None