Test Version

tests/utils/networkx/network_2d_test.py

@@ -0,0 +1,38 @@
+"""Test the initialize Network function."""
+import datetime
+from unittest import TestCase
+
+import networkx as nx
+import pandas as pd
+
+from aki_prj23_transparenzregister.utils.networkx.network_base import initialize_network
+
+tc = TestCase()
+
+
+def test_initialize_network() -> None:
+    edges: list = [
+        {"from": "p_545", "to": "c_53", "type": "HAFTENDER_GESELLSCHAFTER"},
+        {"from": "p_758", "to": "c_77", "type": "HAFTENDER_GESELLSCHAFTER"},
+    ]
+    nodes: dict = {
+        "c_53": {
+            "id": "c_53",
+            "name": "1. Freiburger Solarfonds Beteiligungs-KG",
+            "color": "blue",
+        },
+        "p_545": {
+            "id": "p_545",
+            "firstname": "Jürgen",
+            "lastname": "Wetzel",
+            "date_of_birth": datetime.date(1962, 11, 15),
+            "color": "red",
+        },
+    }
+    graph, metrics = initialize_network(edges=edges, nodes=nodes)
+    assert type(graph) is nx.Graph
+    assert type(metrics) is pd.DataFrame
+    tc.assertListEqual(
+        list(metrics.columns),
+        ["degree", "eigenvector", "betweeness", "closeness", "pagerank"],
+    )

tests/utils/networkx/network_3d_test.py

@@ -0,0 +1,38 @@
+"""Test the initialize Network function."""
+import datetime
+from unittest import TestCase
+
+import networkx as nx
+import pandas as pd
+
+from aki_prj23_transparenzregister.utils.networkx.network_base import initialize_network
+
+tc = TestCase()
+
+
+def test_initialize_network() -> None:
+    edges: list = [
+        {"from": "p_545", "to": "c_53", "type": "HAFTENDER_GESELLSCHAFTER"},
+        {"from": "p_758", "to": "c_77", "type": "HAFTENDER_GESELLSCHAFTER"},
+    ]
+    nodes: dict = {
+        "c_53": {
+            "id": "c_53",
+            "name": "1. Freiburger Solarfonds Beteiligungs-KG",
+            "color": "blue",
+        },
+        "p_545": {
+            "id": "p_545",
+            "firstname": "Jürgen",
+            "lastname": "Wetzel",
+            "date_of_birth": datetime.date(1962, 11, 15),
+            "color": "red",
+        },
+    }
+    graph, metrics = initialize_network(edges=edges, nodes=nodes)
+    assert type(graph) is nx.Graph
+    assert type(metrics) is pd.DataFrame
+    tc.assertListEqual(
+        list(metrics.columns),
+        ["degree", "eigenvector", "betweeness", "closeness", "pagerank"],
+    )

tests/utils/networkx/network_base_test.py

@@ -0,0 +1,38 @@
+"""Test the initialize Network function."""
+import datetime
+from unittest import TestCase
+
+import networkx as nx
+import pandas as pd
+
+from aki_prj23_transparenzregister.utils.networkx.network_base import initialize_network
+
+tc = TestCase()
+
+
+def test_initialize_network() -> None:
+    edges: list = [
+        {"from": "p_545", "to": "c_53", "type": "HAFTENDER_GESELLSCHAFTER"},
+        {"from": "p_758", "to": "c_77", "type": "HAFTENDER_GESELLSCHAFTER"},
+    ]
+    nodes: dict = {
+        "c_53": {
+            "id": "c_53",
+            "name": "1. Freiburger Solarfonds Beteiligungs-KG",
+            "color": "blue",
+        },
+        "p_545": {
+            "id": "p_545",
+            "firstname": "Jürgen",
+            "lastname": "Wetzel",
+            "date_of_birth": datetime.date(1962, 11, 15),
+            "color": "red",
+        },
+    }
+    graph, metrics = initialize_network(edges=edges, nodes=nodes)
+    assert type(graph) is nx.Graph
+    assert type(metrics) is pd.DataFrame
+    tc.assertListEqual(
+        list(metrics.columns),
+        ["degree", "eigenvector", "betweeness", "closeness", "pagerank"],
+    )

tests/utils/networkx/networkx_data_test.py

@@ -0,0 +1,38 @@
+"""Test the initialize Network function."""
+import datetime
+from unittest import TestCase
+
+import networkx as nx
+import pandas as pd
+
+from aki_prj23_transparenzregister.utils.networkx.network_base import initialize_network
+
+tc = TestCase()
+
+
+def test_initialize_network() -> None:
+    edges: list = [
+        {"from": "p_545", "to": "c_53", "type": "HAFTENDER_GESELLSCHAFTER"},
+        {"from": "p_758", "to": "c_77", "type": "HAFTENDER_GESELLSCHAFTER"},
+    ]
+    nodes: dict = {
+        "c_53": {
+            "id": "c_53",
+            "name": "1. Freiburger Solarfonds Beteiligungs-KG",
+            "color": "blue",
+        },
+        "p_545": {
+            "id": "p_545",
+            "firstname": "Jürgen",
+            "lastname": "Wetzel",
+            "date_of_birth": datetime.date(1962, 11, 15),
+            "color": "red",
+        },
+    }
+    graph, metrics = initialize_network(edges=edges, nodes=nodes)
+    assert type(graph) is nx.Graph
+    assert type(metrics) is pd.DataFrame
+    tc.assertListEqual(
+        list(metrics.columns),
+        ["degree", "eigenvector", "betweeness", "closeness", "pagerank"],
+    )