test(discov2): update unit tests for new methods

Author: apaparazzi0329

discovery/src/test/java/com/ibm/watson/discovery/v2/DiscoveryTest.java

@@ -0,0 +1,49 @@
+/*
+ * (C) Copyright IBM Corp. 2022.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with
+ * the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
+ * an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
+ * specific language governing permissions and limitations under the License.
+ */
+
+package com.ibm.watson.discovery.v2.model;
+
+import static org.testng.Assert.*;
+
+import com.ibm.cloud.sdk.core.service.model.FileWithMetadata;
+import com.ibm.watson.discovery.v2.utils.TestUtilities;
+import java.io.InputStream;
+import java.util.HashMap;
+import java.util.List;
+import org.testng.annotations.Test;
+
+/** Unit test class for the ClassifierFederatedModel model. */
+public class ClassifierFederatedModelTest {
+  final HashMap<String, InputStream> mockStreamMap = TestUtilities.createMockStreamMap();
+  final List<FileWithMetadata> mockListFileWithMetadata =
+      TestUtilities.creatMockListFileWithMetadata();
+
+  @Test
+  public void testClassifierFederatedModel() throws Throwable {
+    ClassifierFederatedModel classifierFederatedModelModel =
+        new ClassifierFederatedModel.Builder().field("testString").build();
+    assertEquals(classifierFederatedModelModel.field(), "testString");
+
+    String json = TestUtilities.serialize(classifierFederatedModelModel);
+
+    ClassifierFederatedModel classifierFederatedModelModelNew =
+        TestUtilities.deserialize(json, ClassifierFederatedModel.class);
+    assertTrue(classifierFederatedModelModelNew instanceof ClassifierFederatedModel);
+    assertEquals(classifierFederatedModelModelNew.field(), "testString");
+  }
+
+  @Test(expectedExceptions = IllegalArgumentException.class)
+  public void testClassifierFederatedModelError() throws Throwable {
+    new ClassifierFederatedModel.Builder().build();
+  }
+}

discovery/src/test/java/com/ibm/watson/discovery/v2/model/ClassifierFederatedModelTest.java

@@ -0,0 +1,38 @@
+/*
+ * (C) Copyright IBM Corp. 2022.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with
+ * the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
+ * an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
+ * specific language governing permissions and limitations under the License.
+ */
+
+package com.ibm.watson.discovery.v2.model;
+
+import static org.testng.Assert.*;
+
+import com.ibm.cloud.sdk.core.service.model.FileWithMetadata;
+import com.ibm.watson.discovery.v2.utils.TestUtilities;
+import java.io.InputStream;
+import java.util.HashMap;
+import java.util.List;
+import org.testng.annotations.Test;
+
+/** Unit test class for the ClassifierModelEvaluation model. */
+public class ClassifierModelEvaluationTest {
+  final HashMap<String, InputStream> mockStreamMap = TestUtilities.createMockStreamMap();
+  final List<FileWithMetadata> mockListFileWithMetadata =
+      TestUtilities.creatMockListFileWithMetadata();
+
+  @Test
+  public void testClassifierModelEvaluation() throws Throwable {
+    ClassifierModelEvaluation classifierModelEvaluationModel = new ClassifierModelEvaluation();
+    assertNull(classifierModelEvaluationModel.getMicroAverage());
+    assertNull(classifierModelEvaluationModel.getMacroAverage());
+    assertNull(classifierModelEvaluationModel.getPerClass());
+  }
+}

discovery/src/test/java/com/ibm/watson/discovery/v2/model/ClassifierModelEvaluationTest.java

