(self)
| 35 | f.draw_node.assert_called_with(step, f.blueprint_scheme, {'x': 3}) |
| 36 | |
| 37 | def test_edge(self): |
| 38 | f = bootsteps.StepFormatter() |
| 39 | f.draw_edge = Mock() |
| 40 | a, b = Mock(), Mock() |
| 41 | a.last = True |
| 42 | f.edge(a, b, x=6) |
| 43 | f.draw_edge.assert_called_with(a, b, f.edge_scheme, { |
| 44 | 'x': 6, 'arrowhead': 'none', 'color': 'darkseagreen3', |
| 45 | }) |
| 46 | |
| 47 | a.last = False |
| 48 | f.edge(a, b, x=6) |
| 49 | f.draw_edge.assert_called_with(a, b, f.edge_scheme, { |
| 50 | 'x': 6, |
| 51 | }) |
| 52 | |
| 53 | |
| 54 | class test_Step: |