def _do_save(save_op_tuple: Tuple):
set_io_scopes(caller_io_scopes)
if len(save_op_tuple) == 2:
return save(save_op_tuple[0],
save_op_tuple[1],
save_context=current_save_context)
elif len(save_op_tuple) == 3:
return save(save_op_tuple[0],
save_op_tuple[1],
save_context=current_save_context,
**(save_op_tuple[2]))
else:
raise ValueError(f'unknown save tuple size: {len(save_op_tuple)}')
def test_set_write_scopes():
target_path = "./tests/fixtures/write_scope5.txt"
if os.path.exists(target_path):
os.remove(target_path)
# fake write scopes, such as when called on a remote worker:
_old_scopes = scoping.current_io_scopes()
scoping.set_io_scopes(["./tests", "fixtures"])
write("test", "write_scope5.txt", mode="w")
assert os.path.isfile(target_path)
# restore write scopes for later tests
scoping.set_io_scopes(_old_scopes)
def _do_load(url):
set_io_scopes(caller_io_scopes)
return load(url, cache=cache, **kwargs)