def test_download_image_terms():
with _TestTemporaryDirectory() as temp_dir:
image_info = {'id': 'a'}
collection = {
'relative_path': 'collection',
'absolute_path': os.path.join(temp_dir, 'collection')
}
os.makedirs(collection['absolute_path'])
download_params = {
'temp_dir': temp_dir,
'verbose': 3,
'fetch_neurosynth_words': True
}
with _FailingDownloads():
neurovault._download_image_terms(image_info, collection,
download_params)
download_params['allow_neurosynth_failure'] = False
pytest.raises(RuntimeError, neurovault._download_image_terms,
image_info, collection, download_params)
with open(
os.path.join(collection['absolute_path'],
'neurosynth_words_for_image_a.json'), 'w'):
pass
neurovault._download_image_terms(image_info, collection,
download_params)
def test_download_image_terms(request_mocker):
with tempfile.TemporaryDirectory() as temp_dir:
image_info = {'id': 'a'}
collection = {
'relative_path': 'collection',
'absolute_path': os.path.join(temp_dir, 'collection')
}
os.makedirs(collection['absolute_path'])
download_params = {
'temp_dir': temp_dir,
'verbose': 3,
'fetch_neurosynth_words': True
}
request_mocker.url_mapping["*"] = requests.RequestException()
neurovault._download_image_terms(image_info, collection,
download_params)
download_params['allow_neurosynth_failure'] = False
pytest.raises(RuntimeError, neurovault._download_image_terms,
image_info, collection, download_params)
with open(
os.path.join(collection['absolute_path'],
'neurosynth_words_for_image_a.json'), 'w'):
pass
neurovault._download_image_terms(image_info, collection,
download_params)
def test_download_image_terms():
with _TestTemporaryDirectory() as temp_dir:
image_info = {'id': 'a'}
collection = {'relative_path': 'collection',
'absolute_path': os.path.join(temp_dir, 'collection')}
os.makedirs(collection['absolute_path'])
download_params = {'temp_dir': temp_dir, 'verbose': 3,
'fetch_neurosynth_words': True}
with _FailingDownloads():
neurovault._download_image_terms(
image_info, collection, download_params)
download_params['allow_neurosynth_failure'] = False
assert_raises(RuntimeError,
neurovault._download_image_terms,
image_info, collection, download_params)
with open(os.path.join(
collection['absolute_path'],
'neurosynth_words_for_image_a.json'), 'w'):
pass
neurovault._download_image_terms(
image_info, collection, download_params)