def test_custom_valid_ext(name, suffix, expected): alt_suffix = suffix.replace(".", "") result1 = validate_image_name(name, suffix=suffix) result2 = validate_image_name(name, suffix=alt_suffix) assert result1 == expected assert result2 == expected
def test_custom_valid_ext(): for dext in SUPPORTED_FORMATS: ext = dext.replace(".", "") result1 = validate_image_name(filename_no_ext, suffix=ext) result2 = validate_image_name(filename_no_ext, suffix=dext) expected = f"{filename_no_ext}.{ext}" assert result1 == expected assert result2 == expected
def test_default(): result = validate_image_name(filename_no_ext) expected = f"{filename_no_ext}.png" assert result == expected result = validate_image_name(filename_png) assert result == expected result = validate_image_name(filename_pdf) expected = str(filename_pdf) assert result == expected
def _setup_save(self, fname, render) -> str: self._render_on_demand(render) rensources = self._get_render_sources() if fname is None: # if a volume source present, use its affiliated ds for fname if len(rensources) > 0: rs = rensources[0] basename = rs.data_source.ds.basename if isinstance(rs.field, str): field = rs.field else: field = rs.field[-1] fname = f"{basename}_Render_{field}" # if no volume source present, use a default filename else: fname = "Render_opaque" fname = validate_image_name(fname) mylog.info("Saving rendered image to %s", fname) return fname
def test_default(name, expected): result = validate_image_name(name) assert result == expected