def test_google_vision_api_face_extractor(): ext = GoogleVisionAPIFaceExtractor(num_retries=5) filename = join(get_test_data_path(), 'image', 'obama.jpg') stim = ImageStim(filename) result = ext.extract(stim).to_df() assert 'joyLikelihood' in result.columns assert result['joyLikelihood'][0] == 'VERY_LIKELY' assert result['face_detectionConfidence'][0] > 0.7
def test_google_vision_api_face_extractor_inits(): ext = GoogleVisionAPIFaceExtractor(num_retries=5) assert ext.num_retries == 5 assert ext.max_results == 100 assert ext.service is not None # Test parsing of individual response filename = join(get_test_data_path(), 'payloads', 'google_vision_api_face_payload.json') response = json.load(open(filename, 'r')) features, data = ext._parse_annotations(response['faceAnnotations']) assert len(features) == len(data) assert data[features.index('angerLikelihood')] == 'VERY_UNLIKELY' assert data[features.index('landmark_LEFT_EYE_BOTTOM_BOUNDARY_y')] == 257.023 assert np.isnan(data[features.index('boundingPoly_vertex2_y')])