add test from issue #145

Author: aivanoved

tests/test_adapt.py

@@ -238,7 +238,7 @@ def test_inline_model_custom_node_nested(old_squeeze: onnx.ModelProto):
     build({"a": a}, {"c": c})
 
 
-def test_if_adaptation_const():
+def test_if_adapatation_squeeze():
     cond = argument(Tensor(numpy.bool_, ()))
     b = argument(Tensor(numpy.float32, (1,)))
     squeezed = Squeeze11.squeeze11(b, [0])
@@ -255,3 +255,11 @@ def test_if_adaptation_const():
     out = ort.InferenceSession(model.SerializeToString()).run(
         None, {"b": b, "cond": cond}
     )
+
+
+def test_if_adaptation_const():
+    sq = op19.const(1.1453, dtype=numpy.float32)
+    b = argument(Tensor(numpy.float32, ("N",)))
+    cond = op18.equal(sq, b)
+    out = op18.if_(cond, then_branch=lambda: [sq], else_branch=lambda: [sq])
+    build({"b": b}, {"out": out[0]})