Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
def test_request_file_no_seek():
class File:
def read(self, n):
return b"\x00" * n
def seek(self, i):
raise IOError("Not supported")
def tell(self):
raise Exception("Not supported")
def close(self):
pass
R = Request(File(), "ri")
with raises(IOError):
R.firstbytes
def test_select():
fname1 = get_remote_file("images/cockatoo.mp4", test_dir)
F = imageio.formats["avbin"]
assert F.name == "AVBIN"
assert F.can_read(core.Request(fname1, "rI"))
assert not F.can_write(core.Request(fname1, "wI"))
assert not F.can_read(core.Request(fname1, "ri"))
assert not F.can_read(core.Request(fname1, "rv"))
# Test can read, cannot write
assert format.can_read(Request(lfp_file, "ri"))
# Test cannot read, cannot write
assert not format.can_read(Request(lfp_file, "rv"))
assert not format.can_read(Request(lfp_file, "rI"))
assert not format.can_read(Request(lfp_file, "rV"))
assert not format.can_read(Request(lfr_file, "ri"))
assert not format.can_read(Request(raw_f01_file, "ri"))
assert not format.can_read(Request(raw_illum_file, "ri"))
assert not format.can_read(Request(png_file, "ri"))
assert not format.can_write(Request(lfp_file, "wi"))
assert not format.can_write(Request(lfr_file, "wi"))
assert not format.can_write(Request(raw_f01_file, "wi"))
assert not format.can_write(Request(raw_illum_file, "wi"))
assert not format.can_write(Request(png_file, "wi"))
# Test lytro raw format
format = imageio.formats["lytro-illum-raw"]
assert format.name == "LYTRO-ILLUM-RAW"
assert format.__module__.endswith("lytro")
# Test can read, cannot write
assert format.can_read(Request(raw_illum_file, "ri"))
# Test cannot read, cannot write
assert not format.can_read(Request(raw_illum_file, "rv"))
assert not format.can_read(Request(raw_illum_file, "rI"))
assert not format.can_read(Request(raw_illum_file, "rV"))
assert not format.can_read(Request(lfr_file, "ri"))
assert not format.can_read(Request(lfp_file, "ri"))
assert not format.can_read(Request(png_file, "ri"))
assert not format.can_write(Request(raw_illum_file, "wi"))
assert not format.can_write(Request(lfr_file, "wi"))
assert not format.can_write(Request(lfp_file, "wi"))
assert not format.can_write(Request(raw_f01_file, "wi"))
assert not format.can_write(Request(png_file, "wi"))
lfr_file = get_remote_file(LFR_FILENAME)
raw_illum_file = get_remote_file(RAW_ILLUM_FILENAME)
lfp_file = get_remote_file(LFP_FILENAME)
raw_f01_file = get_remote_file(RAW_F0_FILENAME)
png_file = get_remote_file(PNG_FILENAME)
# Test lytro lfr format
format = imageio.formats["lytro-lfr"]
assert format.name == "LYTRO-LFR"
assert format.__module__.endswith("lytro")
# Test can read
assert format.can_read(Request(lfr_file, "ri"))
# Test cannot read, cannot write
assert not format.can_read(Request(lfr_file, "rv"))
assert not format.can_read(Request(lfr_file, "rI"))
assert not format.can_read(Request(lfr_file, "rV"))
assert not format.can_read(Request(lfp_file, "ri"))
assert not format.can_read(Request(raw_illum_file, "ri"))
assert not format.can_read(Request(raw_f01_file, "ri"))
assert not format.can_read(Request(png_file, "ri"))
assert not format.can_write(Request(lfr_file, "wi"))
assert not format.can_write(Request(lfp_file, "wi"))
assert not format.can_write(Request(raw_f01_file, "wi"))
assert not format.can_write(Request(raw_illum_file, "wi"))
assert not format.can_write(Request(png_file, "wi"))
# Test can read, cannot write
assert format.can_read(Request(raw_illum_file, "ri"))
# Test cannot read, cannot write
assert not format.can_read(Request(raw_illum_file, "rv"))
assert not format.can_read(Request(raw_illum_file, "rI"))
assert not format.can_read(Request(raw_illum_file, "rV"))
assert not format.can_read(Request(lfr_file, "ri"))
assert not format.can_read(Request(lfp_file, "ri"))
assert not format.can_read(Request(png_file, "ri"))
assert not format.can_write(Request(raw_illum_file, "wi"))
assert not format.can_write(Request(lfr_file, "wi"))
assert not format.can_write(Request(lfp_file, "wi"))
assert not format.can_write(Request(raw_f01_file, "wi"))
assert not format.can_write(Request(png_file, "wi"))
the appropriate for you based on the filename.
mode : {'i', 'I', 'v', 'V', '?'}
Used to give the writer a hint on what the user expects (default '?'):
"i" for an image, "I" for multiple images, "v" for a volume,
"V" for multiple volumes, "?" for don't care.
kwargs : ...
Further keyword arguments are passed to the writer. See :func:`.help`
to see what arguments are available for a particular format.
"""
# Signal extension when returning as bytes, needed by e.g. ffmpeg
if uri == RETURN_BYTES and isinstance(format, str):
uri = RETURN_BYTES + "." + format.strip(". ")
# Create request object
request = Request(uri, "w" + mode, **kwargs)
# Get format
if format is not None:
format = formats[format]
else:
format = formats.search_write_format(request)
if format is None:
raise ValueError(
"Could not find a format to write the specified file " "in mode %r" % mode
)
# Return its writer object
return format.get_writer(request)
The resource to load the image from, e.g. a filename, pathlib.Path,
http address or file object, see the docs for more info.
format : str
The format to use to read the file. By default imageio selects
the appropriate for you based on the filename and its contents.
mode : {'i', 'I', 'v', 'V', '?'}
Used to give the reader a hint on what the user expects (default "?"):
"i" for an image, "I" for multiple images, "v" for a volume,
"V" for multiple volumes, "?" for don't care.
kwargs : ...
Further keyword arguments are passed to the reader. See :func:`.help`
to see what arguments are available for a particular format.
"""
# Create request object
request = Request(uri, "r" + mode, **kwargs)
# Get format
if format is not None:
format = formats[format]
else:
format = formats.search_read_format(request)
if format is None:
raise ValueError(
"Could not find a format to read the specified file " "in mode %r" % mode
)
# Return its reader object
return format.get_reader(request)