use of com.ibm.watson.natural_language_classifier.v1.model.ClassifyInput in project java-sdk by watson-developer-cloud.
the class NaturalLanguageClassifierIT method fClassifyCollection.
/**
* Test classifyCollection. Use the pre created classifier to avoid waiting for availability
*/
@Test
public void fClassifyCollection() {
ClassificationCollection classificationCollection = null;
ClassifyInput input1 = new ClassifyInput();
input1.setText("How hot will it be today?");
ClassifyInput input2 = new ClassifyInput();
input2.setText("Is it hot outside?");
try {
ClassifyCollectionOptions classifyOptions = new ClassifyCollectionOptions.Builder().classifierId(preCreatedClassifierId).addClassifyInput(input1).addClassifyInput(input2).build();
classificationCollection = service.classifyCollection(classifyOptions).execute();
} catch (NotFoundException e) {
// The build should not fail here, because this is out of our control.
throw new AssumptionViolatedException(e.getMessage(), e);
}
assertNotNull(classificationCollection);
assertEquals("temperature", classificationCollection.getCollection().get(0).getTopClass());
assertEquals("temperature", classificationCollection.getCollection().get(1).getTopClass());
}
use of com.ibm.watson.natural_language_classifier.v1.model.ClassifyInput in project java-sdk by watson-developer-cloud.
the class NaturalLanguageClassifierTest method testClassifyCollectionWOptions.
@Test
public void testClassifyCollectionWOptions() throws Throwable {
// Schedule some responses.
String mockResponseBody = "{\"classifier_id\": \"classifierId\", \"url\": \"url\", \"collection\": [{\"text\": \"text\", \"top_class\": \"topClass\", \"classes\": [{\"confidence\": 10, \"class_name\": \"className\"}]}]}";
String classifyCollectionPath = "/v1/classifiers/testString/classify_collection";
server.enqueue(new MockResponse().setHeader("Content-type", "application/json").setResponseCode(200).setBody(mockResponseBody));
constructClientService();
// Construct an instance of the ClassifyInput model
ClassifyInput classifyInputModel = new ClassifyInput.Builder().text("How hot will it be today?").build();
// Construct an instance of the ClassifyCollectionOptions model
ClassifyCollectionOptions classifyCollectionOptionsModel = new ClassifyCollectionOptions.Builder().classifierId("testString").collection(new java.util.ArrayList<ClassifyInput>(java.util.Arrays.asList(classifyInputModel))).build();
// Invoke operation with valid options model (positive test)
Response<ClassificationCollection> response = naturalLanguageClassifierService.classifyCollection(classifyCollectionOptionsModel).execute();
assertNotNull(response);
ClassificationCollection responseObj = response.getResult();
assertNotNull(responseObj);
// Verify the contents of the request
RecordedRequest request = server.takeRequest();
assertNotNull(request);
assertEquals(request.getMethod(), "POST");
// Check query
Map<String, String> query = TestUtilities.parseQueryString(request);
assertNull(query);
// Check request path
String parsedPath = TestUtilities.parseReqPath(request);
assertEquals(parsedPath, classifyCollectionPath);
}
use of com.ibm.watson.natural_language_classifier.v1.model.ClassifyInput in project java-sdk by watson-developer-cloud.
the class NaturalLanguageClassifierTest method testClassifyCollection.
/**
* Test classifying a collection.
*
* @throws InterruptedException the interrupted exception
*/
@Test
public void testClassifyCollection() throws InterruptedException {
final String path = String.format(CLASSIFY_COLLECTION_PATH, classifierId);
server.enqueue(jsonResponse(classificationCollection));
ClassifyInput input1 = new ClassifyInput();
input1.setText("How hot will it be today?");
ClassifyInput input2 = new ClassifyInput();
input2.setText("Is it hot outside?");
List<ClassifyInput> inputCollection = Arrays.asList(input1, input2);
ClassifyCollectionOptions classifyOptions = new ClassifyCollectionOptions.Builder().classifierId(classifierId).collection(inputCollection).build();
final ClassificationCollection result = service.classifyCollection(classifyOptions).execute();
final RecordedRequest request = server.takeRequest();
assertEquals(path, request.getPath());
assertEquals("POST", request.getMethod());
assertEquals(classificationCollection, result);
}
use of com.ibm.watson.natural_language_classifier.v1.model.ClassifyInput in project java-sdk by watson-developer-cloud.
the class NaturalLanguageClassifierIT method fClassifyCollection.
/**
* Test classifyCollection. Use the pre created classifier to avoid waiting for availability
*/
@Test
public void fClassifyCollection() {
ClassificationCollection classificationCollection = null;
ClassifyInput input1 = new ClassifyInput.Builder().text("How hot will it be today?").build();
ClassifyInput input2 = new ClassifyInput.Builder().text("Is it hot outside?").build();
try {
ClassifyCollectionOptions classifyOptions = new ClassifyCollectionOptions.Builder().classifierId(preCreatedClassifierId).addClassifyInput(input1).addClassifyInput(input2).build();
classificationCollection = service.classifyCollection(classifyOptions).execute().getResult();
} catch (NotFoundException e) {
// The build should not fail here, because this is out of our control.
throw new AssumptionViolatedException(e.getMessage(), e);
}
assertNotNull(classificationCollection);
assertEquals("temperature", classificationCollection.getCollection().get(0).getTopClass());
assertEquals("temperature", classificationCollection.getCollection().get(1).getTopClass());
}
Aggregations