milvus-io
/
milvus
mirror of https://github.com/milvus-io/milvus.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Change == to is when assert bool variable (#9163) 

Signed-off-by: zhuwenxing <wenxing.zhu@zilliz.com>
pull/8968/head
zhuwenxing 2021-10-03 21:35:56 +08:00 committed by GitHub
parent 3c9de73446
commit 4118e5de26
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 10 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
tests/python_client/testcases/test_alias.py
View File

@ -121,7 +121,6 @@ class TestAliasOperation(TestcaseBase):
check_task=CheckTasks.check_collection_property, check_task=CheckTasks.check_collection_property,
check_items={exp_name: c_2_name, exp_schema: default_schema}) check_items={exp_name: c_2_name, exp_schema: default_schema})
for _ in range(5): for _ in range(5):
partition_name = cf.gen_unique_str("partition") partition_name = cf.gen_unique_str("partition")
# create partition with different names and check the partition exists # create partition with different names and check the partition exists
@ -225,7 +224,7 @@ class TestAliasOperation(TestcaseBase):
collection_w.create_partition(partition_name) collection_w.create_partition(partition_name)
# assert partition # assert partition
pytest.assume(create_partition_flag == True and pytest.assume(create_partition_flag is True and
[p.name for p in collection_alias.partitions] == [p.name for p in collection_w.partitions]) [p.name for p in collection_alias.partitions] == [p.name for p in collection_w.partitions])
# insert data by alias # insert data by alias
@ -238,7 +237,7 @@ class TestAliasOperation(TestcaseBase):
collection_w.insert(data=df) collection_w.insert(data=df)
# assert insert data # assert insert data
pytest.assume(insert_data_flag == True and pytest.assume(insert_data_flag is True and
collection_alias.num_entities == ct.default_nb and collection_alias.num_entities == ct.default_nb and
collection_w.num_entities == ct.default_nb) collection_w.num_entities == ct.default_nb)
@ -252,9 +251,9 @@ class TestAliasOperation(TestcaseBase):
collection_w.create_index(field_name="float_vector", index_params=default_index) collection_w.create_index(field_name="float_vector", index_params=default_index)
# assert create index # assert create index
pytest.assume(create_index_flag == True and pytest.assume(create_index_flag is True and
collection_alias.has_index() == True and collection_alias.has_index() is True and
collection_w.has_index()[0] == True) collection_w.has_index()[0] is True)
# load by alias # load by alias
try: try:
@ -264,7 +263,7 @@ class TestAliasOperation(TestcaseBase):
load_collection_flag = False load_collection_flag = False
collection_w.load() collection_w.load()
# assert load # assert load
pytest.assume(load_collection_flag == True) pytest.assume(load_collection_flag is True)
# search by alias # search by alias
topK = 5 topK = 5
@ -286,7 +285,7 @@ class TestAliasOperation(TestcaseBase):
"int64 >= 0", output_fields=["int64"] "int64 >= 0", output_fields=["int64"]
) )
# assert search # assert search
pytest.assume(search_flag == True and alias_res[0].ids == collection_res[0].ids) pytest.assume(search_flag is True and alias_res[0].ids == collection_res[0].ids)
# release by alias # release by alias
try: try:
@ -296,8 +295,7 @@ class TestAliasOperation(TestcaseBase):
release_collection_flag = False release_collection_flag = False
collection_w.release() collection_w.release()
# assert release # assert release
pytest.assume(release_collection_flag == True) pytest.assume(release_collection_flag is True)
@pytest.mark.tags(CaseLabel.L1) @pytest.mark.tags(CaseLabel.L1)
def test_alias_called_by_utility_has_collection(self): def test_alias_called_by_utility_has_collection(self):
@ -322,7 +320,7 @@ class TestAliasOperation(TestcaseBase):
exp_schema: default_schema}) exp_schema: default_schema})
res, _ = self.utility_wrap.has_collection(alias_name) res, _ = self.utility_wrap.has_collection(alias_name)
assert res == True assert res is True
@pytest.mark.tags(CaseLabel.L1) @pytest.mark.tags(CaseLabel.L1)
def test_alias_called_by_utility_drop_collection(self): def test_alias_called_by_utility_drop_collection(self):
@ -374,9 +372,7 @@ class TestAliasOperation(TestcaseBase):
exp_schema: default_schema}) exp_schema: default_schema})
res, _ = self.utility_wrap.has_partition(alias_name, partition_name) res, _ = self.utility_wrap.has_partition(alias_name, partition_name)
assert res == True assert res is True
class TestAliasOperationInvalid(TestcaseBase): class TestAliasOperationInvalid(TestcaseBase):