From 22c0bd434a63c032432d3bfeb82c37a4194ee8d4 Mon Sep 17 00:00:00 2001
From: auphelia <jakobapk@web.de>
Date: Thu, 30 Apr 2020 17:49:34 +0100
Subject: [PATCH] [Test] Add test for get_node_index modelwrapper function
---
tests/core/test_modelwrapper.py | 5 +++++
1 file changed, 5 insertions(+)
diff --git a/tests/core/test_modelwrapper.py b/tests/core/test_modelwrapper.py
index 9d30ad64e..493dd2988 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
--
GitLab