@@ -0,0 +1,53 @@
+/*
+ * (C) Copyright IBM Corp. 2022.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with
+ * the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
+ * an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
+ * specific language governing permissions and limitations under the License.
+ */
+
+package com.ibm.watson.discovery.v2.model;
+
+import static org.testng.Assert.*;
+
+import com.ibm.cloud.sdk.core.service.model.FileWithMetadata;
+import com.ibm.watson.discovery.v2.utils.TestUtilities;
+import java.io.InputStream;
+import java.util.HashMap;
+import java.util.List;
+import org.testng.annotations.Test;
+
+/** Unit test class for the CollectionDetailsSmartDocumentUnderstanding model. */
+public class CollectionDetailsSmartDocumentUnderstandingTest {
+  final HashMap<String, InputStream> mockStreamMap = TestUtilities.createMockStreamMap();
+  final List<FileWithMetadata> mockListFileWithMetadata =
+      TestUtilities.creatMockListFileWithMetadata();
+
+  @Test
+  public void testCollectionDetailsSmartDocumentUnderstanding() throws Throwable {
+    CollectionDetailsSmartDocumentUnderstanding collectionDetailsSmartDocumentUnderstandingModel =
+        new CollectionDetailsSmartDocumentUnderstanding.Builder()
+            .enabled(true)
+            .model("custom")
+            .build();
+    assertEquals(collectionDetailsSmartDocumentUnderstandingModel.enabled(), Boolean.valueOf(true));
+    assertEquals(collectionDetailsSmartDocumentUnderstandingModel.model(), "custom");
+
+    String json = TestUtilities.serialize(collectionDetailsSmartDocumentUnderstandingModel);
+
+    CollectionDetailsSmartDocumentUnderstanding
+        collectionDetailsSmartDocumentUnderstandingModelNew =
+            TestUtilities.deserialize(json, CollectionDetailsSmartDocumentUnderstanding.class);
+    assertTrue(
+        collectionDetailsSmartDocumentUnderstandingModelNew
+            instanceof CollectionDetailsSmartDocumentUnderstanding);
+    assertEquals(
+        collectionDetailsSmartDocumentUnderstandingModelNew.enabled(), Boolean.valueOf(true));
+    assertEquals(collectionDetailsSmartDocumentUnderstandingModelNew.model(), "custom");
+  }
+}

discovery/src/test/java/com/ibm/watson/discovery/v2/model/CollectionDetailsSmartDocumentUnderstandingTest.java

