From 201c91d6c21c722d891b1958ab7f95bb3c520472 Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <maltanar@gmail.com>
Date: Mon, 21 Oct 2019 23:34:13 +0100
Subject: [PATCH] [Test] add test_give_unique_node_names
---
tests/test_general_transformation.py | 31 ++++++++++++++++++++++++++++
1 file changed, 31 insertions(+)
create mode 100644 tests/test_general_transformation.py
diff --git a/tests/test_general_transformation.py b/tests/test_general_transformation.py
new file mode 100644
index 000000000..037083884
--- /dev/null
+++ b/tests/test_general_transformation.py
@@ -0,0 +1,31 @@
+import hashlib
+import os
+import shutil
+
+import wget
+
+import finn.transformation.general as tg
+from finn.core.modelwrapper import ModelWrapper
+
+mnist_onnx_url_base = "https://onnxzoo.blob.core.windows.net/models/opset_8/mnist"
+mnist_onnx_filename = "mnist.tar.gz"
+mnist_onnx_local_dir = "/tmp/mnist_onnx"
+
+
+def test_give_unique_node_names():
+ try:
+ os.remove("/tmp/" + mnist_onnx_filename)
+ except OSError:
+ pass
+ dl_ret = wget.download(mnist_onnx_url_base + "/" + mnist_onnx_filename, out="/tmp")
+ shutil.unpack_archive(dl_ret, mnist_onnx_local_dir)
+ with open(mnist_onnx_local_dir + "/mnist/model.onnx", "rb") as f:
+ assert hashlib.md5(f.read()).hexdigest() == "d7cd24a0a76cd492f31065301d468c3d"
+ model = ModelWrapper(mnist_onnx_local_dir + "/mnist/model.onnx")
+ 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"
+ # remove the downloaded model and extracted files
+ os.remove(dl_ret)
+ shutil.rmtree(mnist_onnx_local_dir)
--
GitLab