def test_filenames_and_formats():
# Test with a variety of user provided args
filenames = [
'mydaskpdf', 'mydask.pdf', 'mydask.pdf', 'mydaskpdf', 'mydask.pdf.svg'
]
formats = ['svg', None, 'svg', None, None]
targets = [
'mydaskpdf.svg', 'mydask.pdf', 'mydask.pdf.svg', 'mydaskpdf.png',
'mydask.pdf.svg'
]
result_types = {
'png': Image,
'jpeg': Image,
'dot': type(None),
'pdf': type(None),
'svg': SVG,
}
for filename, format, target in zip(filenames, formats, targets):
expected_result_type = result_types[target.split('.')[-1]]
result = dot_graph(dsk, filename=filename, format=format)
assert os.path.isfile(target)
assert isinstance(result, expected_result_type)
ensure_not_exists(target)
def test_dot_graph_defaults():
# Test with default args.
default_name = 'mydask'
default_format = 'png'
target = '.'.join([default_name, default_format])
ensure_not_exists(target)
try:
result = dot_graph(dsk)
assert os.path.isfile(target)
assert isinstance(result, Image)
finally:
ensure_not_exists(target)
def test_dot_graph_defaults():
# Test with default args.
default_name = 'mydask'
default_format = 'png'
target = '.'.join([default_name, default_format])
ensure_not_exists(target)
try:
result = dot_graph(dsk)
assert os.path.isfile(target)
assert isinstance(result, Image)
finally:
ensure_not_exists(target)
def test_filenames_and_formats():
# Test with a variety of user provided args
filenames = ["mydaskpdf", "mydask.pdf", "mydask.pdf", "mydaskpdf"]
formats = ["svg", None, "svg", None]
targets = ["mydaskpdf.svg", "mydask.pdf", "mydask.pdf.svg", "mydaskpdf.png"]
result_types = {"png": Image, "jpeg": Image, "dot": type(None), "pdf": type(None), "svg": SVG}
for filename, format, target in zip(filenames, formats, targets):
expected_result_type = result_types[target.split(".")[-1]]
result = dot_graph(dsk, filename=filename, format=format)
assert os.path.isfile(target)
assert isinstance(result, expected_result_type)
ensure_not_exists(target)
def test_dot_graph(tmpdir, format, typ):
# Use a name that the shell would interpret specially to ensure that we're
# not vulnerable to shell injection when interacting with `dot`.
filename = str(tmpdir.join('$(touch should_not_get_created.txt)'))
target = '.'.join([filename, format])
ensure_not_exists(target)
try:
result = dot_graph(dsk, filename=filename, format=format)
assert not os.path.exists('should_not_get_created.txt')
assert os.path.isfile(target)
assert isinstance(result, typ)
finally:
ensure_not_exists(target)
def test_dot_graph(tmpdir, format, typ):
# Use a name that the shell would interpret specially to ensure that we're
# not vulnerable to shell injection when interacting with `dot`.
filename = str(tmpdir.join('$(touch should_not_get_created.txt)'))
target = '.'.join([filename, format])
ensure_not_exists(target)
try:
result = dot_graph(dsk, filename=filename, format=format)
assert not os.path.exists('should_not_get_created.txt')
assert os.path.isfile(target)
assert isinstance(result, typ)
finally:
ensure_not_exists(target)
def test_dot_graph(tmpdir):
# Use a name that the shell would interpret specially to ensure that we're
# not vulnerable to shell injection when interacting with `dot`.
filename = str(tmpdir.join("$(touch should_not_get_created.txt)"))
# Map from format extension to expected return type.
result_types = {"png": Image, "jpeg": Image, "dot": type(None), "pdf": type(None), "svg": SVG}
for format in result_types:
target = ".".join([filename, format])
ensure_not_exists(target)
try:
result = dot_graph(dsk, filename=filename, format=format)
assert not os.path.exists("should_not_get_created.txt")
assert os.path.isfile(target)
assert isinstance(result, result_types[format])
finally:
ensure_not_exists(target)
def test_filenames_and_formats():
# Test with a variety of user provided args
filenames = ['mydaskpdf', 'mydask.pdf', 'mydask.pdf', 'mydaskpdf', 'mydask.pdf.svg']
formats = ['svg', None, 'svg', None, None]
targets = ['mydaskpdf.svg', 'mydask.pdf', 'mydask.pdf.svg', 'mydaskpdf.png', 'mydask.pdf.svg']
result_types = {
'png': Image,
'jpeg': Image,
'dot': type(None),
'pdf': type(None),
'svg': SVG,
}
for filename, format, target in zip(filenames, formats, targets):
expected_result_type = result_types[target.split('.')[-1]]
result = dot_graph(dsk, filename=filename, format=format)
assert os.path.isfile(target)
assert isinstance(result, expected_result_type)
ensure_not_exists(target)
def test_dot_graph(tmpdir):
# Use a name that the shell would interpret specially to ensure that we're
# not vulnerable to shell injection when interacting with `dot`.
filename = str(tmpdir.join('$(touch should_not_get_created.txt)'))
# Map from format extension to expected return type.
result_types = {
'png': Image,
'jpeg': Image,
'dot': type(None),
'pdf': type(None),
'svg': SVG,
}
for format in result_types:
target = '.'.join([filename, format])
ensure_not_exists(target)
try:
result = dot_graph(dsk, filename=filename, format=format)
assert not os.path.exists('should_not_get_created.txt')
assert os.path.isfile(target)
assert isinstance(result, result_types[format])
finally:
ensure_not_exists(target)
def test_dot_graph(tmpdir):
# Use a name that the shell would interpret specially to ensure that we're
# not vulnerable to shell injection when interacting with `dot`.
filename = str(tmpdir.join('$(touch should_not_get_created.txt)'))
# Map from format extension to expected return type.
result_types = {
'png': Image,
'jpeg': Image,
'dot': type(None),
'pdf': type(None),
'svg': SVG,
}
for format in result_types:
target = '.'.join([filename, format])
ensure_not_exists(target)
try:
result = dot_graph(dsk, filename=filename, format=format)
assert not os.path.exists('should_not_get_created.txt')
assert os.path.isfile(target)
assert isinstance(result, result_types[format])
finally:
ensure_not_exists(target)
def test_filenames_and_formats(filename, format, target, expected_result_type):
result = dot_graph(dsk, filename=filename, format=format)
assert os.path.isfile(target)
assert isinstance(result, expected_result_type)
ensure_not_exists(target)
