use of com.ibm.watson.developer_cloud.visual_recognition.v3.model.FaceIdentity in project jbpm-work-items by kiegroup.
the class WatsonWorkitemHandlerTest method setUp.
// @Mock
// Face recognitionFace;
@Before
public void setUp() {
// image classification
ClassResult classResultObj = new ClassResult();
classResultObj.setClassName("testClassName");
classResultObj.setScore(new Float(1));
classResultObj.setTypeHierarchy("testTypeHierarchy");
List<ClassifiedImage> classifiedImageList = new ArrayList<>();
classifiedImageList.add(associationClassifiedImage);
List<ClassifierResult> classifierResults = new ArrayList<>();
classifierResults.add(associationClassifierResult);
List<ClassResult> classResults = new ArrayList<>();
classResults.add(classResultObj);
when(associationService.classify(any(ClassifyOptions.class))).thenReturn(associationServiceCall);
when(associationServiceCall.execute()).thenReturn(assoiationClassifiedImages);
when(assoiationClassifiedImages.getImages()).thenReturn(classifiedImageList);
when(associationClassifiedImage.getClassifiers()).thenReturn(classifierResults);
when(associationClassifierResult.getClasses()).thenReturn(classResults);
when(associationClassifierResult.getClassifierId()).thenReturn("testClassifierId");
// face detection
List<ImageWithFaces> recognitionImageList = new ArrayList<>();
recognitionImageList.add(recognitionImageWithFaces);
FaceAge recognitionFaceAge = new FaceAge();
recognitionFaceAge.setMin(20);
recognitionFaceAge.setMax(35);
FaceGender recognitionFaceGender = new FaceGender();
recognitionFaceGender.setGender("male");
FaceIdentity recognitionFaceIdentity = new FaceIdentity();
recognitionFaceIdentity.setName("testPerson");
Face recognitionFace = new Face();
recognitionFace.setAge(recognitionFaceAge);
recognitionFace.setGender(recognitionFaceGender);
recognitionFace.setIdentity(recognitionFaceIdentity);
List<Face> recognitionFaceList = new ArrayList<>();
recognitionFaceList.add(recognitionFace);
when(recognitionService.detectFaces(any(DetectFacesOptions.class))).thenReturn(recognitionServiceCall);
when(recognitionServiceCall.execute()).thenReturn(recognitionDetectFaces);
when(recognitionDetectFaces.getImages()).thenReturn(recognitionImageList);
when(recognitionImageWithFaces.getFaces()).thenReturn(recognitionFaceList);
when(recognitionImageWithFaces.getImage()).thenReturn("testImage");
}
Aggregations