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

[skip ci] Test get entity by id (#3417) 

* [skip ci] update for Contasts

Signed-off-by: zongyufen <zongyufen@foxmail.com>

* [skip ci] test get entity by id

Signed-off-by: zongyufen <zongyufen@foxmail.com>

* [skip ci] test get entity by id

Signed-off-by: zongyufen <zongyufen@foxmail.com>
pull/3424/head
ThreadDao 2020-08-24 10:04:10 +08:00 committed by GitHub
parent b0b6aefc21
commit e89a707d6c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
6 changed files with 117 additions and 167 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/milvus-java-test/src/main/java/com/Constants.java
View File

@ -32,6 +32,10 @@ public final class Constants {
public static final List<ByteBuffer> vectorsBinary = Utils.genBinaryVectors(nb, dimension); public static final List<ByteBuffer> vectorsBinary = Utils.genBinaryVectors(nb, dimension);
public static final List<Map<String,Object>> defaultFields = Utils.genDefaultFields(dimension,false);
public static final List<Map<String,Object>> defaultBinaryFields = Utils.genDefaultFields(dimension,true);
public static final List<Map<String,Object>> defaultEntities = Utils.genDefaultEntities(dimension, nb, vectors); public static final List<Map<String,Object>> defaultEntities = Utils.genDefaultEntities(dimension, nb, vectors);
public static final List<Map<String,Object>> defaultBinaryEntities = Utils.genDefaultBinaryEntities(dimension, nb, vectorsBinary); public static final List<Map<String,Object>> defaultBinaryEntities = Utils.genDefaultBinaryEntities(dimension, nb, vectorsBinary);

17
tests/milvus-java-test/src/main/java/com/TestCollectionCount.java
View File

@ -3,19 +3,12 @@ package com;
import io.milvus.client.*; import io.milvus.client.*;
import org.testng.Assert; import org.testng.Assert;
import org.testng.annotations.Test; import org.testng.annotations.Test;
import java.nio.ByteBuffer;
import java.util.List; import java.util.List;
import java.util.Map;
public class TestCollectionCount { public class TestCollectionCount {
int segmentRowCount = 5000; int segmentRowCount = 5000;
int dimension = 128; int dimension = Constants.dimension;
int nb = 10000; int nb = Constants.nb;
List<List<Float>> vectors = Utils.genVectors(nb, dimension, true);
List<ByteBuffer> vectorsBinary = Utils.genBinaryVectors(nb, dimension);
List<Map<String,Object>> defaultEntities = Utils.genDefaultEntities(dimension,nb,vectors);
List<Map<String,Object>> defaultBinaryEntities = Utils.genDefaultBinaryEntities(dimension,nb,vectorsBinary);
@Test(dataProvider = "Collection", dataProviderClass = MainClass.class) @Test(dataProvider = "Collection", dataProviderClass = MainClass.class)
public void testCollectionCountNoVectors(MilvusClient client, String collectionName) { public void testCollectionCountNoVectors(MilvusClient client, String collectionName) {
@ -39,7 +32,7 @@ public class TestCollectionCount {
InsertParam insertParam = InsertParam insertParam =
new InsertParam.Builder(collectionName) new InsertParam.Builder(collectionName)
.withFields(defaultEntities) .withFields(Constants.defaultEntities)
.build(); .build();
InsertResponse insertResponse = client.insert(insertParam); InsertResponse insertResponse = client.insert(insertParam);
// Insert returns a list of entity ids that you will be using (if you did not supply the yourself) to reference the entities you just inserted // Insert returns a list of entity ids that you will be using (if you did not supply the yourself) to reference the entities you just inserted
@ -54,7 +47,7 @@ public class TestCollectionCount {
public void testCollectionCountBinary(MilvusClient client, String collectionName) throws InterruptedException { public void testCollectionCountBinary(MilvusClient client, String collectionName) throws InterruptedException {
// Add vectors // Add vectors
InsertParam insertParam = new InsertParam.Builder(collectionName) InsertParam insertParam = new InsertParam.Builder(collectionName)
.withFields(defaultBinaryEntities) .withFields(Constants.defaultBinaryEntities)
.build(); .build();
client.insert(insertParam); client.insert(insertParam);
client.flush(collectionName); client.flush(collectionName);
@ -75,7 +68,7 @@ public class TestCollectionCount {
Assert.assertEquals(cteateRes.ok(), true); Assert.assertEquals(cteateRes.ok(), true);
// Add vectors // Add vectors
InsertParam insertParam = new InsertParam.Builder(collectionNameNew) InsertParam insertParam = new InsertParam.Builder(collectionNameNew)
.withFields(defaultEntities) .withFields(Constants.defaultEntities)
.build(); .build();
InsertResponse insertRes = client.insert(insertParam); InsertResponse insertRes = client.insert(insertParam);
Assert.assertEquals(insertRes.ok(), true); Assert.assertEquals(insertRes.ok(), true);

25
tests/milvus-java-test/src/main/java/com/TestCollectionInfo.java
View File

@ -1,34 +1,19 @@
package com; package com;
import com.alibaba.fastjson.JSONObject; import com.alibaba.fastjson.JSONObject;
import io.milvus.client.*; import io.milvus.client.*;
import org.testng.annotations.Test; import org.testng.annotations.Test;
import java.nio.ByteBuffer;
import java.util.Collections; import java.util.Collections;
import java.util.List; import java.util.List;
import java.util.Map;
public class TestCollectionInfo { public class TestCollectionInfo {
int dimension = 128; int nb = Constants.nb;
int nb = 8000;
int nList = 1024;
int defaultNList = 16384;
String indexType = "IVF_SQ8";
String defaultIndexType = "FLAT";
String metricType = "L2";
String indexParam = Utils.setIndexParam(indexType,metricType,nList);
List<List<Float>> vectors = Utils.genVectors(nb, dimension, true);
List<ByteBuffer> vectorsBinary = Utils.genBinaryVectors(nb, dimension);
List<Map<String,Object>> defaultEntities = Utils.genDefaultEntities(dimension,nb,vectors);
List<Map<String,Object>> defaultBinaryEntities = Utils.genDefaultBinaryEntities(dimension,nb,vectorsBinary);
@Test(dataProvider = "Collection", dataProviderClass = MainClass.class) @Test(dataProvider = "Collection", dataProviderClass = MainClass.class)
public void testGetEntityIdsAfterDeleteEntities(MilvusClient client, String collectionName) { public void testGetEntityIdsAfterDeleteEntities(MilvusClient client, String collectionName) {
InsertParam insertParam = new InsertParam.Builder(collectionName) InsertParam insertParam = new InsertParam.Builder(collectionName)
.withFields(defaultEntities) .withFields(Constants.defaultEntities)
.build(); .build();
InsertResponse resInsert = client.insert(insertParam); InsertResponse resInsert = client.insert(insertParam);
client.flush(collectionName); client.flush(collectionName);
@ -45,12 +30,12 @@ public class TestCollectionInfo {
@Test(dataProvider = "Collection", dataProviderClass = MainClass.class) @Test(dataProvider = "Collection", dataProviderClass = MainClass.class)
public void testGetEntityIdsAterDeleteEntitiesIndexed(MilvusClient client, String collectionName) { public void testGetEntityIdsAterDeleteEntitiesIndexed(MilvusClient client, String collectionName) {
InsertParam insertParam = new InsertParam.Builder(collectionName) InsertParam insertParam = new InsertParam.Builder(collectionName)
.withFields(defaultEntities) .withFields(Constants.defaultEntities)
.build(); .build();
InsertResponse resInsert = client.insert(insertParam); InsertResponse resInsert = client.insert(insertParam);
client.flush(collectionName); client.flush(collectionName);
Index index = new Index.Builder(collectionName, "float_vector") Index index = new Index.Builder(collectionName, "float_vector")
.withParamsInJson(indexParam).build(); .withParamsInJson(Constants.indexParam).build();
Response createIndexResponse = client.createIndex(index); Response createIndexResponse = client.createIndex(index);
assert(createIndexResponse.ok()); assert(createIndexResponse.ok());
List<Long> idsBefore = resInsert.getEntityIds(); List<Long> idsBefore = resInsert.getEntityIds();
@ -66,7 +51,7 @@ public class TestCollectionInfo {
@Test(dataProvider = "BinaryCollection", dataProviderClass = MainClass.class) @Test(dataProvider = "BinaryCollection", dataProviderClass = MainClass.class)
public void testGetEntityIdsAfterDeleteEntitiesBinary(MilvusClient client, String collectionName) { public void testGetEntityIdsAfterDeleteEntitiesBinary(MilvusClient client, String collectionName) {
InsertParam insertParam = new InsertParam.Builder(collectionName) InsertParam insertParam = new InsertParam.Builder(collectionName)
.withFields(defaultBinaryEntities) .withFields(Constants.defaultBinaryEntities)
.build(); .build();
InsertResponse resInsert = client.insert(insertParam); InsertResponse resInsert = client.insert(insertParam);
client.flush(collectionName); client.flush(collectionName);

15
tests/milvus-java-test/src/main/java/com/TestCompact.java
View File

@ -3,23 +3,12 @@ package com;
import io.milvus.client.*; import io.milvus.client.*;
import org.testng.Assert; import org.testng.Assert;
import org.testng.annotations.Test; import org.testng.annotations.Test;
import java.nio.ByteBuffer;
import java.util.List; import java.util.List;
import java.util.Map;
public class TestCompact { public class TestCompact {
int dimension = 128;
int nb = 8000;
List<List<Float>> vectors = Utils.genVectors(nb, dimension, true);
List<ByteBuffer> vectorsBinary = Utils.genBinaryVectors(nb, dimension);
List<Map<String,Object>> defaultEntities = Utils.genDefaultEntities(dimension,nb,vectors);
List<Map<String,Object>> defaultBinaryEntities = Utils.genDefaultBinaryEntities(dimension,nb,vectorsBinary);
@Test(dataProvider = "Collection", dataProviderClass = MainClass.class) @Test(dataProvider = "Collection", dataProviderClass = MainClass.class)
public void testCompactAfterDelete(MilvusClient client, String collectionName) { public void testCompactAfterDelete(MilvusClient client, String collectionName) {
InsertParam insertParam = new InsertParam.Builder(collectionName).withFields(defaultEntities).build(); InsertParam insertParam = new InsertParam.Builder(collectionName).withFields(Constants.defaultEntities).build();
InsertResponse res = client.insert(insertParam); InsertResponse res = client.insert(insertParam);
assert(res.getResponse().ok()); assert(res.getResponse().ok());
List<Long> ids = res.getEntityIds(); List<Long> ids = res.getEntityIds();
@ -37,7 +26,7 @@ public class TestCompact {
@Test(dataProvider = "BinaryCollection", dataProviderClass = MainClass.class) @Test(dataProvider = "BinaryCollection", dataProviderClass = MainClass.class)
public void testCompactAfterDeleteBinary(MilvusClient client, String collectionName) { public void testCompactAfterDeleteBinary(MilvusClient client, String collectionName) {
InsertParam insertParam = new InsertParam.Builder(collectionName).withFields(defaultBinaryEntities).build(); InsertParam insertParam = new InsertParam.Builder(collectionName).withFields(Constants.defaultBinaryEntities).build();
InsertResponse res = client.insert(insertParam); InsertResponse res = client.insert(insertParam);
assert(res.getResponse().ok()); assert(res.getResponse().ok());
List<Long> ids = res.getEntityIds(); List<Long> ids = res.getEntityIds();

39
tests/milvus-java-test/src/main/java/com/TestFlush.java
View File

@ -1,33 +1,19 @@
package com; package com;
import com.google.common.util.concurrent.ListenableFuture; import com.google.common.util.concurrent.ListenableFuture;
import java.util.Map;
import java.util.concurrent.ExecutionException; import java.util.concurrent.ExecutionException;
import io.milvus.client.*; import io.milvus.client.*;
import org.apache.commons.lang3.RandomStringUtils; import org.apache.commons.lang3.RandomStringUtils;
import org.testng.Assert; import org.testng.Assert;
import org.testng.annotations.Test; import org.testng.annotations.Test;
import java.nio.ByteBuffer;
import java.util.ArrayList; import java.util.ArrayList;
import java.util.List; import java.util.List;
public class TestFlush { public class TestFlush {
int segmentRowCount = 50; int segmentRowCount = 50;
int dimension = 128; int nb = Constants.nb;
int nb = 8000;
List<List<Float>> vectors = Utils.genVectors(nb, dimension, true); @Test(dataProvider = "ConnectInstance", dataProviderClass = MainClass.class)
List<ByteBuffer> vectorsBinary = Utils.genBinaryVectors(nb, dimension);
List<Map<String,Object>> defaultFields = Utils.genDefaultFields(dimension,false);
List<Map<String,Object>> defaultEntities = Utils.genDefaultEntities(dimension,nb,vectors);
List<Map<String,Object>> defaultBinaryEntities = Utils.genDefaultBinaryEntities(dimension,nb,vectorsBinary);
@Test(dataProvider = "Collection", dataProviderClass = MainClass.class)
public void testFlushCollectionNotExisted(MilvusClient client, String collectionName) { public void testFlushCollectionNotExisted(MilvusClient client, String collectionName) {
String newCollection = "not_existed"; String newCollection = "not_existed";
Response res = client.flush(newCollection); Response res = client.flush(newCollection);
@ -40,39 +26,40 @@ public class TestFlush {
assert(res.ok()); assert(res.ok());
} }
@Test(dataProvider = "Collection", dataProviderClass = MainClass.class) @Test(dataProvider = "ConnectInstance", dataProviderClass = MainClass.class)
public void testAddCollectionsFlush(MilvusClient client, String collectionName) { public void testAddCollectionsFlush(MilvusClient client, String collectionName) {
List<String> names = new ArrayList<>(); List<String> names = new ArrayList<>();
for (int i = 0; i < 10; i++) { int collectionNum = 10;
for (int i = 0; i < collectionNum; i++) {
names.add(RandomStringUtils.randomAlphabetic(10)); names.add(RandomStringUtils.randomAlphabetic(10));
CollectionMapping collectionSchema = new CollectionMapping.Builder(names.get(i)) CollectionMapping collectionSchema = new CollectionMapping.Builder(names.get(i))
.withFields(defaultFields) .withFields(Constants.defaultFields)
.withParamsInJson(String.format("{\"segment_row_count\": %s}",segmentRowCount)) .withParamsInJson(String.format("{\"segment_row_count\": %s}",segmentRowCount))
.build(); .build();
client.createCollection(collectionSchema); client.createCollection(collectionSchema);
InsertParam insertParam = new InsertParam.Builder(names.get(i)).withFields(defaultEntities).build(); InsertParam insertParam = new InsertParam.Builder(names.get(i)).withFields(Constants.defaultEntities).build();
client.insert(insertParam); client.insert(insertParam);
System.out.println("Table " + names.get(i) + " created."); System.out.println("Table " + names.get(i) + " created.");
} }
Response res = client.flush(names); Response res = client.flush(names);
assert(res.ok()); assert(res.ok());
for (int i = 0; i < 10; i++) { for (int i = 0; i < collectionNum; i++) {
// check row count // check row count
Assert.assertEquals(client.countEntities(names.get(i)).getCollectionEntityCount(), nb); Assert.assertEquals(client.countEntities(names.get(i)).getCollectionEntityCount(), nb);
} }
} }
@Test(dataProvider = "Collection", dataProviderClass = MainClass.class) @Test(dataProvider = "ConnectInstance", dataProviderClass = MainClass.class)
public void testAddCollectionsFlushAsync(MilvusClient client, String collectionName) throws ExecutionException, InterruptedException { public void testAddCollectionsFlushAsync(MilvusClient client, String collectionName) throws ExecutionException, InterruptedException {
List<String> names = new ArrayList<>(); List<String> names = new ArrayList<>();
for (int i = 0; i < 100; i++) { for (int i = 0; i < 100; i++) {
names.add(RandomStringUtils.randomAlphabetic(10)); names.add(RandomStringUtils.randomAlphabetic(10));
CollectionMapping collectionSchema = new CollectionMapping.Builder(names.get(i)) CollectionMapping collectionSchema = new CollectionMapping.Builder(names.get(i))
.withFields(defaultFields) .withFields(Constants.defaultFields)
.withParamsInJson(String.format("{\"segment_row_count\": %s}",segmentRowCount)) .withParamsInJson(String.format("{\"segment_row_count\": %s}",segmentRowCount))
.build(); .build();
client.createCollection(collectionSchema); client.createCollection(collectionSchema);
InsertParam insertParam = new InsertParam.Builder(names.get(i)).withFields(defaultEntities).build(); InsertParam insertParam = new InsertParam.Builder(names.get(i)).withFields(Constants.defaultEntities).build();
client.insert(insertParam); client.insert(insertParam);
System.out.println("Collection " + names.get(i) + " created."); System.out.println("Collection " + names.get(i) + " created.");
} }
@ -87,7 +74,7 @@ public class TestFlush {
@Test(dataProvider = "Collection", dataProviderClass = MainClass.class) @Test(dataProvider = "Collection", dataProviderClass = MainClass.class)
public void testAddFlushMultipleTimes(MilvusClient client, String collectionName) { public void testAddFlushMultipleTimes(MilvusClient client, String collectionName) {
for (int i = 0; i < 10; i++) { for (int i = 0; i < 10; i++) {
InsertParam insertParam = new InsertParam.Builder(collectionName).withFields(defaultEntities).build(); InsertParam insertParam = new InsertParam.Builder(collectionName).withFields(Constants.defaultEntities).build();
client.insert(insertParam); client.insert(insertParam);
Response res = client.flush(collectionName); Response res = client.flush(collectionName);
assert(res.ok()); assert(res.ok());
@ -98,7 +85,7 @@ public class TestFlush {
@Test(dataProvider = "BinaryCollection", dataProviderClass = MainClass.class) @Test(dataProvider = "BinaryCollection", dataProviderClass = MainClass.class)
public void testAddFlushMultipleTimesBinary(MilvusClient client, String collectionName) { public void testAddFlushMultipleTimesBinary(MilvusClient client, String collectionName) {
for (int i = 0; i < 10; i++) { for (int i = 0; i < 10; i++) {
InsertParam insertParam = new InsertParam.Builder(collectionName).withFields(defaultBinaryEntities).build(); InsertParam insertParam = new InsertParam.Builder(collectionName).withFields(Constants.defaultBinaryEntities).build();
client.insert(insertParam); client.insert(insertParam);
Response res = client.flush(collectionName); Response res = client.flush(collectionName);
assert(res.ok()); assert(res.ok());

184
tests/milvus-java-test/src/main/java/com/TestGetEntityByID.java
View File

@ -1,97 +1,89 @@
//package com; package com;
//
//import io.milvus.client.*; import io.milvus.client.*;
//import org.testng.annotations.Test; import org.testng.annotations.Test;
//
//import java.nio.ByteBuffer; import java.util.Collections;
//import java.util.ArrayList; import java.util.List;
//import java.util.Collections; import java.util.Map;
//import java.util.List;
//import java.util.Map; public class TestGetEntityByID {
// public List<Long> get_ids = Utils.toListIds(1111);
//public class TestGetEntityByID {
// int dimension = 128; @Test(dataProvider = "Collection", dataProviderClass = MainClass.class)
// int nb = 8000; public void testGetEntitiesByIdValid(MilvusClient client, String collectionName) {
// public List<Long> get_ids = Utils.toListIds(1111); int get_length = 100;
//// List<List<Float>> vectors = Utils.genVectors(nb, dimension, true); InsertParam insertParam = new InsertParam.Builder(collectionName).withFields(Constants.defaultEntities).build();
//// List<ByteBuffer> vectorsBinary = Utils.genBinaryVectors(nb, dimension); InsertResponse resInsert = client.insert(insertParam);
// List<Map<String,Object>> defaultEntities = Utils.genDefaultEntities(dimension,nb,false); List<Long> ids = resInsert.getEntityIds();
// List<Map<String,Object>> defaultBinaryEntities = Utils.genDefaultEntities(dimension,nb,true); client.flush(collectionName);
// GetEntityByIDResponse res = client.getEntityByID(collectionName, ids.subList(0, get_length));
// @Test(dataProvider = "Collection", dataProviderClass = MainClass.class) assert (res.getResponse().ok());
// public void test_get_vector_by_id_valid(MilvusClient client, String collectionName) {
// int get_length = 100;
// InsertParam insertParam = new InsertParam.Builder(collectionName).withFields(defaultEntities).build();
// InsertResponse resInsert = client.insert(insertParam);
// List<Long> ids = resInsert.getEntityIds();
// client.flush(collectionName);
// GetEntityByIDResponse res = client.getEntityByID(collectionName, ids.subList(0, get_length));
// assert (res.getResponse().ok());
// assert (res.getValidIds(), ids.subList(0, get_length)); // assert (res.getValidIds(), ids.subList(0, get_length));
// for (int i = 0; i < get_length; i++) { for (int i = 0; i < get_length; i++) {
// List<Map<String,Object>> fieldsMap = res.getFieldsMap(); List<Map<String,Object>> fieldsMap = res.getFieldsMap();
// assert (fieldsMap.get(i).get("float_vector").equals(defaultEntities.get(defaultEntities.size()-1).get("values").get("float_vector"))); assert (fieldsMap.get(i).get("float_vector").equals(Constants.vectors.get(i)));
// } }
// } }
//
// @Test(dataProvider = "Collection", dataProviderClass = MainClass.class) @Test(dataProvider = "Collection", dataProviderClass = MainClass.class)
// public void test_get_vector_by_id_after_delete(MilvusClient client, String collectionName) { public void testGetEntityByIdAfterDelete(MilvusClient client, String collectionName) {
// InsertParam insertParam = new InsertParam.Builder(collectionName).withFloatVectors(vectors).build(); InsertParam insertParam = new InsertParam.Builder(collectionName).withFields(Constants.defaultEntities).build();
// InsertResponse resInsert = client.insert(insertParam); InsertResponse resInsert = client.insert(insertParam);
// List<Long> ids = resInsert.getVectorIds(); List<Long> ids = resInsert.getEntityIds();
// Response res_delete = client.deleteEntityByID(collectionName, Collections.singletonList(ids.get(0))); Response res_delete = client.deleteEntityByID(collectionName, Collections.singletonList(ids.get(0)));
// assert(res_delete.ok()); assert(res_delete.ok());
// client.flush(collectionName); client.flush(collectionName);
// GetEntityByIDResponse res = client.getEntityByID(collectionName, ids.subList(0, 1)); GetEntityByIDResponse res = client.getEntityByID(collectionName, ids.subList(0, 1));
// assert (res.getResponse().ok()); assert (res.getResponse().ok());
// assert (res.getFloatVectors().get(0).size() == 0); assert (res.getFieldsMap().size() == 0);
// } }
//
// @Test(dataProvider = "Collection", dataProviderClass = MainClass.class) @Test(dataProvider = "ConnectInstance", dataProviderClass = MainClass.class)
// public void test_get_vector_by_id_collection_name_not_existed(MilvusClient client, String collectionName) { public void testGetEntityByIdCollectionNameNotExisted(MilvusClient client, String collectionName) {
// String newCollection = "not_existed"; String newCollection = "not_existed";
// GetEntityByIDResponse res = client.getEntityByID(newCollection, get_ids); GetEntityByIDResponse res = client.getEntityByID(newCollection, get_ids);
// assert(!res.getResponse().ok()); assert(!res.getResponse().ok());
// } }
//
// @Test(dataProvider = "Collection", dataProviderClass = MainClass.class) @Test(dataProvider = "Collection", dataProviderClass = MainClass.class)
// public void test_get_vector_id_not_existed(MilvusClient client, String collectionName) { public void testGetVectorIdNotExisted(MilvusClient client, String collectionName) {
// InsertParam insertParam = new InsertParam.Builder(collectionName).withFloatVectors(vectors).build(); InsertParam insertParam = new InsertParam.Builder(collectionName).withFields(Constants.defaultEntities).build();
// client.insert(insertParam); client.insert(insertParam);
// client.flush(collectionName); client.flush(collectionName);
// GetEntityByIDResponse res = client.getEntityByID(collectionName, get_ids); GetEntityByIDResponse res = client.getEntityByID(collectionName, get_ids);
// assert (res.getFloatVectors().get(0).size() == 0); assert (res.getFieldsMap().size() == 0);
// } }
//
// // Binary tests // Binary tests
// @Test(dataProvider = "BinaryCollection", dataProviderClass = MainClass.class) @Test(dataProvider = "BinaryCollection", dataProviderClass = MainClass.class)
// public void test_get_vector_by_id_valid_binary(MilvusClient client, String collectionName) { public void testGetEntityByIdValidBinary(MilvusClient client, String collectionName) {
// InsertParam insertParam = new InsertParam.Builder(collectionName).withBinaryVectors(vectorsBinary).build(); InsertParam insertParam = new InsertParam.Builder(collectionName).withFields(Constants.defaultBinaryEntities).build();
// InsertResponse resInsert = client.insert(insertParam); InsertResponse resInsert = client.insert(insertParam);
// List<Long> ids = resInsert.getVectorIds(); List<Long> ids = resInsert.getEntityIds();
// client.flush(collectionName); client.flush(collectionName);
// GetEntityByIDResponse res = client.getEntityByID(collectionName, ids.subList(0, 1)); GetEntityByIDResponse res = client.getEntityByID(collectionName, ids.subList(0, 1));
// assert res.getBinaryVectors().get(0).equals(vectorsBinary.get(0).rewind()); assert res.getFieldsMap().get(0).get(Constants.binaryFieldName).equals(Constants.vectorsBinary.get(0).rewind());
// } }
//
// @Test(dataProvider = "BinaryCollection", dataProviderClass = MainClass.class) @Test(dataProvider = "BinaryCollection", dataProviderClass = MainClass.class)
// public void test_get_vector_by_id_after_delete_binary(MilvusClient client, String collectionName) { public void testGetEntityByIdAfterDeleteBinary(MilvusClient client, String collectionName) {
// InsertParam insertParam = new InsertParam.Builder(collectionName).withBinaryVectors(vectorsBinary).build(); InsertParam insertParam = new InsertParam.Builder(collectionName).withFields(Constants.defaultBinaryEntities).build();
// InsertResponse resInsert = client.insert(insertParam); InsertResponse resInsert = client.insert(insertParam);
// List<Long> ids = resInsert.getVectorIds(); List<Long> ids = resInsert.getEntityIds();
// Response res_delete = client.deleteEntityByID(collectionName, Collections.singletonList(ids.get(0))); Response res_delete = client.deleteEntityByID(collectionName, Collections.singletonList(ids.get(0)));
// assert(res_delete.ok()); assert(res_delete.ok());
// client.flush(collectionName); client.flush(collectionName);
// GetEntityByIDResponse res = client.getEntityByID(collectionName, ids.subList(0, 1)); GetEntityByIDResponse res = client.getEntityByID(collectionName, ids.subList(0, 1));
// assert (res.getFloatVectors().get(0).size() == 0); assert (res.getFieldsMap().size() == 0);
// } }
//
// @Test(dataProvider = "BinaryCollection", dataProviderClass = MainClass.class) @Test(dataProvider = "BinaryCollection", dataProviderClass = MainClass.class)
// public void test_get_vector_id_not_existed_binary(MilvusClient client, String collectionName) { public void testGetEntityIdNotExistedBinary(MilvusClient client, String collectionName) {
// InsertParam insertParam = new InsertParam.Builder(collectionName).withBinaryVectors(vectorsBinary).build(); InsertParam insertParam = new InsertParam.Builder(collectionName).withFields(Constants.defaultBinaryEntities).build();
// client.insert(insertParam); client.insert(insertParam);
// client.flush(collectionName); client.flush(collectionName);
// GetEntityByIDResponse res = client.getEntityByID(collectionName, get_ids); GetEntityByIDResponse res = client.getEntityByID(collectionName, get_ids);
// assert (res.getFloatVectors().get(0).size() == 0); assert (res.getFieldsMap().size() == 0);
// } }
//} }