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_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_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
def test_strict_graph():
dotter = Dotter(strict=True)
dotter.add_node('a')
expected = [' strict ', 'digraph', ' {', 'gb', 'gb [label="a"]']
assert dotter.commands == expected
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
