From 60dde716b6ac08b57406ebffaad5a3d88c1d32ce Mon Sep 17 00:00:00 2001 From: Thijs Baaijen <13253091+Thijss@users.noreply.github.com> Date: Thu, 23 Jan 2025 15:36:41 +0100 Subject: [PATCH] Add test to verify reversed branches also work with .has_branch() Signed-off-by: Thijs Baaijen <13253091+Thijss@users.noreply.github.com> --- tests/unit/model/graphs/test_graph_model.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/tests/unit/model/graphs/test_graph_model.py b/tests/unit/model/graphs/test_graph_model.py index cea5477..628c77b 100644 --- a/tests/unit/model/graphs/test_graph_model.py +++ b/tests/unit/model/graphs/test_graph_model.py @@ -27,6 +27,16 @@ def test_graph_initialize(graph): assert 1 == graph.nr_branches +def test_graph_has_branch(graph): + graph.add_node(1) + graph.add_node(2) + graph.add_branch(1, 2) + + assert graph.has_branch(1, 2) + assert graph.has_branch(2, 1) # reversed should work too + assert not graph.has_branch(1, 3) + + def test_graph_delete_branch(graph): """Test whether a branch is deleted correctly""" graph.add_node(1)