erivlis / graphinate

# Drawing a non-planar graph uses spring_layout directly

def test_draw_non_planar_graph(mocker):

    # Arrange

    mock_nx_draw = mocker.patch('networkx.draw')

    mock_planar_layout = mocker.patch('networkx.planar_layout')

    mock_spring_layout = mocker.patch('networkx.spring_layout', return_value='spring_pos')

    mock_is_planar = mocker.patch('networkx.is_planar', return_value=False)

    mock_node_color_mapping = mocker.patch('graphinate.renderers.matplotlib.node_color_mapping',

                                           return_value={0: 'red', 1: 'blue'})

    graph = nx.Graph()

    graph.add_node(0)

    graph.add_node(1)

    graph.add_edge(0, 1)

    # Act

    draw(graph)

    # Assert

    mock_is_planar.assert_called_once_with(graph)

    mock_planar_layout.assert_not_called()

    mock_spring_layout.assert_called_once_with(graph)

    mock_node_color_mapping.assert_called_once_with(graph)

    mock_nx_draw.assert_called_once()

# Drawing an empty graph

# Drawing a planar graph uses planar_layout first

def test_draw_planar_graph(mocker):

    # Arrange

    mock_nx_draw = mocker.patch('networkx.draw')

    mock_planar_layout = mocker.patch('networkx.planar_layout', return_value='planar_pos')

    mock_spring_layout = mocker.patch('networkx.spring_layout', return_value='spring_pos')

    mock_is_planar = mocker.patch('networkx.is_planar', return_value=True)

    mock_node_color_mapping = mocker.patch('graphinate.renderers.matplotlib.node_color_mapping',

                                           return_value={0: 'red', 1: 'blue'})

    graph = nx.Graph()

    graph.add_node(0)

    graph.add_node(1)

    graph.add_edge(0, 1)

    # Act

    draw(graph)

    # Assert

    mock_is_planar.assert_called_once_with(graph)

    mock_planar_layout.assert_called_once_with(graph)

    mock_spring_layout.assert_called_once_with(graph, pos='planar_pos')

    mock_node_color_mapping.assert_called_once_with(graph)

    mock_nx_draw.assert_called_once()

# Drawing a non-planar graph uses spring_layout directly