Added Realtion_count MEthod

This commit is contained in:
Tim
2023-11-05 15:50:28 +01:00
parent 76af89ff32
commit f2ac0eda91
13 changed files with 175 additions and 108 deletions

View File

@ -1,14 +1,11 @@
"""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 = [
@ -32,7 +29,7 @@ def test_initialize_network() -> None:
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"],
assert (
list(metrics.columns)
== ["degree", "eigenvector", "betweeness", "closeness", "pagerank"],
)

View File

@ -1,16 +1,14 @@
"""Test the initialize Network function."""
import datetime
from unittest import TestCase
import networkx as nx
import pandas as pd
import pytest
from aki_prj23_transparenzregister.utils.networkx.network_base import initialize_network
tc = TestCase()
import pytest
@pytest.mark.tim
@pytest.mark.tim()
def test_initialize_network() -> None:
edges: list = [
{"from": "p_545", "to": "c_53", "type": "HAFTENDER_GESELLSCHAFTER"},
@ -33,7 +31,7 @@ def test_initialize_network() -> None:
graph, metrics = initialize_network(edges=edges, nodes=nodes)
assert isinstance(graph, nx.Graph)
assert isinstance(metrics, pd.DataFrame)
tc.assertListEqual(
list(metrics.columns),
["degree", "eigenvector", "betweeness", "closeness", "pagerank"],
assert (
list(metrics.columns)
== ["degree", "eigenvector", "betweeness", "closeness", "pagerank"],
)