def test_google_vision_api_safe_search(): ext = GoogleVisionAPISafeSearchExtractor(num_retries=5) filename = join(get_test_data_path(), 'image', 'obama.jpg') stim = ImageStim(filename) result = ext.transform(stim).to_df() assert 'adult' in result.columns assert result['violence'][0] == 'VERY_UNLIKELY'