From b36f8523e6f2c7ba4a2cc18d75f1a831a56855e8 Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <maltanar@gmail.com>
Date: Tue, 27 Apr 2021 15:10:45 +0100
Subject: [PATCH] [Test] improve the end2end cybsec mlp test
---
tests/end2end/test_end2end_cybsec_mlp.py | 6 +++++-
1 file changed, 5 insertions(+), 1 deletion(-)
diff --git a/tests/end2end/test_end2end_cybsec_mlp.py b/tests/end2end/test_end2end_cybsec_mlp.py
index 4dba19f58..57e873516 100644
--- a/tests/end2end/test_end2end_cybsec_mlp.py
+++ b/tests/end2end/test_end2end_cybsec_mlp.py
@@ -123,11 +123,15 @@ def test_end2end_cybsec_mlp_export():
finn_model.set_tensor_datatype(finnonnx_in_tensor_name, DataType.BIPOLAR)
finn_model.save(export_onnx_path)
assert tuple(finn_model.get_tensor_shape(finnonnx_in_tensor_name)) == (1, 600)
- assert len(finn_model.graph.node) == 30
+ # verify a few exported ops
assert finn_model.graph.node[0].op_type == "Add"
assert finn_model.graph.node[1].op_type == "Div"
assert finn_model.graph.node[2].op_type == "MatMul"
assert finn_model.graph.node[-1].op_type == "MultiThreshold"
+ # verify datatypes on some tensors
+ assert finn_model.get_tensor_datatype(finnonnx_in_tensor_name) == DataType.BIPOLAR
+ first_matmul_w_name = finn_model.graph.node[2].input[1]
+ assert finn_model.get_tensor_datatype(first_matmul_w_name) == DataType.INT2
@pytest.mark.slow
--
GitLab