diff --git a/tests/milvus-java-test/MilvusSDkJavaTest.iml b/tests/milvus-java-test/MilvusSDkJavaTest.iml index 953b98ba92..cac27b0e7c 100644 --- a/tests/milvus-java-test/MilvusSDkJavaTest.iml +++ b/tests/milvus-java-test/MilvusSDkJavaTest.iml @@ -9,13 +9,12 @@ - - + - + diff --git a/tests/milvus-java-test/src/main/java/com/Partition.java b/tests/milvus-java-test/src/main/java/com/Partition.java deleted file mode 100644 index 89a12663ea..0000000000 --- a/tests/milvus-java-test/src/main/java/com/Partition.java +++ /dev/null @@ -1,205 +0,0 @@ -package com; - -import io.milvus.client.MilvusClient; -import io.milvus.client.Response; -import io.milvus.client.ShowPartitionsResponse; -import org.apache.commons.lang3.RandomStringUtils; -import org.testng.Assert; -import org.testng.annotations.Test; - -import java.util.ArrayList; -import java.util.LinkedList; -import java.util.List; -import java.util.Random; - -public class Partition { - int dimension = 128; - - public List> gen_vectors(Integer nb) { - List> xb = new LinkedList<>(); - Random random = new Random(); - for (int i = 0; i < nb; ++i) { - LinkedList vector = new LinkedList<>(); - for (int j = 0; j < dimension; j++) { - vector.add(random.nextFloat()); - } - xb.add(vector); - } - return xb; - } - - // ----------------------------- create partition cases in --------------------------------- - - // create partition - @Test(dataProvider = "Table", dataProviderClass = MainClass.class) - public void test_create_partition(MilvusClient client, String tableName) { - String partitionName = RandomStringUtils.randomAlphabetic(10); - String tag = RandomStringUtils.randomAlphabetic(10); - io.milvus.client.Partition partition = new io.milvus.client.Partition.Builder(tableName, partitionName, tag).build(); - Response createpResponse = client.createPartition(partition); - assert (createpResponse.ok()); - // show partitions - List partitions = client.showPartitions(tableName).getPartitionList(); - System.out.println(partitions); - List partitionNames = new ArrayList<>(); - for (int i=0; i partitionNames = new ArrayList<>(); - for (int i=0; i partitionNames = response.getPartitionNameList(); -// for (int i=0; i tagNames = response.getPartitionTagList(); - Assert.assertTrue(tagNames.contains(tag)); - Assert.assertTrue(tagNames.contains(tagNew)); - } - -} diff --git a/tests/milvus-java-test/src/main/java/com/TestSearchByIds.java b/tests/milvus-java-test/src/main/java/com/TestSearchByIds.java index 51f0a0b13f..94a8cb2498 100644 --- a/tests/milvus-java-test/src/main/java/com/TestSearchByIds.java +++ b/tests/milvus-java-test/src/main/java/com/TestSearchByIds.java @@ -61,9 +61,9 @@ public class TestSearchByIds { .build(); List> res_search = client.searchByIds(searchParam).getQueryResultsList(); assert (client.searchByIds(searchParam).getResponse().ok()); - Assert.assertEquals(res_search.get(0).size(), top_k); - Assert.assertEquals(res_search.size(), default_ids.size()); - Assert.assertEquals(res_search.get(0).get(0).getVectorId(), -1); + Assert.assertEquals(res_search.get(0).size(), 0); +// Assert.assertEquals(res_search.size(), default_ids.size()); +// Assert.assertEquals(res_search.get(0).get(0).getVectorId(), -1); } @Test(dataProvider = "Collection", dataProviderClass = MainClass.class) diff --git a/tests/milvus-java-test/src/main/java/com/TestTable.java b/tests/milvus-java-test/src/main/java/com/TestTable.java deleted file mode 100644 index e722db23df..0000000000 --- a/tests/milvus-java-test/src/main/java/com/TestTable.java +++ /dev/null @@ -1,142 +0,0 @@ -package com; - - -import io.milvus.client.*; -import org.testng.Assert; -import org.testng.annotations.Test; - -import java.util.List; - -public class TestTable { - int index_file_size = 50; - int dimension = 128; - - @Test(dataProvider = "ConnectInstance", dataProviderClass = MainClass.class) - public void test_create_table(MilvusClient client, String tableName){ - TableSchema tableSchema = new TableSchema.Builder(tableName, dimension) - .withIndexFileSize(index_file_size) - .withMetricType(MetricType.L2) - .build(); - Response res = client.createTable(tableSchema); - assert(res.ok()); - Assert.assertEquals(res.ok(), true); - } - - @Test(dataProvider = "DisConnectInstance", dataProviderClass = MainClass.class) - public void test_create_table_disconnect(MilvusClient client, String tableName){ - TableSchema tableSchema = new TableSchema.Builder(tableName, dimension) - .withIndexFileSize(index_file_size) - .withMetricType(MetricType.L2) - .build(); - Response res = client.createTable(tableSchema); - assert(!res.ok()); - } - - @Test(dataProvider = "ConnectInstance", dataProviderClass = MainClass.class) - public void test_create_table_repeatably(MilvusClient client, String tableName){ - TableSchema tableSchema = new TableSchema.Builder(tableName, dimension) - .withIndexFileSize(index_file_size) - .withMetricType(MetricType.L2) - .build(); - Response res = client.createTable(tableSchema); - Assert.assertEquals(res.ok(), true); - Response res_new = client.createTable(tableSchema); - Assert.assertEquals(res_new.ok(), false); - } - - @Test(dataProvider = "ConnectInstance", dataProviderClass = MainClass.class) - public void test_create_table_wrong_params(MilvusClient client, String tableName){ - Integer dimension = 0; - TableSchema tableSchema = new TableSchema.Builder(tableName, dimension) - .withIndexFileSize(index_file_size) - .withMetricType(MetricType.L2) - .build(); - Response res = client.createTable(tableSchema); - System.out.println(res.toString()); - Assert.assertEquals(res.ok(), false); - } - - @Test(dataProvider = "ConnectInstance", dataProviderClass = MainClass.class) - public void test_show_tables(MilvusClient client, String tableName){ - Integer tableNum = 10; - ShowTablesResponse res = null; - for (int i = 0; i < tableNum; ++i) { - String tableNameNew = tableName+"_"+Integer.toString(i); - TableSchema tableSchema = new TableSchema.Builder(tableNameNew, dimension) - .withIndexFileSize(index_file_size) - .withMetricType(MetricType.L2) - .build(); - client.createTable(tableSchema); - List tableNames = client.showTables().getTableNames(); - Assert.assertTrue(tableNames.contains(tableNameNew)); - } - } - - @Test(dataProvider = "DisConnectInstance", dataProviderClass = MainClass.class) - public void test_show_tables_without_connect(MilvusClient client, String tableName){ - ShowTablesResponse res = client.showTables(); - assert(!res.getResponse().ok()); - } - - @Test(dataProvider = "Table", dataProviderClass = MainClass.class) - public void test_drop_table(MilvusClient client, String tableName) throws InterruptedException { - Response res = client.dropTable(tableName); - assert(res.ok()); - Thread.currentThread().sleep(1000); - List tableNames = client.showTables().getTableNames(); - Assert.assertFalse(tableNames.contains(tableName)); - } - - @Test(dataProvider = "Table", dataProviderClass = MainClass.class) - public void test_drop_table_not_existed(MilvusClient client, String tableName) throws InterruptedException { - Response res = client.dropTable(tableName+"_"); - assert(!res.ok()); - List tableNames = client.showTables().getTableNames(); - Assert.assertTrue(tableNames.contains(tableName)); - } - - @Test(dataProvider = "DisConnectInstance", dataProviderClass = MainClass.class) - public void test_drop_table_without_connect(MilvusClient client, String tableName) throws InterruptedException { - Response res = client.dropTable(tableName); - assert(!res.ok()); - } - - @Test(dataProvider = "Table", dataProviderClass = MainClass.class) - public void test_describe_table(MilvusClient client, String tableName) throws InterruptedException { - DescribeTableResponse res = client.describeTable(tableName); - assert(res.getResponse().ok()); - TableSchema tableSchema = res.getTableSchema().get(); - Assert.assertEquals(tableSchema.getDimension(), dimension); - Assert.assertEquals(tableSchema.getTableName(), tableName); - Assert.assertEquals(tableSchema.getIndexFileSize(), index_file_size); - Assert.assertEquals(tableSchema.getMetricType().name(), tableName.substring(0,2)); - } - - @Test(dataProvider = "DisConnectInstance", dataProviderClass = MainClass.class) - public void test_describe_table_without_connect(MilvusClient client, String tableName) throws InterruptedException { - DescribeTableResponse res = client.describeTable(tableName); - assert(!res.getResponse().ok()); - } - - @Test(dataProvider = "Table", dataProviderClass = MainClass.class) - public void test_has_table_not_existed(MilvusClient client, String tableName) throws InterruptedException { - HasTableResponse res = client.hasTable(tableName+"_"); - assert(res.getResponse().ok()); - Assert.assertFalse(res.hasTable()); - } - - @Test(dataProvider = "DisConnectInstance", dataProviderClass = MainClass.class) - public void test_has_table_without_connect(MilvusClient client, String tableName) throws InterruptedException { - HasTableResponse res = client.hasTable(tableName); - assert(!res.getResponse().ok()); - } - - @Test(dataProvider = "Table", dataProviderClass = MainClass.class) - public void test_has_table(MilvusClient client, String tableName) throws InterruptedException { - HasTableResponse res = client.hasTable(tableName); - assert(res.getResponse().ok()); - Assert.assertTrue(res.hasTable()); - } - - -} diff --git a/tests/milvus-java-test/src/main/java/com/TestTableCount.java b/tests/milvus-java-test/src/main/java/com/TestTableCount.java deleted file mode 100644 index 5cda18e812..0000000000 --- a/tests/milvus-java-test/src/main/java/com/TestTableCount.java +++ /dev/null @@ -1,83 +0,0 @@ -package com; - -import io.milvus.client.*; -import org.testng.Assert; -import org.testng.annotations.Test; - -import java.util.ArrayList; -import java.util.List; -import java.util.Random; - -public class TestTableCount { - int index_file_size = 50; - int dimension = 128; - - public List> gen_vectors(Integer nb) { - List> xb = new ArrayList<>(); - Random random = new Random(); - for (int i = 0; i < nb; ++i) { - ArrayList vector = new ArrayList<>(); - for (int j = 0; j < dimension; j++) { - vector.add(random.nextFloat()); - } - xb.add(vector); - } - return xb; - } - - @Test(dataProvider = "Table", dataProviderClass = MainClass.class) - public void test_table_count_no_vectors(MilvusClient client, String tableName) { - Assert.assertEquals(client.getTableRowCount(tableName).getTableRowCount(), 0); - } - - @Test(dataProvider = "Table", dataProviderClass = MainClass.class) - public void test_table_count_table_not_existed(MilvusClient client, String tableName) { - GetTableRowCountResponse res = client.getTableRowCount(tableName+"_"); - assert(!res.getResponse().ok()); - } - - @Test(dataProvider = "DisConnectInstance", dataProviderClass = MainClass.class) - public void test_table_count_without_connect(MilvusClient client, String tableName) { - GetTableRowCountResponse res = client.getTableRowCount(tableName+"_"); - assert(!res.getResponse().ok()); - } - - @Test(dataProvider = "Table", dataProviderClass = MainClass.class) - public void test_table_count(MilvusClient client, String tableName) throws InterruptedException { - int nb = 10000; - List> vectors = gen_vectors(nb); - // Add vectors - InsertParam insertParam = new InsertParam.Builder(tableName, vectors).build();; - client.insert(insertParam); - Thread.currentThread().sleep(2000); - Assert.assertEquals(client.getTableRowCount(tableName).getTableRowCount(), nb); - } - - @Test(dataProvider = "Table", dataProviderClass = MainClass.class) - public void test_table_count_multi_tables(MilvusClient client, String tableName) throws InterruptedException { - int nb = 10000; - List> vectors = gen_vectors(nb); - Integer tableNum = 10; - GetTableRowCountResponse res = null; - for (int i = 0; i < tableNum; ++i) { - String tableNameNew = tableName + "_" + Integer.toString(i); - TableSchema tableSchema = new TableSchema.Builder(tableNameNew, dimension) - .withIndexFileSize(index_file_size) - .withMetricType(MetricType.L2) - .build(); - client.createTable(tableSchema); - // Add vectors - InsertParam insertParam = new InsertParam.Builder(tableNameNew, vectors).build(); - client.insert(insertParam); - } - Thread.currentThread().sleep(1000); - for (int i = 0; i < tableNum; ++i) { - String tableNameNew = tableName + "_" + Integer.toString(i); - res = client.getTableRowCount(tableNameNew); - Assert.assertEquals(res.getTableRowCount(), nb); - } - } - -} - -