|
@@ 1177-1192 (lines=16) @@
|
| 1174 |
|
logging.getLogger().info(ids) |
| 1175 |
|
assert not status.OK() |
| 1176 |
|
|
| 1177 |
|
@pytest.mark.timeout(ADD_TIMEOUT) |
| 1178 |
|
def test_insert_twice_not_ids_ids(self, connect, ip_collection): |
| 1179 |
|
''' |
| 1180 |
|
target: check the result of insert, with params ids and no ids |
| 1181 |
|
method: test add vectors twice, use not ids first, and then use customize ids |
| 1182 |
|
expected: status not OK |
| 1183 |
|
''' |
| 1184 |
|
nq = 5 |
| 1185 |
|
top_k = 1 |
| 1186 |
|
vectors = gen_vectors(nq, dim) |
| 1187 |
|
ids = [i for i in range(nq)] |
| 1188 |
|
status, ids = connect.insert(ip_collection, vectors) |
| 1189 |
|
assert status.OK() |
| 1190 |
|
status, ids = connect.insert(ip_collection, vectors, ids) |
| 1191 |
|
logging.getLogger().info(status) |
| 1192 |
|
logging.getLogger().info(ids) |
| 1193 |
|
assert not status.OK() |
| 1194 |
|
|
| 1195 |
|
@pytest.mark.timeout(ADD_TIMEOUT) |
|
@@ 1159-1174 (lines=16) @@
|
| 1156 |
|
for i in range(nq): |
| 1157 |
|
assert result[i][0].id == i |
| 1158 |
|
|
| 1159 |
|
@pytest.mark.timeout(ADD_TIMEOUT) |
| 1160 |
|
def test_insert_twice_ids_no_ids(self, connect, ip_collection): |
| 1161 |
|
''' |
| 1162 |
|
target: check the result of insert, with params ids and no ids |
| 1163 |
|
method: test add vectors twice, use customize ids first, and then use no ids |
| 1164 |
|
expected: status not OK |
| 1165 |
|
''' |
| 1166 |
|
nq = 5 |
| 1167 |
|
top_k = 1 |
| 1168 |
|
vectors = gen_vectors(nq, dim) |
| 1169 |
|
ids = [i for i in range(nq)] |
| 1170 |
|
status, ids = connect.insert(ip_collection, vectors, ids) |
| 1171 |
|
assert status.OK() |
| 1172 |
|
status, ids = connect.insert(ip_collection, vectors) |
| 1173 |
|
logging.getLogger().info(status) |
| 1174 |
|
logging.getLogger().info(ids) |
| 1175 |
|
assert not status.OK() |
| 1176 |
|
|
| 1177 |
|
@pytest.mark.timeout(ADD_TIMEOUT) |
|
@@ 415-430 (lines=16) @@
|
| 412 |
|
logging.getLogger().info(ids) |
| 413 |
|
assert not status.OK() |
| 414 |
|
|
| 415 |
|
@pytest.mark.timeout(ADD_TIMEOUT) |
| 416 |
|
def test_insert_twice_not_ids_ids(self, connect, collection): |
| 417 |
|
''' |
| 418 |
|
target: check the result of insert, with params ids and no ids |
| 419 |
|
method: test add vectors twice, use not ids first, and then use customize ids |
| 420 |
|
expected: status not OK |
| 421 |
|
''' |
| 422 |
|
nq = 5 |
| 423 |
|
top_k = 1 |
| 424 |
|
vectors = gen_vectors(nq, dim) |
| 425 |
|
ids = [i for i in range(nq)] |
| 426 |
|
status, ids = connect.insert(collection, vectors) |
| 427 |
|
assert status.OK() |
| 428 |
|
status, ids = connect.insert(collection, vectors, ids) |
| 429 |
|
logging.getLogger().info(status) |
| 430 |
|
logging.getLogger().info(ids) |
| 431 |
|
assert not status.OK() |
| 432 |
|
|
| 433 |
|
@pytest.mark.timeout(ADD_TIMEOUT) |
milvus-io /
milvus
