diff --git a/tests/core/test_modelwrapper.py b/tests/core/test_modelwrapper.py
index 9d30ad64e3eea7421c9d7bad4d510966ae40e5a3..493dd29886f1d3d748085be1718d70fda1b469fd 100644
--- a/tests/core/test_modelwrapper.py
+++ b/tests/core/test_modelwrapper.py
@@ -116,3 +116,8 @@ def test_modelwrapper_graph_order():
     assert model.find_predecessors(Round_node) == [Neg_node]
     assert model.find_predecessors(Ceil_node) == [Neg_node]
     assert model.find_predecessors(Add_node) == [Round_node, Ceil_node]
+
+    assert model.get_node_index(Neg_node) == 0
+    assert model.get_node_index(Round_node) == 1
+    assert model.get_node_index(Ceil_node) == 2
+    assert model.get_node_index(Add_node) == 3