提交 becce20f 编写于 作者: J Jin Hai 提交者: GitHub

Merge pull request #73 from del-zhenwu/0.5.0

Update test cases 

Former-commit-id: 8a2213d6b51d6dd760d192a4adeaae38dce92283
......@@ -50,8 +50,7 @@ class TestAddBase:
'''
vector = gen_single_vector(dim)
status, ids = connect.add_vectors(table, vector)
ret = connect.has_table(table)
assert ret == True
assert assert_has_table(connect, table)
@pytest.mark.timeout(ADD_TIMEOUT)
def test_delete_table_add_vector(self, connect, table):
......@@ -618,8 +617,7 @@ class TestAddIP:
'''
vector = gen_single_vector(dim)
status, ids = connect.add_vectors(ip_table, vector)
ret = connect.has_table(ip_table)
assert ret == True
assert assert_has_table(connect, ip_table)
@pytest.mark.timeout(ADD_TIMEOUT)
def test_delete_table_add_vector(self, connect, ip_table):
......
......@@ -264,7 +264,7 @@ class TestTable:
expected: status ok, and no table in tables
'''
status = connect.delete_table(table)
assert not connect.has_table(table)
assert not assert_has_table(connect, table)
def test_delete_table_ip(self, connect, ip_table):
'''
......@@ -274,7 +274,7 @@ class TestTable:
expected: status ok, and no table in tables
'''
status = connect.delete_table(ip_table)
assert not connect.has_table(ip_table)
assert not assert_has_table(connect, ip_table)
@pytest.mark.level(2)
def test_table_delete_without_connection(self, table, dis_connect):
......@@ -314,7 +314,7 @@ class TestTable:
connect.create_table(param)
status = connect.delete_table(table_name)
time.sleep(1)
assert not connect.has_table(table_name)
assert not assert_has_table(connect, table_name)
def test_delete_create_table_repeatedly(self, connect):
'''
......@@ -371,7 +371,7 @@ class TestTable:
def deletetable(milvus):
status = milvus.delete_table(table)
# assert not status.code==0
assert milvus.has_table(table)
assert assert_has_table(milvus, table)
assert status.OK()
for i in range(process_num):
......@@ -411,11 +411,10 @@ class TestTable:
def delete(connect,ids):
i = 0
while i < loop_num:
# assert connect.has_table(table[ids*8+i])
status = connect.delete_table(table[ids*process_num+i])
time.sleep(2)
assert status.OK()
assert not connect.has_table(table[ids*process_num+i])
assert not assert_has_table(connect, table[ids*process_num+i])
i = i + 1
for i in range(process_num):
......@@ -444,7 +443,7 @@ class TestTable:
'index_file_size': index_file_size,
'metric_type': MetricType.L2}
connect.create_table(param)
assert connect.has_table(table_name)
assert assert_has_table(connect, table_name)
def test_has_table_ip(self, connect):
'''
......@@ -458,7 +457,7 @@ class TestTable:
'index_file_size': index_file_size,
'metric_type': MetricType.IP}
connect.create_table(param)
assert connect.has_table(table_name)
assert assert_has_table(connect, table_name)
@pytest.mark.level(2)
def test_has_table_without_connection(self, table, dis_connect):
......@@ -468,7 +467,7 @@ class TestTable:
expected: has table raise exception
'''
with pytest.raises(Exception) as e:
status = dis_connect.has_table(table)
assert_has_table(dis_connect, table)
def test_has_table_not_existed(self, connect):
'''
......@@ -478,7 +477,7 @@ class TestTable:
expected: False
'''
table_name = gen_unique_str("test_table")
assert not connect.has_table(table_name)
assert not assert_has_table(connect, table_name)
"""
******************************************************************
......@@ -700,7 +699,7 @@ class TestCreateTableDimInvalid(object):
'dimension': dimension,
'index_file_size': index_file_size,
'metric_type': MetricType.L2}
if isinstance(dimension, int) and dimension > 0:
if isinstance(dimension, int):
status = connect.create_table(param)
assert not status.OK()
else:
......@@ -778,7 +777,7 @@ def preload_table(connect, **params):
return status
def has(connect, **params):
status = connect.has_table(params["table_name"])
status = assert_has_table(connect, params["table_name"])
return status
def show(connect, **params):
......
......@@ -462,6 +462,11 @@ def gen_simple_index_params():
return gen_params(index_types, nlists)
def assert_has_table(conn, table_name):
status, ok = conn.has_table(table_name)
return status.OK() and ok
if __name__ == "__main__":
import numpy
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册