test_builders.test_build_graph_with_custom_layout() - Code Metrics - erivlis/networkx-mermaid - Measure and Improve Code Quality continuously with Scrutinizer

test_build_graph_with_custom_layout() A
last analyzed 2025-11-08 19:20 UTC

↳ Parent: test_builders

Complexity

Conditions

Size

Total Lines	12
Code Lines	7

Duplication

Lines	0
Ratio	0 %

Importance

Changes

Metric	Value
cc	1
eloc	7
nop	0
dl	0
loc	12
rs	10
c	0
b	0
f	0

import networkx as nx
import pytest

from networkx_mermaid import DiagramNodeShape, DiagramOrientation
from networkx_mermaid.builders import (
    DEFAULT_LAYOUT,
    DEFAULT_LOOK,
    DEFAULT_THEME,
    DiagramBuilder,
    _contrast_color,
    _edge_label,
    _graph_title,
    _node_style,
)


def test_default_initialization():
    # Arrange
    builder = DiagramBuilder()

    # Act & Assert
    assert builder.orientation == DiagramOrientation.LEFT_RIGHT
    assert builder.node_shape == DiagramNodeShape.DEFAULT
    assert builder.layout == DEFAULT_LAYOUT
    assert builder.look == DEFAULT_LOOK
    assert builder.theme == DEFAULT_THEME


def test_custom_initialization():
    # Arrange
    builder = DiagramBuilder(
        orientation=DiagramOrientation.TOP_DOWN,
        node_shape=DiagramNodeShape.RECTANGLE,
        layout="elk",
        look="neo",
        theme="neutral"
    )

    # Act & Assert
    assert builder.orientation == DiagramOrientation.TOP_DOWN
    assert builder.node_shape == DiagramNodeShape.RECTANGLE
    assert builder.layout == "elk"
    assert builder.look == "neo"
    assert builder.theme == "neutral"


def test_invalid_orientation_type():
    # Arrange & Act & Assert
    with pytest.raises(TypeError):
        DiagramBuilder(orientation="invalid")


def test_invalid_node_shape_type():
    # Arrange & Act & Assert
    with pytest.raises(TypeError):
        DiagramBuilder(node_shape="invalid")


def test_build_empty_graph():
    # Arrange
    graph = nx.Graph()
    builder = DiagramBuilder()

    # Act
    diagram = builder.build(graph)

    # Assert
    assert "graph LR" in diagram


def test_build_simple_graph():
    # Arrange
    graph = nx.Graph()
    graph.add_edge(1, 2)
    builder = DiagramBuilder()

    # Act
    diagram = builder.build(graph)

    # Assert
    assert "graph LR" in diagram
    assert "A(1)" in diagram
    assert "B(2)" in diagram
    assert "A --> B" in diagram


def test_build_graph_with_node_labels():
    # Arrange
    graph = nx.Graph()
    graph.add_node(1, label="Node 1")
    graph.add_node(2, label="Node 2")
    graph.add_edge(1, 2)
    builder = DiagramBuilder()

    # Act
    diagram = builder.build(graph)

    # Assert
    assert "A(Node 1)" in diagram
    assert "B(Node 2)" in diagram
    assert "A --> B" in diagram


def test_build_graph_with_edge_labels():
    # Arrange
    graph = nx.Graph()
    graph.add_edge(1, 2, label="Edge 1-2")
    builder = DiagramBuilder()

    # Act
    diagram = builder.build(graph)

    # Assert
    assert "A -->|Edge 1-2| B" in diagram


def test_build_graph_without_edge_labels():
    # Arrange
    graph = nx.Graph()
    graph.add_edge(1, 2, label="Edge 1-2")
    builder = DiagramBuilder()

    # Act
    diagram = builder.build(graph, with_edge_labels=False)

    # Assert
    assert "A --> B" in diagram
    assert "|Edge 1-2|" not in diagram


