未验证 提交 9ca1839c 编写于 作者: D del-zhenwu 提交者: GitHub

[skip ci] remove int8 (#3026)

Signed-off-by: Nzw <zw@milvus.io>
Co-authored-by: Nzw <zw@milvus.io>
上级 43cfe506
......@@ -322,7 +322,7 @@ class TestCreateCollectionInvalid(object):
fields = copy.deepcopy(default_fields)
for i in range(limit_num):
field_name = gen_unique_str("field_name")
field = {"field": field_name, "type": DataType.INT8}
field = {"field": field_name, "type": DataType.INT64}
fields["fields"].append(field)
with pytest.raises(Exception) as e:
connect.create_collection(collection_name, fields)
......@@ -332,7 +332,7 @@ class TestCreateCollectionInvalid(object):
collection_name = gen_unique_str(collection_id)
fields = copy.deepcopy(default_fields)
field_name = get_invalid_string
field = {"field": field_name, "type": DataType.INT8}
field = {"field": field_name, "type": DataType.INT64}
fields["fields"].append(field)
with pytest.raises(Exception) as e:
connect.create_collection(collection_name, fields)
......
......@@ -307,7 +307,7 @@ class TestGetBase:
ids = connect.insert(collection, entities)
connect.flush([collection])
get_ids = [ids[get_pos]]
fields = ["int8"]
fields = ["int64"]
res = connect.get_entity_by_id(collection, get_ids, fields = fields)
# assert fields
......@@ -321,7 +321,7 @@ class TestGetBase:
ids = connect.insert(collection, entities)
connect.flush([collection])
get_ids = [ids[get_pos]]
fields = ["int8", "int64", "float", "vector"]
fields = ["int64", "float", "vector"]
res = connect.get_entity_by_id(collection, get_ids, fields = fields)
# assert fields
......@@ -349,7 +349,7 @@ class TestGetBase:
ids = connect.insert(collection, entities)
connect.flush([collection])
get_ids = [ids[get_pos]]
fields = ["int1288", "int8"]
fields = ["int1288"]
with pytest.raises(Exception) as e:
res = connect.get_entity_by_id(collection, get_ids, fields = fields)
......@@ -589,4 +589,4 @@ class TestGetInvalid(object):
ids = [1]
fields = [field_name]
with pytest.raises(Exception):
res = connect.get_entity_by_id(collection, ids, fields=fields)
\ No newline at end of file
res = connect.get_entity_by_id(collection, ids, fields=fields)
......@@ -391,7 +391,7 @@ class TestInsertBase:
method: update entity field name
expected: error raised
'''
tmp_entity = update_field_name(copy.deepcopy(entity), "int8", "int8new")
tmp_entity = update_field_name(copy.deepcopy(entity), "int64", "int64new")
with pytest.raises(Exception):
connect.insert(collection, tmp_entity)
......@@ -402,7 +402,7 @@ class TestInsertBase:
method: update entity field type
expected: error raised
'''
tmp_entity = update_field_type(copy.deepcopy(entity), DataType.INT8, DataType.FLOAT)
tmp_entity = update_field_type(copy.deepcopy(entity), DataType.INT64, DataType.FLOAT)
with pytest.raises(Exception):
connect.insert(collection, tmp_entity)
......@@ -413,7 +413,7 @@ class TestInsertBase:
method: update entity field value
expected: error raised
'''
tmp_entity = update_field_value(copy.deepcopy(entity), 'int8', 's')
tmp_entity = update_field_value(copy.deepcopy(entity), 'int64', 's')
with pytest.raises(Exception):
connect.insert(collection, tmp_entity)
......@@ -848,7 +848,7 @@ class TestInsertInvalid(object):
def test_insert_with_invalid_field_name(self, connect, collection, get_field_name):
field_name = get_field_name
tmp_entity = update_field_name(copy.deepcopy(entity), "int8", get_field_name)
tmp_entity = update_field_name(copy.deepcopy(entity), "int64", get_field_name)
with pytest.raises(Exception):
connect.insert(collection, tmp_entity)
......@@ -860,7 +860,7 @@ class TestInsertInvalid(object):
def test_insert_with_invalid_field_value(self, connect, collection, get_field_int_value):
field_value = get_field_int_value
tmp_entity = update_field_type(copy.deepcopy(entity), 'int8', field_value)
tmp_entity = update_field_type(copy.deepcopy(entity), 'int64', field_value)
with pytest.raises(Exception):
connect.insert(collection, tmp_entity)
......@@ -951,7 +951,7 @@ class TestInsertInvalidIP(object):
@pytest.mark.level(2)
def test_insert_with_invalid_field_name(self, connect, ip_collection, get_field_name):
field_name = get_field_name
tmp_entity = update_field_name(copy.deepcopy(entity), "int8", get_field_name)
tmp_entity = update_field_name(copy.deepcopy(entity), "int64", get_field_name)
with pytest.raises(Exception):
connect.insert(ip_collection, tmp_entity)
......@@ -965,7 +965,7 @@ class TestInsertInvalidIP(object):
@pytest.mark.level(2)
def test_insert_with_invalid_field_value(self, connect, ip_collection, get_field_int_value):
field_value = get_field_int_value
tmp_entity = update_field_type(copy.deepcopy(entity), 'int8', field_value)
tmp_entity = update_field_type(copy.deepcopy(entity), 'int64', field_value)
with pytest.raises(Exception):
connect.insert(ip_collection, tmp_entity)
......@@ -1057,14 +1057,14 @@ class TestInsertInvalidBinary(object):
@pytest.mark.level(2)
def test_insert_with_invalid_field_name(self, connect, jac_collection, get_field_name):
field_name = get_field_name
tmp_entity = update_field_name(copy.deepcopy(binary_entity), "int8", get_field_name)
tmp_entity = update_field_name(copy.deepcopy(binary_entity), "int64", get_field_name)
with pytest.raises(Exception):
connect.insert(jac_collection, tmp_entity)
@pytest.mark.level(2)
def test_insert_with_invalid_field_value(self, connect, jac_collection, get_field_int_value):
field_value = get_field_int_value
tmp_entity = update_field_type(copy.deepcopy(binary_entity), 'int8', field_value)
tmp_entity = update_field_type(copy.deepcopy(binary_entity), 'int64', field_value)
with pytest.raises(Exception):
connect.insert(jac_collection, tmp_entity)
......@@ -1091,21 +1091,21 @@ class TestInsertInvalidBinary(object):
@pytest.mark.level(2)
def test_insert_with_invalid_field_name(self, connect, jac_collection, get_field_name):
field_name = get_field_name
tmp_entity = update_field_name(copy.deepcopy(binary_entity), "int8", get_field_name)
tmp_entity = update_field_name(copy.deepcopy(binary_entity), "int64", get_field_name)
with pytest.raises(Exception):
connect.insert(jac_collection, tmp_entity)
@pytest.mark.level(2)
def test_insert_with_invalid_field_type(self, connect, jac_collection, get_field_type):
field_type = get_field_type
tmp_entity = update_field_type(copy.deepcopy(binary_entity), 'int8', field_type)
tmp_entity = update_field_type(copy.deepcopy(binary_entity), 'int64', field_type)
with pytest.raises(Exception):
connect.insert(jac_collection, tmp_entity)
@pytest.mark.level(2)
def test_insert_with_invalid_field_value(self, connect, jac_collection, get_field_int_value):
field_value = get_field_int_value
tmp_entity = update_field_type(copy.deepcopy(binary_entity), 'int8', field_value)
tmp_entity = update_field_type(copy.deepcopy(binary_entity), 'int64', field_value)
with pytest.raises(Exception):
connect.insert(jac_collection, tmp_entity)
......
......@@ -192,7 +192,7 @@ def gen_unique_str(str_value=None):
def gen_single_filter_fields():
fields = []
for data_type in DataType:
if data_type in [DataType.INT8, DataType.INT16, DataType.INT32, DataType.INT64, DataType.FLOAT, DataType.DOUBLE]:
if data_type in [DataType.INT32, DataType.INT64, DataType.FLOAT, DataType.DOUBLE]:
fields.append({"field": data_type.name, "type": data_type})
return fields
......@@ -213,7 +213,6 @@ def gen_single_vector_fields():
def gen_default_fields():
default_fields = {
"fields": [
{"field": "int8", "type": DataType.INT8},
{"field": "int64", "type": DataType.INT64},
{"field": "float", "type": DataType.FLOAT},
{"field": "vector", "type": DataType.FLOAT_VECTOR, "params": {"metric_type": "L2", "dimension": dimension}}
......@@ -226,7 +225,6 @@ def gen_default_fields():
def gen_entities(nb, is_normal=False):
vectors = gen_vectors(nb, dimension, is_normal)
entities = [
{"field": "int8", "type": DataType.INT8, "values": [1 for i in range(nb)]},
{"field": "int64", "type": DataType.INT64, "values": [2 for i in range(nb)]},
{"field": "float", "type": DataType.FLOAT, "values": [3.0 for i in range(nb)]},
{"field": "vector", "type": DataType.FLOAT_VECTOR, "values": vectors}
......@@ -237,7 +235,6 @@ def gen_entities(nb, is_normal=False):
def gen_binary_entities(nb):
raw_vectors, vectors = gen_binary_vectors(nb, dimension)
entities = [
{"field": "int8", "type": DataType.INT8, "values": [1 for i in range(nb)]},
{"field": "int64", "type": DataType.INT64, "values": [2 for i in range(nb)]},
{"field": "float", "type": DataType.FLOAT, "values": [3.0 for i in range(nb)]},
{"field": "binary_vector", "type": DataType.BINARY_VECTOR, "values": vectors}
......@@ -248,7 +245,7 @@ def gen_binary_entities(nb):
def gen_entities_by_fields(fields, nb, dimension):
entities = []
for field in fields:
if field["type"] in [DataType.INT8, DataType.INT16, DataType.INT32, DataType.INT64]:
if field["type"] in [DataType.INT32, DataType.INT64]:
field_value = [1 for i in range(nb)]
elif field["type"] in [DataType.FLOAT, DataType.DOUBLE]:
field_value = [3.0 for i in range(nb)]
......@@ -295,7 +292,7 @@ def add_field(entities):
nb = len(entities[0]["values"])
field = {
"field": gen_unique_str(),
"type": DataType.INT8,
"type": DataType.INT64,
"values": [1 for i in range(nb)]
}
entities.append(field)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册