def test_render_svg(capsys):
output = dump_graphviz(tree, output_format='svg')
print_graphviz(output)
out, _ = capsys.readouterr()
assert out.startswith('<?xml')
assert '<svg' in out
assert out.strip().endswith('</svg>')
def test_render_svg(capsys):
output = dump_graphviz(tree, output_format='svg')
print_graphviz(output)
out, _ = capsys.readouterr()
assert out.startswith('<?xml')
assert '<svg' in out
assert out.strip().endswith('</svg>')
def test_render_json(capsys):
output = jsonify_tree(tree, indent=4)
print_graphviz(output)
out, _ = capsys.readouterr()
assert out.startswith('[\n {\n "')
assert out.strip().endswith('}\n]')
data = json.loads(out)
assert 'package' in data[0]
assert 'dependencies' in data[0]
def test_render_json(capsys):
output = render_json(tree, indent=4)
print_graphviz(output)
out, _ = capsys.readouterr()
assert out.startswith('[\n {\n "')
assert out.strip().endswith('}\n]')
data = json.loads(out)
assert 'package' in data[0]
assert 'dependencies' in data[0]
def test_render_pdf():
output = p.dump_graphviz(t, output_format='pdf')
@contextmanager
def redirect_stdout(new_target):
old_target, sys.stdout = sys.stdout, new_target
try:
yield new_target
finally:
sys.stdout = old_target
with NamedTemporaryFile(delete=True) as f:
with redirect_stdout(f):
p.print_graphviz(output)
rf = open(f.name, 'rb')
assert b'%PDF' == rf.read()[:4]
def test_render_pdf():
output = dump_graphviz(tree, output_format='pdf')
@contextmanager
def redirect_stdout(new_target):
old_target, sys.stdout = sys.stdout, new_target
try:
yield new_target
finally:
sys.stdout = old_target
f = NamedTemporaryFile(delete=False)
with redirect_stdout(f):
print_graphviz(output)
with open(f.name, 'rb') as rf:
out = rf.read()
os.remove(f.name)
assert out[:4] == b'%PDF'
def test_render_pdf():
output = dump_graphviz(tree, output_format='pdf')
@contextmanager
def redirect_stdout(new_target):
old_target, sys.stdout = sys.stdout, new_target
try:
yield new_target
finally:
sys.stdout = old_target
f = NamedTemporaryFile(delete=False)
with redirect_stdout(f):
print_graphviz(output)
with open(f.name, 'rb') as rf:
out = rf.read()
os.remove(f.name)
assert out[:4] == b'%PDF'