def test_build_graph_with_node_colors():
    # Arrange
    graph = nx.Graph()
    graph.add_node(1, color="#FF0000")
    graph.add_node(2, color="#00FF00")
    graph.add_edge(1, 2)
    builder = DiagramBuilder()

    # Act
    diagram = builder.build(graph)

    # Assert
    assert "style A fill:#FF0000, color:#ffffff" in diagram
    assert "style B fill:#00FF00, color:#ffffff" in diagram


def test_build_graph_with_custom_node_shape():
    # Arrange
    graph = nx.Graph()
    graph.add_node(1)
    graph.add_edge(1, 1)
    builder = DiagramBuilder(node_shape=DiagramNodeShape.RECTANGLE)

    # Act
    diagram = builder.build(graph)

    # Assert
    assert "A[1]" in diagram


def test_build_graph_with_custom_orientation():
    # Arrange
    graph = nx.Graph()
    graph.add_node(1)
    graph.add_edge(1, 1)
    builder = DiagramBuilder(orientation=DiagramOrientation.TOP_DOWN)

    # Act
    diagram = builder.build(graph)

    # Assert
    assert "graph TD" in diagram


def test_build_graph_with_custom_layout():
    # Arrange
    graph = nx.Graph()
    graph.add_node(1)
    graph.add_edge(1, 1)
    builder = DiagramBuilder(layout='elk')

    # Act
    diagram = builder.build(graph)

    # Assert
    assert "layout: elk" in diagram


def test_build_graph_with_custom_look():
    # Arrange
    graph = nx.Graph()
    graph.add_node(1)
    graph.add_edge(1, 1)
    builder = DiagramBuilder(look='neo')

    # Act
    diagram = builder.build(graph)

    # Assert
    assert "look: neo" in diagram


def test_build_graph_with_custom_theme():
    # Arrange
    graph = nx.Graph()
    graph.add_node(1)
    graph.add_edge(1, 1)
    builder = DiagramBuilder(theme='neutral')

    # Act
    diagram = builder.build(graph)

    # Assert
    assert "theme: neutral" in diagram


def test_build_graph_with_title():
    # Arrange
    graph = nx.Graph(name="My Graph")
    graph.add_node(1)
    graph.add_edge(1, 1)
    builder = DiagramBuilder()

    # Act
    diagram = builder.build(graph)

    # Assert
    assert "title: My Graph" in diagram


def test_build_graph_without_title():
    # Arrange
    graph = nx.Graph()
    graph.add_node(1)
    graph.add_edge(1, 1)
    builder = DiagramBuilder()

    # Act
    diagram = builder.build(graph)

    # Assert
    assert "title" not in diagram


def test_build_graph_with_title_overwrite():
    # Arrange
    graph = nx.Graph(name="My Graph")
    graph.add_node(1)
    graph.add_edge(1, 1)
    builder = DiagramBuilder()

    # Act
    diagram = builder.build(graph, title="Custom Title")

    # Assert
    assert "title: Custom Title" in diagram
    assert "title: My Graph" not in diagram


def test_build_graph_with_empty_title_overwrite():
    # Arrange
    graph = nx.Graph(name="My Graph")
    graph.add_node(1)
    graph.add_edge(1, 1)
    builder = DiagramBuilder()

    # Act
    diagram = builder.build(graph, title="")

    # Assert
    assert "title" not in diagram


def test_edge_label_with_label():
    # Arrange
    data = {"label": "Edge Label"}

    # Act
    result = _edge_label(data)

    # Assert
    assert result == "|Edge Label|"


def test_edge_label_without_label():
    # Arrange
    data = {}

    # Act
    result = _edge_label(data)

    # Assert
    assert result == ""


