erivlis / graphinate

# Drawing a graph with no edge attributes for labels when with_edge_labels=True

def test_draw_no_edge_labels_attribute(mocker):

    # Arrange

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

    mock_nx_draw_edge_labels = mocker.patch('networkx.draw_networkx_edge_labels')

    mock_get_edge_attributes = mocker.patch('networkx.get_edge_attributes', return_value={})

    mock_spring_layout = mocker.patch('networkx.spring_layout', return_value='mock_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)  # No label attribute

    # Act

    draw(graph, with_edge_labels=True)

    # Assert

    mock_is_planar.assert_called_once_with(graph)

    mock_spring_layout.assert_called_once_with(graph)

    mock_node_color_mapping.assert_called_once_with(graph)

    mock_get_edge_attributes.assert_called_once_with(graph, 'label')

    mock_nx_draw.assert_called_once()

    mock_nx_draw_edge_labels.assert_called_once()

    # Check that the edge_labels parameter is an empty dict

    args, kwargs = mock_nx_draw_edge_labels.call_args

    assert kwargs.get('edge_labels') == {}

# Drawing a graph with no node attributes for labels when with_node_labels=True

def test_draw_no_node_labels_attribute(mocker):

    # Arrange

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

    mock_get_node_attributes = mocker.patch('networkx.get_node_attributes', return_value={})

    mock_spring_layout = mocker.patch('networkx.spring_layout', return_value='mock_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)  # No label attribute

    graph.add_node(1)  # No label attribute

    graph.add_edge(0, 1)

    # Act

    draw(graph, with_node_labels=True)

    # Assert

    mock_is_planar.assert_called_once_with(graph)

    mock_spring_layout.assert_called_once_with(graph)

    mock_node_color_mapping.assert_called_once_with(graph)

    mock_get_node_attributes.assert_called_once_with(graph, 'label')

    mock_nx_draw.assert_called_once()

    # Check that labels parameter is empty dict

    args, kwargs = mock_nx_draw.call_args

    assert kwargs.get('labels') == {}

# Drawing a graph with no edge attributes for labels when with_edge_labels=True