solved errors

This commit is contained in:
Tim
2023-10-25 20:10:29 +02:00
parent 7e8adfafd5
commit 30f9e4506f
6 changed files with 69 additions and 71 deletions

View File

@ -8,8 +8,9 @@ import pandas as pd
from aki_prj23_transparenzregister.utils.networkx.network_base import initialize_network
tc = TestCase()
import pytest
@pytest.mark.tim
def test_initialize_network() -> None:
edges: list = [
{"from": "p_545", "to": "c_53", "type": "HAFTENDER_GESELLSCHAFTER"},
@ -30,8 +31,8 @@ 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
assert isinstance(graph, nx.Graph)
assert isinstance(metrics, pd.DataFrame)
tc.assertListEqual(
list(metrics.columns),
["degree", "eigenvector", "betweeness", "closeness", "pagerank"],