@@ -1,5 +1,5 @@
 /*
- * (C) Copyright IBM Corp. 2020, 2021.
+ * (C) Copyright IBM Corp. 2022.
  *
  * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with
  * the License. You may obtain a copy of the License at
@@ -33,29 +33,35 @@ public void testCollectionDetails() throws Throwable {
     CollectionEnrichment collectionEnrichmentModel =
         new CollectionEnrichment.Builder()
             .enrichmentId("testString")
-            .fields(new java.util.ArrayList<String>(java.util.Arrays.asList("testString")))
+            .fields(java.util.Arrays.asList("testString"))
             .build();
     assertEquals(collectionEnrichmentModel.enrichmentId(), "testString");
-    assertEquals(
-        collectionEnrichmentModel.fields(),
-        new java.util.ArrayList<String>(java.util.Arrays.asList("testString")));
+    assertEquals(collectionEnrichmentModel.fields(), java.util.Arrays.asList("testString"));
+
+    CollectionDetailsSmartDocumentUnderstanding collectionDetailsSmartDocumentUnderstandingModel =
+        new CollectionDetailsSmartDocumentUnderstanding.Builder()
+            .enabled(true)
+            .model("custom")
+            .build();
+    assertEquals(collectionDetailsSmartDocumentUnderstandingModel.enabled(), Boolean.valueOf(true));
+    assertEquals(collectionDetailsSmartDocumentUnderstandingModel.model(), "custom");
 
     CollectionDetails collectionDetailsModel =
         new CollectionDetails.Builder()
             .name("testString")
             .description("testString")
             .language("en")
-            .enrichments(
-                new java.util.ArrayList<CollectionEnrichment>(
-                    java.util.Arrays.asList(collectionEnrichmentModel)))
+            .enrichments(java.util.Arrays.asList(collectionEnrichmentModel))
+            .smartDocumentUnderstanding(collectionDetailsSmartDocumentUnderstandingModel)
             .build();
     assertEquals(collectionDetailsModel.name(), "testString");
     assertEquals(collectionDetailsModel.description(), "testString");
     assertEquals(collectionDetailsModel.language(), "en");
     assertEquals(
-        collectionDetailsModel.enrichments(),
-        new java.util.ArrayList<CollectionEnrichment>(
-            java.util.Arrays.asList(collectionEnrichmentModel)));
+        collectionDetailsModel.enrichments(), java.util.Arrays.asList(collectionEnrichmentModel));
+    assertEquals(
+        collectionDetailsModel.smartDocumentUnderstanding(),
+        collectionDetailsSmartDocumentUnderstandingModel);
 
     String json = TestUtilities.serialize(collectionDetailsModel);
 
@@ -65,6 +71,9 @@ public void testCollectionDetails() throws Throwable {
     assertEquals(collectionDetailsModelNew.name(), "testString");
     assertEquals(collectionDetailsModelNew.description(), "testString");
     assertEquals(collectionDetailsModelNew.language(), "en");
+    assertEquals(
+        collectionDetailsModelNew.smartDocumentUnderstanding().toString(),
+        collectionDetailsSmartDocumentUnderstandingModel.toString());
   }
 
   @Test(expectedExceptions = IllegalArgumentException.class)

discovery/src/test/java/com/ibm/watson/discovery/v2/model/CollectionDetailsTest.java

@@ -1,5 +1,5 @@
 /*
- * (C) Copyright IBM Corp. 2020.
+ * (C) Copyright IBM Corp. 2022.
  *
  * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with
  * the License. You may obtain a copy of the License at
@@ -33,12 +33,10 @@ public void testCollectionEnrichment() throws Throwable {
     CollectionEnrichment collectionEnrichmentModel =
         new CollectionEnrichment.Builder()
             .enrichmentId("testString")
-            .fields(new java.util.ArrayList<String>(java.util.Arrays.asList("testString")))
+            .fields(java.util.Arrays.asList("testString"))
             .build();
     assertEquals(collectionEnrichmentModel.enrichmentId(), "testString");
-    assertEquals(
-        collectionEnrichmentModel.fields(),
-        new java.util.ArrayList<String>(java.util.Arrays.asList("testString")));
+    assertEquals(collectionEnrichmentModel.fields(), java.util.Arrays.asList("testString"));
 
     String json = TestUtilities.serialize(collectionEnrichmentModel);
 

discovery/src/test/java/com/ibm/watson/discovery/v2/model/CollectionEnrichmentTest.java

@@ -1,5 +1,5 @@
 /*
- * (C) Copyright IBM Corp. 2020, 2021.
+ * (C) Copyright IBM Corp. 2022.
  *
  * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with
  * the License. You may obtain a copy of the License at
@@ -33,31 +33,38 @@ public void testCreateCollectionOptions() throws Throwable {
     CollectionEnrichment collectionEnrichmentModel =
         new CollectionEnrichment.Builder()
             .enrichmentId("testString")
-            .fields(new java.util.ArrayList<String>(java.util.Arrays.asList("testString")))
+            .fields(java.util.Arrays.asList("testString"))
             .build();
     assertEquals(collectionEnrichmentModel.enrichmentId(), "testString");
-    assertEquals(
-        collectionEnrichmentModel.fields(),
-        new java.util.ArrayList<String>(java.util.Arrays.asList("testString")));
+    assertEquals(collectionEnrichmentModel.fields(), java.util.Arrays.asList("testString"));
+
+    CollectionDetailsSmartDocumentUnderstanding collectionDetailsSmartDocumentUnderstandingModel =
+        new CollectionDetailsSmartDocumentUnderstanding.Builder()
+            .enabled(true)
+            .model("custom")
+            .build();
+    assertEquals(collectionDetailsSmartDocumentUnderstandingModel.enabled(), Boolean.valueOf(true));
+    assertEquals(collectionDetailsSmartDocumentUnderstandingModel.model(), "custom");
 
     CreateCollectionOptions createCollectionOptionsModel =
         new CreateCollectionOptions.Builder()
             .projectId("testString")
             .name("testString")
             .description("testString")
             .language("en")
-            .enrichments(
-                new java.util.ArrayList<CollectionEnrichment>(
-                    java.util.Arrays.asList(collectionEnrichmentModel)))
+            .enrichments(java.util.Arrays.asList(collectionEnrichmentModel))
+            .smartDocumentUnderstanding(collectionDetailsSmartDocumentUnderstandingModel)
             .build();
     assertEquals(createCollectionOptionsModel.projectId(), "testString");
     assertEquals(createCollectionOptionsModel.name(), "testString");
     assertEquals(createCollectionOptionsModel.description(), "testString");
     assertEquals(createCollectionOptionsModel.language(), "en");
     assertEquals(
         createCollectionOptionsModel.enrichments(),
-        new java.util.ArrayList<CollectionEnrichment>(
-            java.util.Arrays.asList(collectionEnrichmentModel)));
+        java.util.Arrays.asList(collectionEnrichmentModel));
+    assertEquals(
+        createCollectionOptionsModel.smartDocumentUnderstanding(),
+        collectionDetailsSmartDocumentUnderstandingModel);
   }
 
   @Test(expectedExceptions = IllegalArgumentException.class)

discovery/src/test/java/com/ibm/watson/discovery/v2/model/CreateCollectionOptionsTest.java

@@ -0,0 +1,64 @@
+/*
+ * (C) Copyright IBM Corp. 2022.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with
+ * the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
+ * an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
+ * specific language governing permissions and limitations under the License.
+ */
+
+package com.ibm.watson.discovery.v2.model;
+
+import static org.testng.Assert.*;
+
+import com.ibm.cloud.sdk.core.service.model.FileWithMetadata;
+import com.ibm.watson.discovery.v2.utils.TestUtilities;
+import java.io.InputStream;
+import java.util.HashMap;
+import java.util.List;
+import org.testng.annotations.Test;
+
+/** Unit test class for the CreateDocumentClassifierModelOptions model. */
+public class CreateDocumentClassifierModelOptionsTest {
+  final HashMap<String, InputStream> mockStreamMap = TestUtilities.createMockStreamMap();
+  final List<FileWithMetadata> mockListFileWithMetadata =
+      TestUtilities.creatMockListFileWithMetadata();
+
+  @Test
+  public void testCreateDocumentClassifierModelOptions() throws Throwable {
+    CreateDocumentClassifierModelOptions createDocumentClassifierModelOptionsModel =
+        new CreateDocumentClassifierModelOptions.Builder()
+            .projectId("testString")
+            .classifierId("testString")
+            .name("testString")
+            .description("testString")
+            .learningRate(Double.valueOf("0"))
+            .l1RegularizationStrengths(java.util.Arrays.asList(Double.valueOf("1.0E-6")))
+            .l2RegularizationStrengths(java.util.Arrays.asList(Double.valueOf("1.0E-6")))
+            .trainingMaxSteps(Long.valueOf("0"))
+            .improvementRatio(Double.valueOf("0"))
+            .build();
+    assertEquals(createDocumentClassifierModelOptionsModel.projectId(), "testString");
+    assertEquals(createDocumentClassifierModelOptionsModel.classifierId(), "testString");
+    assertEquals(createDocumentClassifierModelOptionsModel.name(), "testString");
+    assertEquals(createDocumentClassifierModelOptionsModel.description(), "testString");
+    assertEquals(createDocumentClassifierModelOptionsModel.learningRate(), Double.valueOf("0"));
+    assertEquals(
+        createDocumentClassifierModelOptionsModel.l1RegularizationStrengths(),
+        java.util.Arrays.asList(Double.valueOf("1.0E-6")));
+    assertEquals(
+        createDocumentClassifierModelOptionsModel.l2RegularizationStrengths(),
+        java.util.Arrays.asList(Double.valueOf("1.0E-6")));
+    assertEquals(createDocumentClassifierModelOptionsModel.trainingMaxSteps(), Long.valueOf("0"));
+    assertEquals(createDocumentClassifierModelOptionsModel.improvementRatio(), Double.valueOf("0"));
+  }
+
+  @Test(expectedExceptions = IllegalArgumentException.class)
+  public void testCreateDocumentClassifierModelOptionsError() throws Throwable {
+    new CreateDocumentClassifierModelOptions.Builder().build();
+  }
+}