| @@ 652-675 (lines=24) @@ | ||
| 649 | status, res = connect.count_entities(ham_collection) |
|
| 650 | assert res == 0 |
|
| 651 | ||
| 652 | def test_collection_rows_count_multi_collections(self, connect): |
|
| 653 | ''' |
|
| 654 | target: test collection rows_count is correct or not with multiple collections of IP |
|
| 655 | method: create collection and add vectors in it, |
|
| 656 | assert the value returned by count_entities method is equal to length of vectors |
|
| 657 | expected: the count is equal to the length of vectors |
|
| 658 | ''' |
|
| 659 | nq = 100 |
|
| 660 | tmp, vectors = gen_binary_vectors(nq, dim) |
|
| 661 | collection_list = [] |
|
| 662 | for i in range(20): |
|
| 663 | collection_name = gen_unique_str('test_collection_rows_count_multi_collections') |
|
| 664 | collection_list.append(collection_name) |
|
| 665 | param = {'collection_name': collection_name, |
|
| 666 | 'dimension': dim, |
|
| 667 | 'index_file_size': index_file_size, |
|
| 668 | 'metric_type': MetricType.HAMMING} |
|
| 669 | connect.create_collection(param) |
|
| 670 | res = connect.insert(collection_name=collection_name, records=vectors) |
|
| 671 | connect.flush(collection_list) |
|
| 672 | for i in range(20): |
|
| 673 | status, res = connect.count_entities(collection_list[i]) |
|
| 674 | assert status.OK() |
|
| 675 | assert res == nq |
|
| 676 | ||
| 677 | ||
| 678 | class TestCollectionCountTANIMOTO: |
|
| @@ 459-482 (lines=24) @@ | ||
| 456 | status, res = connect.count_entities(jac_collection) |
|
| 457 | assert res == 0 |
|
| 458 | ||
| 459 | def test_collection_rows_count_multi_collections(self, connect): |
|
| 460 | ''' |
|
| 461 | target: test collection rows_count is correct or not with multiple collections of IP |
|
| 462 | method: create collection and add vectors in it, |
|
| 463 | assert the value returned by count_entities method is equal to length of vectors |
|
| 464 | expected: the count is equal to the length of vectors |
|
| 465 | ''' |
|
| 466 | nq = 100 |
|
| 467 | tmp, vectors = gen_binary_vectors(nq, dim) |
|
| 468 | collection_list = [] |
|
| 469 | for i in range(20): |
|
| 470 | collection_name = gen_unique_str('test_collection_rows_count_multi_collections') |
|
| 471 | collection_list.append(collection_name) |
|
| 472 | param = {'collection_name': collection_name, |
|
| 473 | 'dimension': dim, |
|
| 474 | 'index_file_size': index_file_size, |
|
| 475 | 'metric_type': MetricType.JACCARD} |
|
| 476 | connect.create_collection(param) |
|
| 477 | res = connect.insert(collection_name=collection_name, records=vectors) |
|
| 478 | connect.flush(collection_list) |
|
| 479 | for i in range(20): |
|
| 480 | status, res = connect.count_entities(collection_list[i]) |
|
| 481 | assert status.OK() |
|
| 482 | assert res == nq |
|
| 483 | ||
| 484 | class TestCollectionCountBinary: |
|
| 485 | """ |
|
| @@ 347-370 (lines=24) @@ | ||
| 344 | for p in processes: |
|
| 345 | p.join() |
|
| 346 | ||
| 347 | def test_collection_rows_count_multi_collections(self, connect): |
|
| 348 | ''' |
|
| 349 | target: test collection rows_count is correct or not with multiple collections of IP |
|
| 350 | method: create collection and add vectors in it, |
|
| 351 | assert the value returned by count_entities method is equal to length of vectors |
|
| 352 | expected: the count is equal to the length of vectors |
|
| 353 | ''' |
|
| 354 | nq = 100 |
|
| 355 | vectors = gen_vectors(nq, dim) |
|
| 356 | collection_list = [] |
|
| 357 | for i in range(20): |
|
| 358 | collection_name = gen_unique_str('test_collection_rows_count_multi_collections') |
|
| 359 | collection_list.append(collection_name) |
|
| 360 | param = {'collection_name': collection_name, |
|
| 361 | 'dimension': dim, |
|
| 362 | 'index_file_size': index_file_size, |
|
| 363 | 'metric_type': MetricType.IP} |
|
| 364 | connect.create_collection(param) |
|
| 365 | res = connect.insert(collection_name=collection_name, records=vectors) |
|
| 366 | connect.flush(collection_list) |
|
| 367 | for i in range(20): |
|
| 368 | status, res = connect.count_entities(collection_list[i]) |
|
| 369 | assert status.OK() |
|
| 370 | assert res == nq |
|
| 371 | ||
| 372 | ||
| 373 | class TestCollectionCountJAC: |
|
| @@ 204-227 (lines=24) @@ | ||
| 201 | for p in processes: |
|
| 202 | p.join() |
|
| 203 | ||
| 204 | def test_collection_rows_count_multi_collections(self, connect): |
|
| 205 | ''' |
|
| 206 | target: test collection rows_count is correct or not with multiple collections of L2 |
|
| 207 | method: create collection and add vectors in it, |
|
| 208 | assert the value returned by count_entities method is equal to length of vectors |
|
| 209 | expected: the count is equal to the length of vectors |
|
| 210 | ''' |
|
| 211 | nq = 100 |
|
| 212 | vectors = gen_vectors(nq, dim) |
|
| 213 | collection_list = [] |
|
| 214 | for i in range(20): |
|
| 215 | collection_name = gen_unique_str() |
|
| 216 | collection_list.append(collection_name) |
|
| 217 | param = {'collection_name': collection_name, |
|
| 218 | 'dimension': dim, |
|
| 219 | 'index_file_size': index_file_size, |
|
| 220 | 'metric_type': MetricType.L2} |
|
| 221 | connect.create_collection(param) |
|
| 222 | res = connect.insert(collection_name=collection_name, records=vectors) |
|
| 223 | connect.flush(collection_list) |
|
| 224 | for i in range(20): |
|
| 225 | status, res = connect.count_entities(collection_list[i]) |
|
| 226 | assert status.OK() |
|
| 227 | assert res == nq |
|
| 228 | ||
| 229 | ||
| 230 | class TestCollectionCountIP: |
|
| @@ 1289-1309 (lines=21) @@ | ||
| 1286 | assert status.OK() |
|
| 1287 | assert len(result) == 1 |
|
| 1288 | ||
| 1289 | def test_add_vector_multi_collections(self, connect): |
|
| 1290 | ''' |
|
| 1291 | target: test add vectors is correct or not with multiple collections of IP |
|
| 1292 | method: create 50 collections and add vectors into them in turn |
|
| 1293 | expected: status ok |
|
| 1294 | ''' |
|
| 1295 | nq = 100 |
|
| 1296 | vectors = gen_vectors(nq, dim) |
|
| 1297 | collection_list = [] |
|
| 1298 | for i in range(20): |
|
| 1299 | collection_name = gen_unique_str( |
|
| 1300 | 'test_add_vector_multi_collections') |
|
| 1301 | collection_list.append(collection_name) |
|
| 1302 | param = {'collection_name': collection_name, |
|
| 1303 | 'dimension': dim, |
|
| 1304 | 'index_file_size': index_file_size, |
|
| 1305 | 'metric_type': MetricType.IP} |
|
| 1306 | connect.create_collection(param) |
|
| 1307 | for j in range(10): |
|
| 1308 | for i in range(20): |
|
| 1309 | status, ids = connect.insert( |
|
| 1310 | collection_name=collection_list[i], records=vectors) |
|
| 1311 | assert status.OK() |
|
| 1312 | ||
| @@ 646-666 (lines=21) @@ | ||
| 643 | status, res = milvus.count_entities(collection) |
|
| 644 | assert res == thread_num * nb |
|
| 645 | ||
| 646 | def test_add_vector_multi_collections(self, connect): |
|
| 647 | ''' |
|
| 648 | target: test add vectors is correct or not with multiple collections of L2 |
|
| 649 | method: create 50 collections and add vectors into them in turn |
|
| 650 | expected: status ok |
|
| 651 | ''' |
|
| 652 | nq = 100 |
|
| 653 | vectors = gen_vectors(nq, dim) |
|
| 654 | collection_list = [] |
|
| 655 | for i in range(20): |
|
| 656 | collection_name = gen_unique_str( |
|
| 657 | 'test_add_vector_multi_collections') |
|
| 658 | collection_list.append(collection_name) |
|
| 659 | param = {'collection_name': collection_name, |
|
| 660 | 'dimension': dim, |
|
| 661 | 'index_file_size': index_file_size, |
|
| 662 | 'metric_type': MetricType.L2} |
|
| 663 | connect.create_collection(param) |
|
| 664 | for j in range(5): |
|
| 665 | for i in range(20): |
|
| 666 | status, ids = connect.insert( |
|
| 667 | collection_name=collection_list[i], records=vectors) |
|
| 668 | assert status.OK() |
|
| 669 | ||