Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
milvus
milvus
提交
55266d8d
M
milvus
项目概览
milvus
/
milvus
11 个月 前同步成功
通知
261
Star
22476
Fork
2472
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
DevOps
流水线
流水线任务
计划
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
M
milvus
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
DevOps
DevOps
流水线
流水线任务
计划
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
流水线任务
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
未验证
提交
55266d8d
编写于
6月 07, 2023
作者:
Y
yanliang567
提交者:
GitHub
6月 07, 2023
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Add partition_key tests (#24701)
Signed-off-by:
N
yanliang567
<
yanliang.qiao@zilliz.com
>
上级
4048caa8
变更
3
展开全部
隐藏空白更改
内联
并排
Showing
3 changed file
with
706 addition
and
25 deletion
+706
-25
tests/python_client/common/common_func.py
tests/python_client/common/common_func.py
+20
-20
tests/python_client/testcases/test_partition_key.py
tests/python_client/testcases/test_partition_key.py
+681
-0
tests/python_client/testcases/test_resourcegroup.py
tests/python_client/testcases/test_resourcegroup.py
+5
-5
未找到文件。
tests/python_client/common/common_func.py
浏览文件 @
55266d8d
...
...
@@ -124,23 +124,23 @@ def gen_double_field(name=ct.default_double_field_name, is_primary=False, descri
def
gen_float_vec_field
(
name
=
ct
.
default_float_vec_field_name
,
is_primary
=
False
,
dim
=
ct
.
default_dim
,
description
=
ct
.
default_desc
):
description
=
ct
.
default_desc
,
**
kwargs
):
float_vec_field
,
_
=
ApiFieldSchemaWrapper
().
init_field_schema
(
name
=
name
,
dtype
=
DataType
.
FLOAT_VECTOR
,
description
=
description
,
dim
=
dim
,
is_primary
=
is_primary
)
is_primary
=
is_primary
,
**
kwargs
)
return
float_vec_field
def
gen_binary_vec_field
(
name
=
ct
.
default_binary_vec_field_name
,
is_primary
=
False
,
dim
=
ct
.
default_dim
,
description
=
ct
.
default_desc
):
description
=
ct
.
default_desc
,
**
kwargs
):
binary_vec_field
,
_
=
ApiFieldSchemaWrapper
().
init_field_schema
(
name
=
name
,
dtype
=
DataType
.
BINARY_VECTOR
,
description
=
description
,
dim
=
dim
,
is_primary
=
is_primary
)
is_primary
=
is_primary
,
**
kwargs
)
return
binary_vec_field
def
gen_default_collection_schema
(
description
=
ct
.
default_desc
,
primary_field
=
ct
.
default_int64_field_name
,
auto_id
=
False
,
dim
=
ct
.
default_dim
,
enable_dynamic_field
=
False
,
with_json
=
True
):
auto_id
=
False
,
dim
=
ct
.
default_dim
,
enable_dynamic_field
=
False
,
with_json
=
True
,
**
kwargs
):
if
enable_dynamic_field
:
if
primary_field
is
ct
.
default_int64_field_name
:
fields
=
[
gen_int64_field
(),
gen_float_vec_field
(
dim
=
dim
)]
...
...
@@ -157,50 +157,50 @@ def gen_default_collection_schema(description=ct.default_desc, primary_field=ct.
schema
,
_
=
ApiCollectionSchemaWrapper
().
init_collection_schema
(
fields
=
fields
,
description
=
description
,
primary_field
=
primary_field
,
auto_id
=
auto_id
,
enable_dynamic_field
=
enable_dynamic_field
)
enable_dynamic_field
=
enable_dynamic_field
,
**
kwargs
)
return
schema
def
gen_general_collection_schema
(
description
=
ct
.
default_desc
,
primary_field
=
ct
.
default_int64_field_name
,
auto_id
=
False
,
is_binary
=
False
,
dim
=
ct
.
default_dim
):
auto_id
=
False
,
is_binary
=
False
,
dim
=
ct
.
default_dim
,
**
kwargs
):
if
is_binary
:
fields
=
[
gen_int64_field
(),
gen_float_field
(),
gen_string_field
(),
gen_binary_vec_field
(
dim
=
dim
)]
else
:
fields
=
[
gen_int64_field
(),
gen_float_field
(),
gen_string_field
(),
gen_float_vec_field
(
dim
=
dim
)]
schema
,
_
=
ApiCollectionSchemaWrapper
().
init_collection_schema
(
fields
=
fields
,
description
=
description
,
primary_field
=
primary_field
,
auto_id
=
auto_id
)
primary_field
=
primary_field
,
auto_id
=
auto_id
,
**
kwargs
)
return
schema
def
gen_string_pk_default_collection_schema
(
description
=
ct
.
default_desc
,
primary_field
=
ct
.
default_string_field_name
,
auto_id
=
False
,
dim
=
ct
.
default_dim
):
auto_id
=
False
,
dim
=
ct
.
default_dim
,
**
kwargs
):
fields
=
[
gen_int64_field
(),
gen_float_field
(),
gen_string_field
(),
gen_json_field
(),
gen_float_vec_field
(
dim
=
dim
)]
schema
,
_
=
ApiCollectionSchemaWrapper
().
init_collection_schema
(
fields
=
fields
,
description
=
description
,
primary_field
=
primary_field
,
auto_id
=
auto_id
)
primary_field
=
primary_field
,
auto_id
=
auto_id
,
**
kwargs
)
return
schema
def
gen_json_default_collection_schema
(
description
=
ct
.
default_desc
,
primary_field
=
ct
.
default_int64_field_name
,
auto_id
=
False
,
dim
=
ct
.
default_dim
):
auto_id
=
False
,
dim
=
ct
.
default_dim
,
**
kwargs
):
fields
=
[
gen_int64_field
(),
gen_float_field
(),
gen_string_field
(),
gen_json_field
(),
gen_float_vec_field
(
dim
=
dim
)]
schema
,
_
=
ApiCollectionSchemaWrapper
().
init_collection_schema
(
fields
=
fields
,
description
=
description
,
primary_field
=
primary_field
,
auto_id
=
auto_id
)
primary_field
=
primary_field
,
auto_id
=
auto_id
,
**
kwargs
)
return
schema
def
gen_multiple_json_default_collection_schema
(
description
=
ct
.
default_desc
,
primary_field
=
ct
.
default_int64_field_name
,
auto_id
=
False
,
dim
=
ct
.
default_dim
):
auto_id
=
False
,
dim
=
ct
.
default_dim
,
**
kwargs
):
fields
=
[
gen_int64_field
(),
gen_float_field
(),
gen_string_field
(),
gen_json_field
(
name
=
"json1"
),
gen_json_field
(
name
=
"json2"
),
gen_float_vec_field
(
dim
=
dim
)]
schema
,
_
=
ApiCollectionSchemaWrapper
().
init_collection_schema
(
fields
=
fields
,
description
=
description
,
primary_field
=
primary_field
,
auto_id
=
auto_id
)
primary_field
=
primary_field
,
auto_id
=
auto_id
,
**
kwargs
)
return
schema
def
gen_collection_schema_all_datatype
(
description
=
ct
.
default_desc
,
primary_field
=
ct
.
default_int64_field_name
,
auto_id
=
False
,
dim
=
ct
.
default_dim
,
enable_dynamic_field
=
False
,
with_json
=
True
):
enable_dynamic_field
=
False
,
with_json
=
True
,
**
kwargs
):
if
enable_dynamic_field
:
fields
=
[
gen_int64_field
(),
gen_float_vec_field
(
dim
=
dim
)]
else
:
...
...
@@ -211,22 +211,22 @@ def gen_collection_schema_all_datatype(description=ct.default_desc,
fields
.
remove
(
gen_json_field
())
schema
,
_
=
ApiCollectionSchemaWrapper
().
init_collection_schema
(
fields
=
fields
,
description
=
description
,
primary_field
=
primary_field
,
auto_id
=
auto_id
,
enable_dynamic_field
=
enable_dynamic_field
)
enable_dynamic_field
=
enable_dynamic_field
,
**
kwargs
)
return
schema
def
gen_collection_schema
(
fields
,
primary_field
=
None
,
description
=
ct
.
default_desc
,
auto_id
=
False
):
def
gen_collection_schema
(
fields
,
primary_field
=
None
,
description
=
ct
.
default_desc
,
auto_id
=
False
,
**
kwargs
):
schema
,
_
=
ApiCollectionSchemaWrapper
().
init_collection_schema
(
fields
=
fields
,
primary_field
=
primary_field
,
description
=
description
,
auto_id
=
auto_id
)
description
=
description
,
auto_id
=
auto_id
,
**
kwargs
)
return
schema
def
gen_default_binary_collection_schema
(
description
=
ct
.
default_desc
,
primary_field
=
ct
.
default_int64_field_name
,
auto_id
=
False
,
dim
=
ct
.
default_dim
):
auto_id
=
False
,
dim
=
ct
.
default_dim
,
**
kwargs
):
fields
=
[
gen_int64_field
(),
gen_float_field
(),
gen_string_field
(),
gen_binary_vec_field
(
dim
=
dim
)]
binary_schema
,
_
=
ApiCollectionSchemaWrapper
().
init_collection_schema
(
fields
=
fields
,
description
=
description
,
primary_field
=
primary_field
,
auto_id
=
auto_id
)
auto_id
=
auto_id
,
**
kwargs
)
return
binary_schema
...
...
tests/python_client/testcases/test_partition_key.py
0 → 100644
浏览文件 @
55266d8d
此差异已折叠。
点击以展开。
tests/python_client/testcases/test_resourcegroup.py
浏览文件 @
55266d8d
...
...
@@ -183,7 +183,7 @@ class TestResourceGroupParams(TestcaseBase):
check_task
=
ct
.
CheckTasks
.
check_rg_property
,
check_items
=
{
"name"
:
rg_name
})
error
=
{
ct
.
err_code
:
999
,
ct
.
err_msg
:
"
failed to create resource group, err=
resource group already exist"
}
ct
.
err_msg
:
"resource group already exist"
}
self
.
init_resource_group
(
name
=
rg_name
,
check_task
=
ct
.
CheckTasks
.
err_res
,
check_items
=
error
)
...
...
@@ -243,7 +243,7 @@ class TestResourceGroupParams(TestcaseBase):
rgs
=
self
.
utility_wrap
.
list_resource_groups
()[
0
]
assert
len
(
rgs
)
==
max_rg_num
error
=
{
ct
.
err_code
:
999
,
ct
.
err_msg
:
'
failed to create resource group, err=
resource group num reach limit'
}
error
=
{
ct
.
err_code
:
999
,
ct
.
err_msg
:
'resource group num reach limit'
}
self
.
init_resource_group
(
name
=
cf
.
gen_unique_str
(
'rg'
),
check_task
=
CheckTasks
.
err_res
,
check_items
=
error
)
...
...
@@ -577,7 +577,7 @@ class TestTransferNode(TestcaseBase):
check_items
=
default_rg_info
)
@
pytest
.
mark
.
tags
(
CaseLabel
.
MultiQueryNodes
)
@
pytest
.
mark
.
parametrize
(
"num_node"
,
[
0
,
99
,
-
1
,
0.5
,
True
,
"str"
])
@
pytest
.
mark
.
parametrize
(
"num_node"
,
[
0
,
99
,
-
1
,
0.5
,
"str"
])
def
test_transfer_node_with_wrong_num_node
(
self
,
num_node
):
"""
Method:
...
...
@@ -1426,7 +1426,7 @@ class TestResourceGroupMultiNodes(TestcaseBase):
rgB_name
=
cf
.
gen_unique_str
(
'rgB'
)
self
.
init_resource_group
(
name
=
rgB_name
)
# transfer
1 nodes to rgA, 2
nodes to rgB
# transfer
2 nodes to rgA, 3
nodes to rgB
self
.
utility_wrap
.
transfer_node
(
source
=
ct
.
default_resource_group_name
,
target
=
rgA_name
,
num_node
=
2
)
...
...
@@ -1509,7 +1509,7 @@ class TestResourceGroupMultiNodes(TestcaseBase):
)
# wait transfer completed for replicas
time
.
sleep
(
6
0
)
time
.
sleep
(
12
0
)
# make growing
if
with_growing
:
res
,
_
=
collection_w
.
insert
(
cf
.
gen_default_list_data
(
nb
,
dim
=
dim
,
start
=
7
*
nb
))
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录