Skip to content

Commit b33ce86

Browse files
committed
test(Natural Language Classifier): Update tests
1 parent b982379 commit b33ce86

File tree

2 files changed

+20
-23
lines changed

2 files changed

+20
-23
lines changed

natural-language-classifier/src/test/java/com/ibm/watson/natural_language_classifier/v1/NaturalLanguageClassifierIT.java

Lines changed: 10 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -12,8 +12,8 @@
1212
*/
1313
package com.ibm.watson.natural_language_classifier.v1;
1414

15+
import com.ibm.cloud.sdk.core.security.IamAuthenticator;
1516
import com.ibm.cloud.sdk.core.service.exception.NotFoundException;
16-
import com.ibm.cloud.sdk.core.service.security.IamOptions;
1717
import com.ibm.watson.common.WatsonServiceTest;
1818
import com.ibm.watson.natural_language_classifier.v1.model.Classification;
1919
import com.ibm.watson.natural_language_classifier.v1.model.ClassificationCollection;
@@ -64,12 +64,9 @@ public void setUp() throws Exception {
6464

6565
Assume.assumeFalse("config.properties doesn't have valid credentials.", apiKey == null);
6666

67-
service = new NaturalLanguageClassifier();
67+
IamAuthenticator authenticator = new IamAuthenticator(apiKey);
68+
service = new NaturalLanguageClassifier(authenticator);
6869
service.setDefaultHeaders(getDefaultHeaders());
69-
IamOptions iamOptions = new IamOptions.Builder()
70-
.apiKey(apiKey)
71-
.build();
72-
service.setIamCredentials(iamOptions);
7370
service.setEndPoint(getProperty("natural_language_classifier.url"));
7471

7572
preCreatedClassifierId = getProperty("natural_language_classifier.classifier_id");
@@ -86,7 +83,7 @@ public void aCreate() throws Exception {
8683
final File metadata = new File("src/test/resources/natural_language_classifier/metadata.json");
8784

8885
CreateClassifierOptions createOptions = new CreateClassifierOptions.Builder()
89-
.metadata(metadata)
86+
.trainingMetadata(metadata)
9087
.trainingData(trainingData)
9188
.build();
9289
Classifier classifier = service.createClassifier(createOptions).execute().getResult();
@@ -179,10 +176,12 @@ public void eDelete() throws InterruptedException {
179176
@Test
180177
public void fClassifyCollection() {
181178
ClassificationCollection classificationCollection = null;
182-
ClassifyInput input1 = new ClassifyInput();
183-
input1.setText("How hot will it be today?");
184-
ClassifyInput input2 = new ClassifyInput();
185-
input2.setText("Is it hot outside?");
179+
ClassifyInput input1 = new ClassifyInput.Builder()
180+
.text("How hot will it be today?")
181+
.build();
182+
ClassifyInput input2 = new ClassifyInput.Builder()
183+
.text("Is it hot outside?")
184+
.build();
186185

187186
try {
188187
ClassifyCollectionOptions classifyOptions = new ClassifyCollectionOptions.Builder()

natural-language-classifier/src/test/java/com/ibm/watson/natural_language_classifier/v1/NaturalLanguageClassifierTest.java

Lines changed: 10 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@
1313
package com.ibm.watson.natural_language_classifier.v1;
1414

1515
import com.google.gson.JsonObject;
16-
import com.ibm.cloud.sdk.core.security.basicauth.BasicAuthConfig;
16+
import com.ibm.cloud.sdk.core.security.NoAuthAuthenticator;
1717
import com.ibm.watson.common.WatsonServiceUnitTest;
1818
import com.ibm.watson.natural_language_classifier.v1.model.Classification;
1919
import com.ibm.watson.natural_language_classifier.v1.model.ClassificationCollection;
@@ -62,11 +62,7 @@ public class NaturalLanguageClassifierTest extends WatsonServiceUnitTest {
6262
@Before
6363
public void setUp() throws Exception {
6464
super.setUp();
65-
BasicAuthConfig authConfig = new BasicAuthConfig.Builder()
66-
.username("")
67-
.password("")
68-
.build();
69-
service = new NaturalLanguageClassifier(authConfig);
65+
service = new NaturalLanguageClassifier(new NoAuthAuthenticator());
7066
service.setEndPoint(getMockWebServerUrl());
7167

7268
classifierId = "foo";
@@ -114,10 +110,12 @@ public void testClassifyCollection() throws InterruptedException {
114110

115111
server.enqueue(jsonResponse(classificationCollection));
116112

117-
ClassifyInput input1 = new ClassifyInput();
118-
input1.setText("How hot will it be today?");
119-
ClassifyInput input2 = new ClassifyInput();
120-
input2.setText("Is it hot outside?");
113+
ClassifyInput input1 = new ClassifyInput.Builder()
114+
.text("How hot will it be today?")
115+
.build();
116+
ClassifyInput input2 = new ClassifyInput.Builder()
117+
.text("Is it hot outside?")
118+
.build();
121119
List<ClassifyInput> inputCollection = Arrays.asList(input1, input2);
122120

123121
ClassifyCollectionOptions classifyOptions = new ClassifyCollectionOptions.Builder()
@@ -176,7 +174,7 @@ public void testCreateClassifier() throws InterruptedException, FileNotFoundExce
176174
File metadata = new File(RESOURCE + "metadata.json");
177175
File trainingData = new File(RESOURCE + "weather_data_train.csv");
178176
CreateClassifierOptions createOptions = new CreateClassifierOptions.Builder()
179-
.metadata(metadata)
177+
.trainingMetadata(metadata)
180178
.trainingData(trainingData)
181179
.build();
182180
final Classifier response = service.createClassifier(createOptions).execute().getResult();
@@ -241,7 +239,7 @@ public void testNullTrainingDataFile() throws FileNotFoundException {
241239
File metadata = new File(RESOURCE + "metadata.json");
242240
File trainingData = new File(RESOURCE + "notfound.txt");
243241
CreateClassifierOptions createOptions = new CreateClassifierOptions.Builder()
244-
.metadata(metadata)
242+
.trainingMetadata(metadata)
245243
.trainingData(trainingData)
246244
.build();
247245
service.createClassifier(createOptions).execute();

0 commit comments

Comments
 (0)