aboutsummaryrefslogtreecommitdiff
path: root/python/pyarmnn/test/test_generated.py
diff options
context:
space:
mode:
authorwangg <gemini910621@gmail.com>2020-08-26 01:44:32 +0000
committerJim Flynn <jim.flynn@arm.com>2020-10-21 12:16:32 +0000
commit5f960d92dbefabf708e8f299789b35f6cdf2d919 (patch)
tree7309f46bfdef2a319c987d0492888d4aee93a509 /python/pyarmnn/test/test_generated.py
parent45274909b06a4882ada92899c58ee66194446135 (diff)
downloadarmnn-5f960d92dbefabf708e8f299789b35f6cdf2d919.tar.gz
Add IDeserializer support in pyarmnn
Resources required for new unit tests are included for review. Signed-off-by: Guanqun Wang gemini910621@gmail.com Change-Id: Iead6cb5beaf824a6f467ad9da4aede5719ebe4ec
Diffstat (limited to 'python/pyarmnn/test/test_generated.py')
-rw-r--r--python/pyarmnn/test/test_generated.py4
1 files changed, 3 insertions, 1 deletions
diff --git a/python/pyarmnn/test/test_generated.py b/python/pyarmnn/test/test_generated.py
index 24765c73ab..0e1d663234 100644
--- a/python/pyarmnn/test/test_generated.py
+++ b/python/pyarmnn/test/test_generated.py
@@ -1,4 +1,4 @@
-# Copyright © 2020 Arm Ltd. All rights reserved.
+# Copyright © 2020 Arm Ltd and Contributors. All rights reserved.
# SPDX-License-Identifier: MIT
import inspect
from typing import Tuple
@@ -6,6 +6,7 @@ from typing import Tuple
import pytest
import pyarmnn._generated.pyarmnn as generated_armnn
+import pyarmnn._generated.pyarmnn as generated_deserializer
import pyarmnn._generated.pyarmnn_caffeparser as generated_caffe
import pyarmnn._generated.pyarmnn_onnxparser as generated_onnx
import pyarmnn._generated.pyarmnn_tfliteparser as generated_tflite
@@ -26,6 +27,7 @@ def get_classes(swig_independent_classes: Tuple):
ignored_class_names = (*swig_independent_classes, '_SwigNonDynamicMeta')
return list(filter(lambda x: x[0] not in ignored_class_names,
inspect.getmembers(generated_armnn, inspect.isclass) +
+ inspect.getmembers(generated_deserializer, inspect.isclass) +
inspect.getmembers(generated_caffe, inspect.isclass) +
inspect.getmembers(generated_tflite, inspect.isclass) +
inspect.getmembers(generated_onnx, inspect.isclass) +