@pytest.mark.parametrize(
    ('color', 'expected_contrast'),
    [
        ("#FFFFFF", "#000000"),
        ("#000000", "#ffffff"),
        ("#FF0000", "#ffffff"),
        ("#00FF00", "#ffffff"),
        ("#0000FF", "#ffffff"),
        ("#ABCDEF", "#000000"),
        ("#123456", "#ffffff"),
    ],
)
def test_contrast_color_valid_hex(color, expected_contrast):
    # Arrange (done by parametrizing)
    # Act
    result = _contrast_color(color)

    # Assert
    assert result == expected_contrast


@pytest.mark.parametrize(
    "color",
    ["FFFFFF", "#FFFFF", "#GGGGGG", 123456, None],
)
def test_contrast_color_invalid_hex(color):
    # Arrange (done by parametrizing)
    # Act & Assert
    with pytest.raises(ValueError):  # noqa: PT011
        _contrast_color(color)


def test_node_style_with_color():
    # Arrange
    data = {"color": "#FF0000"}

    # Act
    result = _node_style("1", data)

    # Assert
    assert result == "\nstyle 1 fill:#FF0000, color:#ffffff"


def test_node_style_without_color():
    # Arrange
    data = {}

    # Act
    result = _node_style("1", data)

    # Assert
    assert result == ""


def test_graph_title_with_name():
    # Arrange
    graph = nx.Graph(name="My Graph")

    # Act
    result = _graph_title(graph)

    # Assert
    assert result == "title: My Graph\n"


def test_graph_title_without_name():
    # Arrange
    graph = nx.Graph()

    # Act
    result = _graph_title(graph)

    # Assert
    assert result == ""


