add option to disable edge rescaling to provide edge weights as integers, closes #1

This commit is contained in:
2025-06-20 09:06:51 +02:00
parent a7718c12cd
commit 63eb274975
7 changed files with 53 additions and 14 deletions

View File

@@ -28,14 +28,26 @@ def test_build_tk_graph_post_pipe():
assert len(pipe.actions) == 3
def test_build_tk_graph_rescaling_pipe():
def test_build_tk_graph_rescaling_pipe_with_edge_rescaling():
pipe = pre.build_tk_graph_rescaling_pipe(
save_result=False, exit_point=EntryPoints.TK_GRAPH_ANALYSIS_RESCALED
save_result=False,
exit_point=EntryPoints.TK_GRAPH_ANALYSIS_RESCALED,
enable_rescaling=True,
)
assert pipe.name == 'Graph_Rescaling'
assert len(pipe.actions) == 2
def test_build_tk_graph_rescaling_pipe_without_edge_rescaling():
pipe = pre.build_tk_graph_rescaling_pipe(
save_result=False,
exit_point=EntryPoints.TK_GRAPH_ANALYSIS_RESCALED,
enable_rescaling=False,
)
assert pipe.name == 'Graph_Rescaling'
assert len(pipe.actions) == 1
@pytest.mark.parametrize('with_subgraphs', [True, False])
def test_build_tk_graph_render_pipe(with_subgraphs):
pipe = pre.build_tk_graph_render_pipe(with_subgraphs=with_subgraphs)