未验证 提交 ccfca780 编写于 作者: B BossZou 提交者: GitHub

[skip ci] Rename dimension to dim in dev test case (#3030)

Signed-off-by: Nyhz <413554850@qq.com>
上级 1307ba4e
...@@ -243,7 +243,7 @@ class TestCreateCollectionInvalid(object): ...@@ -243,7 +243,7 @@ class TestCreateCollectionInvalid(object):
dimension = get_dim dimension = get_dim
collection_name = gen_unique_str() collection_name = gen_unique_str()
fields = copy.deepcopy(default_fields) fields = copy.deepcopy(default_fields)
fields["fields"][-1]["params"]["dimension"] = dimension fields["fields"][-1]["params"]["dim"] = dimension
with pytest.raises(Exception) as e: with pytest.raises(Exception) as e:
connect.create_collection(collection_name, fields) connect.create_collection(collection_name, fields)
...@@ -282,7 +282,7 @@ class TestCreateCollectionInvalid(object): ...@@ -282,7 +282,7 @@ class TestCreateCollectionInvalid(object):
''' '''
collection_name = gen_unique_str(collection_id) collection_name = gen_unique_str(collection_id)
fields = copy.deepcopy(default_fields) fields = copy.deepcopy(default_fields)
fields["fields"][-1]["params"].pop("dimension") fields["fields"][-1]["params"].pop("dim")
with pytest.raises(Exception) as e: with pytest.raises(Exception) as e:
connect.create_collection(collection_name, fields) connect.create_collection(collection_name, fields)
......
...@@ -138,7 +138,7 @@ def jac_collection(request, connect): ...@@ -138,7 +138,7 @@ def jac_collection(request, connect):
ori_collection_name = getattr(request.module, "collection_id", "test") ori_collection_name = getattr(request.module, "collection_id", "test")
collection_name = gen_unique_str(ori_collection_name) collection_name = gen_unique_str(ori_collection_name)
fields = gen_default_fields() fields = gen_default_fields()
fields["fields"][-1] = {"field": "binary_vector", "type": DataType.BINARY_VECTOR, "params": {"dimension": dimension, "metric_type": "JACCARD"}} fields["fields"][-1] = {"field": "binary_vector", "type": DataType.BINARY_VECTOR, "params": {"dim": dimension, "metric_type": "JACCARD"}}
logging.getLogger().info(fields) logging.getLogger().info(fields)
try: try:
connect.create_collection(collection_name, fields) connect.create_collection(collection_name, fields)
...@@ -158,7 +158,7 @@ def ham_collection(request, connect): ...@@ -158,7 +158,7 @@ def ham_collection(request, connect):
ori_collection_name = getattr(request.module, "collection_id", "test") ori_collection_name = getattr(request.module, "collection_id", "test")
collection_name = gen_unique_str(ori_collection_name) collection_name = gen_unique_str(ori_collection_name)
fields = gen_default_fields() fields = gen_default_fields()
fields["fields"][-1] = {"field": "binary_vector", "type": DataType.BINARY_VECTOR, "params": {"dimension": dimension, "metric_type": "HAMMING"}} fields["fields"][-1] = {"field": "binary_vector", "type": DataType.BINARY_VECTOR, "params": {"dim": dimension, "metric_type": "HAMMING"}}
try: try:
connect.create_collection(collection_name, fields) connect.create_collection(collection_name, fields)
except Exception as e: except Exception as e:
...@@ -177,7 +177,7 @@ def tanimoto_collection(request, connect): ...@@ -177,7 +177,7 @@ def tanimoto_collection(request, connect):
ori_collection_name = getattr(request.module, "collection_id", "test") ori_collection_name = getattr(request.module, "collection_id", "test")
collection_name = gen_unique_str(ori_collection_name) collection_name = gen_unique_str(ori_collection_name)
fields = gen_default_fields() fields = gen_default_fields()
fields["fields"][-1] = {"field": "binary_vector", "type": DataType.BINARY_VECTOR, "params": {"dimension": dimension, "metric_type": "TANIMOTO"}} fields["fields"][-1] = {"field": "binary_vector", "type": DataType.BINARY_VECTOR, "params": {"dim": dimension, "metric_type": "TANIMOTO"}}
try: try:
connect.create_collection(collection_name, fields) connect.create_collection(collection_name, fields)
except Exception as e: except Exception as e:
...@@ -196,7 +196,7 @@ def substructure_collection(request, connect): ...@@ -196,7 +196,7 @@ def substructure_collection(request, connect):
ori_collection_name = getattr(request.module, "collection_id", "test") ori_collection_name = getattr(request.module, "collection_id", "test")
collection_name = gen_unique_str(ori_collection_name) collection_name = gen_unique_str(ori_collection_name)
fields = gen_default_fields() fields = gen_default_fields()
fields["fields"][-1] = {"field": "binary_vector", "type": DataType.BINARY_VECTOR, "params": {"dimension": dimension, "metric_type": "SUBSTRUCTURE"}} fields["fields"][-1] = {"field": "binary_vector", "type": DataType.BINARY_VECTOR, "params": {"dim": dimension, "metric_type": "SUBSTRUCTURE"}}
try: try:
connect.create_collection(collection_name, fields) connect.create_collection(collection_name, fields)
except Exception as e: except Exception as e:
...@@ -216,7 +216,7 @@ def superstructure_collection(request, connect): ...@@ -216,7 +216,7 @@ def superstructure_collection(request, connect):
ori_collection_name = getattr(request.module, "collection_id", "test") ori_collection_name = getattr(request.module, "collection_id", "test")
collection_name = gen_unique_str(ori_collection_name) collection_name = gen_unique_str(ori_collection_name)
fields = gen_default_fields() fields = gen_default_fields()
fields["fields"][-1] = {"field": "binary_vector", "type": DataType.BINARY_VECTOR, "params": {"dimension": dimension, "metric_type": MetricType.SUPERSTRUCTURE}} fields["fields"][-1] = {"field": "binary_vector", "type": DataType.BINARY_VECTOR, "params": {"dim": dimension, "metric_type": MetricType.SUPERSTRUCTURE}}
try: try:
connect.create_collection(collection_name, fields) connect.create_collection(collection_name, fields)
except Exception as e: except Exception as e:
......
...@@ -205,7 +205,7 @@ def gen_single_vector_fields(): ...@@ -205,7 +205,7 @@ def gen_single_vector_fields():
continue continue
if metric_type not in ["L2", "IP"] and data_type == DataType.FLOAT_VECTOR: if metric_type not in ["L2", "IP"] and data_type == DataType.FLOAT_VECTOR:
continue continue
field = {"field": data_type.name, "type": data_type, "params": {"metric_type": metric_type, "dimension": dimension}} field = {"field": data_type.name, "type": data_type, "params": {"metric_type": metric_type, "dim": dimension}}
fields.append(field) fields.append(field)
return fields return fields
...@@ -215,7 +215,7 @@ def gen_default_fields(): ...@@ -215,7 +215,7 @@ def gen_default_fields():
"fields": [ "fields": [
{"field": "int64", "type": DataType.INT64}, {"field": "int64", "type": DataType.INT64},
{"field": "float", "type": DataType.FLOAT}, {"field": "float", "type": DataType.FLOAT},
{"field": "vector", "type": DataType.FLOAT_VECTOR, "params": {"metric_type": "L2", "dimension": dimension}} {"field": "vector", "type": DataType.FLOAT_VECTOR, "params": {"metric_type": "L2", "dim": dimension}}
], ],
"segment_size": segment_size "segment_size": segment_size
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册