erivlis / graphinate

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

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

    def test_draw_no_edge_labels_attribute(self, 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 edge_labels parameter is empty dict

        args, kwargs = mock_nx_draw_edge_labels.call_args

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

        assert kwargs.get('node_color') == []

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

    def test_draw_no_node_labels_attribute(self, 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

    def test_draw_no_edge_labels_attribute(self, mocker):