def test_output(): dotter = Dotter(output_to_file=False, output_type='dot') for node in ['a', 'b', 'c', 'd']: dotter.add_node(node, label=node) output = dotter.commands expected = [x.replace('\t', '') for x in load_data('test_output.dot')] assert output == expected
def test_output(): dotter = Dotter(output_to_file=False, output_type='dot') for node in ['a', 'b', 'c', 'd']: dotter.add_node(node, label=node) output = dotter.close() output = output.splitlines() expected = load_data('test_output.dot') assert output == expected
def test_addnode_styles(): dotter = Dotter() dotter.add_node('a', styles=["diagonals", "filled", "bold"]) expected = ['digraph', ' {', 'gb', 'gb [label="a"]', 'gb [style="diagonals, filled, bold"]', ] assert expected == dotter.commands
def test_addnode_styles(): dotter = Dotter() dotter.add_node('a', styles=["diagonals", "filled", "bold"]) expected = [ 'digraph', ' {', 'gb', 'gb [label="a"]', 'gb [style="diagonals, filled, bold"]', ] assert expected == dotter.commands
def test_rank(): dotter = Dotter() for node in ['a', 'b', 'c', 'd']: dotter.add_node(node, label=node) dotter.add_edge('a', 'b') dotter.add_edge('a', 'c') dotter.add_edge('a', 'd') dotter.rank(['a', 'b'], RankType.Min) dotter.rank(['c'], RankType.Same) dotter.rank(['d'], RankType.Max) expected = load_data('test_rank.dot') assert expected == dotter.commands
def test_set_position_wrong_program(): dotter = Dotter() dotter.add_node('a') with raises(Warning): dotter.set_position('a', 5, 4)
def test_set_position(): dotter = Dotter(program=Program.Neato) dotter.add_node('a') dotter.set_position('a', 5, 4) expected = ['digraph', ' {', 'gb', 'gb [label="a"]', 'gb [pos="5,4!"]'] assert expected == dotter.commands
def test_addnode_label(): dotter = Dotter() dotter.add_node('a', label='b') expected = ['digraph', ' {', 'gb', 'gb [label="b"]'] assert expected == dotter.commands
def test_dotter_str(): dotter = Dotter(strict=True, output_to_file=False, output_type='svg') dotter.add_node('a') dotter.close() expected = ' strict \ndigraph\n {\ngb\ngb [label="a"]\n}' assert str(dotter) == expected
def test_strict_graph(): dotter = Dotter(strict=True) dotter.add_node('a') expected = [' strict ', 'digraph', ' {', 'gb', 'gb [label="a"]'] assert dotter.commands == expected