def connectivity(): nrrd.read = MagicMock(name='nrrd_read_file', return_value=('mock_annotation_data', 'mock_annotation_image')) conn_api = MouseConnectivityApi() download_link = '/path/to/link' conn_api.do_query = MagicMock(return_value=download_link) conn_api.json_msg_query = MagicMock(name='json_msg_query') conn_api.retrieve_file_over_http = \ MagicMock(name='retrieve_file_over_http') return conn_api