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'