From ff6f18604b03937ba70a03d94df152500e60895e Mon Sep 17 00:00:00 2001
From: auphelia <jakobapk@web.de>
Date: Tue, 10 Dec 2019 10:58:47 +0000
Subject: [PATCH] [Test] Added test function to test verification of
 MultiThreshold node

---
 tests/test_verify_custom_ops.py | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)
 create mode 100644 tests/test_verify_custom_ops.py

diff --git a/tests/test_verify_custom_ops.py b/tests/test_verify_custom_ops.py
new file mode 100644
index 000000000..2f0e654a4
--- /dev/null
+++ b/tests/test_verify_custom_ops.py
@@ -0,0 +1,17 @@
+import onnx
+from finn.custom_op.layout_custom_ops import CustomOp_Layout
+
+def test_verify_layout_custom_ops():
+    m_node = onnx.helper.make_node(
+        "MultiThreshold",
+        ["v", "thresholds"],
+        ["out"],
+        domain="finn",
+        out_scale=2.0,
+        out_bias=-1.0,
+        out_dtype="",
+    ) 
+
+    inst = CustomOp_Layout[m_node.op_type] 
+    inst.verify_layout(m_node) 
+
-- 
GitLab