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')])
