Exemple #1
0
 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
Exemple #2
0
    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]
Exemple #3
0
 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']
Exemple #4
0
 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