1			import networkx as nx
2			import pytest
3
4			from networkx_mermaid import DiagramNodeShape, DiagramOrientation
5			from networkx_mermaid.builders import (
6			DEFAULT_LAYOUT,
7			DEFAULT_LOOK,
8			DEFAULT_THEME,
9			DiagramBuilder,
10			_contrast_color,
11			_edge_label,
12			_graph_title,
13			_node_style,
14			)
15
16
17			def test_default_initialization():
18			# Arrange
19			builder = DiagramBuilder()
20
21			# Act & Assert
22			assert builder.orientation == DiagramOrientation.LEFT_RIGHT
23			assert builder.node_shape == DiagramNodeShape.DEFAULT
24			assert builder.layout == DEFAULT_LAYOUT
25			assert builder.look == DEFAULT_LOOK
26			assert builder.theme == DEFAULT_THEME
27
28
29			def test_custom_initialization():
30			# Arrange
31			builder = DiagramBuilder(
32			orientation=DiagramOrientation.TOP_DOWN,
33			node_shape=DiagramNodeShape.RECTANGLE,
34			layout="elk",
35			look="neo",
36			theme="neutral"
37			)
38
39			# Act & Assert
40			assert builder.orientation == DiagramOrientation.TOP_DOWN
41			assert builder.node_shape == DiagramNodeShape.RECTANGLE
42			assert builder.layout == "elk"
43			assert builder.look == "neo"
44			assert builder.theme == "neutral"
45
46
47			def test_invalid_orientation_type():
48			# Arrange & Act & Assert
49			with pytest.raises(TypeError):
50			DiagramBuilder(orientation="invalid")
51
52
53			def test_invalid_node_shape_type():
54			# Arrange & Act & Assert
55			with pytest.raises(TypeError):
56			DiagramBuilder(node_shape="invalid")
57
58
59			def test_build_empty_graph():
60			# Arrange
61			graph = nx.Graph()
62			builder = DiagramBuilder()
63
64			# Act
65			diagram = builder.build(graph)
66
67			# Assert
68			assert "graph LR" in diagram
69
70
71			def test_build_simple_graph():
72			# Arrange
73			graph = nx.Graph()
74			graph.add_edge(1, 2)
75			builder = DiagramBuilder()
76
77			# Act
78			diagram = builder.build(graph)
79
80			# Assert
81			assert "graph LR" in diagram
82			assert "A(1)" in diagram
83			assert "B(2)" in diagram
84			assert "A --> B" in diagram
85
86
87			def test_build_graph_with_node_labels():
88			# Arrange
89			graph = nx.Graph()
90			graph.add_node(1, label="Node 1")
91			graph.add_node(2, label="Node 2")
92			graph.add_edge(1, 2)
93			builder = DiagramBuilder()
94
95			# Act
96			diagram = builder.build(graph)
97
98			# Assert
99			assert "A(Node 1)" in diagram
100			assert "B(Node 2)" in diagram
101			assert "A --> B" in diagram
102
103
104			def test_build_graph_with_edge_labels():
105			# Arrange
106			graph = nx.Graph()
107			graph.add_edge(1, 2, label="Edge 1-2")
108			builder = DiagramBuilder()
109
110			# Act
111			diagram = builder.build(graph)
112
113			# Assert
114			assert "A -->\|Edge 1-2\| B" in diagram
115
116
117			def test_build_graph_without_edge_labels():
118			# Arrange
119			graph = nx.Graph()
120			graph.add_edge(1, 2, label="Edge 1-2")
121			builder = DiagramBuilder()
122
123			# Act
124			diagram = builder.build(graph, with_edge_labels=False)
125
126			# Assert
127			assert "A --> B" in diagram
128			assert "\|Edge 1-2\|" not in diagram
129
130
131			def test_build_graph_with_node_colors():
132			# Arrange
133			graph = nx.Graph()
134			graph.add_node(1, color="#FF0000")
135			graph.add_node(2, color="#00FF00")
136			graph.add_edge(1, 2)
137			builder = DiagramBuilder()
138
139			# Act
140			diagram = builder.build(graph)
141
142			# Assert
143			assert "style A fill:#FF0000, color:#ffffff" in diagram
144			assert "style B fill:#00FF00, color:#ffffff" in diagram
145
146
147			def test_build_graph_with_custom_node_shape():
148			# Arrange
149			graph = nx.Graph()
150			graph.add_node(1)
151			graph.add_edge(1, 1)
152			builder = DiagramBuilder(node_shape=DiagramNodeShape.RECTANGLE)
153
154			# Act
155			diagram = builder.build(graph)
156
157			# Assert
158			assert "A[1]" in diagram
159
160
161			def test_build_graph_with_custom_orientation():
162			# Arrange
163			graph = nx.Graph()
164			graph.add_node(1)
165			graph.add_edge(1, 1)
166			builder = DiagramBuilder(orientation=DiagramOrientation.TOP_DOWN)
167
168			# Act
169			diagram = builder.build(graph)
170
171			# Assert
172			assert "graph TD" in diagram
173
174
175			def test_build_graph_with_custom_layout():
176			# Arrange
177			graph = nx.Graph()
178			graph.add_node(1)
179			graph.add_edge(1, 1)
180			builder = DiagramBuilder(layout='elk')
181
182			# Act
183			diagram = builder.build(graph)
184
185			# Assert
186			assert "layout: elk" in diagram
187
188
189			def test_build_graph_with_custom_look():
190			# Arrange
191			graph = nx.Graph()
192			graph.add_node(1)
193			graph.add_edge(1, 1)
194			builder = DiagramBuilder(look='neo')
195
196			# Act
197			diagram = builder.build(graph)
198
199			# Assert
200			assert "look: neo" in diagram
201
202
203			def test_build_graph_with_custom_theme():
204			# Arrange
205			graph = nx.Graph()
206			graph.add_node(1)
207			graph.add_edge(1, 1)
208			builder = DiagramBuilder(theme='neutral')
209
210			# Act
211			diagram = builder.build(graph)
212
213			# Assert
214			assert "theme: neutral" in diagram
215
216
217			def test_build_graph_with_title():
218			# Arrange
219			graph = nx.Graph(name="My Graph")
220			graph.add_node(1)
221			graph.add_edge(1, 1)
222			builder = DiagramBuilder()
223
224			# Act
225			diagram = builder.build(graph)
226
227			# Assert
228			assert "title: My Graph" in diagram
229
230
231			def test_build_graph_without_title():
232			# Arrange
233			graph = nx.Graph()
234			graph.add_node(1)
235			graph.add_edge(1, 1)
236			builder = DiagramBuilder()
237
238			# Act
239			diagram = builder.build(graph)
240
241			# Assert
242			assert "title" not in diagram
243
244
245			def test_build_graph_with_title_overwrite():
246			# Arrange
247			graph = nx.Graph(name="My Graph")
248			graph.add_node(1)
249			graph.add_edge(1, 1)
250			builder = DiagramBuilder()
251
252			# Act
253			diagram = builder.build(graph, title="Custom Title")
254
255			# Assert
256			assert "title: Custom Title" in diagram
257			assert "title: My Graph" not in diagram
258
259
260			def test_build_graph_with_empty_title_overwrite():
261			# Arrange
262			graph = nx.Graph(name="My Graph")
263			graph.add_node(1)
264			graph.add_edge(1, 1)
265			builder = DiagramBuilder()
266
267			# Act
268			diagram = builder.build(graph, title="")
269
270			# Assert
271			assert "title" not in diagram
272
273
274			def test_edge_label_with_label():
275			# Arrange
276			data = {"label": "Edge Label"}
277
278			# Act
279			result = _edge_label(data)
280
281			# Assert
282			assert result == "\|Edge Label\|"
283
284
285			def test_edge_label_without_label():
286			# Arrange
287			data = {}
288
289			# Act
290			result = _edge_label(data)
291
292			# Assert
293			assert result == ""
294
295
296			@pytest.mark.parametrize(
297			('color', 'expected_contrast'),
298			[
299			("#FFFFFF", "#000000"),
300			("#000000", "#ffffff"),
301			("#FF0000", "#ffffff"),
302			("#00FF00", "#ffffff"),
303			("#0000FF", "#ffffff"),
304			("#ABCDEF", "#000000"),
305			("#123456", "#ffffff"),
306			],
307			)
308			def test_contrast_color_valid_hex(color, expected_contrast):
309			# Arrange (done by parametrizing)
310			# Act
311			result = _contrast_color(color)
312
313			# Assert
314			assert result == expected_contrast
315
316
317			@pytest.mark.parametrize(
318			"color",
319			["FFFFFF", "#FFFFF", "#GGGGGG", 123456, None],
320			)
321			def test_contrast_color_invalid_hex(color):
322			# Arrange (done by parametrizing)
323			# Act & Assert
324			with pytest.raises(ValueError): # noqa: PT011
325			_contrast_color(color)
326
327
328			def test_node_style_with_color():
329			# Arrange
330			data = {"color": "#FF0000"}
331
332			# Act
333			result = _node_style("1", data)
334
335			# Assert
336			assert result == "\nstyle 1 fill:#FF0000, color:#ffffff"
337
338
339			def test_node_style_without_color():
340			# Arrange
341			data = {}
342
343			# Act
344			result = _node_style("1", data)
345
346			# Assert
347			assert result == ""
348
349
350			def test_graph_title_with_name():
351			# Arrange
352			graph = nx.Graph(name="My Graph")
353
354			# Act
355			result = _graph_title(graph)
356
357			# Assert
358			assert result == "title: My Graph\n"
359
360
361			def test_graph_title_without_name():
362			# Arrange
363			graph = nx.Graph()
364
365			# Act
366			result = _graph_title(graph)
367
368			# Assert
369			assert result == ""
370

erivlis / networkx-mermaid

test_build_graph_with_custom_layout() A last analyzed 2025-11-08 19:20 UTC

Complexity

Size

Duplication

Importance

Duplication Side-by-Side

Filter issues like

test_build_graph_with_custom_layout() A
last analyzed 2025-11-08 19:20 UTC