/** * 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); }
/** * 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) { // #324: Classifiers may be empty, because of other tests interfering. // 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()); }