diff --git a/tests/test_general_transformation.py b/tests/test_general_transformation.py index 090c72371a6a14a70c69e2f5f3889dcc1e3b5617..d4376cca6051bd45669c5bb1a6102dc7db76e3b4 100644 --- a/tests/test_general_transformation.py +++ b/tests/test_general_transformation.py @@ -9,5 +9,5 @@ def test_give_unique_node_names(): model = ModelWrapper(raw_m) model = model.transform_single(tg.give_unique_node_names) assert model.graph.node[0].name == "Reshape_0" - assert model.graph.node[1].name == "Conv_1" - assert model.graph.node[11].name == "Add_11" + assert model.graph.node[1].name == "Conv_0" + assert model.graph.node[11].name == "Add_2"