ONNX import: Add missing tests

apache · Dec 14, 2018 · e4083db · e4083db
1 parent fb15d32
commit e4083db
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 8 deletions.
diff --git a/tests/python-pytest/onnx/import/mxnet_backend_test.py b/tests/python-pytest/onnx/import/mxnet_backend_test.py
@@ -47,6 +47,8 @@
 
 BACKEND_TESTS.exclude('.*broadcast.*')
 BACKEND_TESTS.exclude('.*bcast.*')
+BACKEND_TESTS.exclude('.*logsoftmax.*')
+BACKEND_TESTS.exclude('.*expand.*')
 
 # import all test cases at global scope to make them visible to python.unittest
 globals().update(BACKEND_TESTS.enable_report().test_cases)

diff --git a/tests/python-pytest/onnx/import/test_cases.py b/tests/python-pytest/onnx/import/test_cases.py
@@ -19,7 +19,6 @@
 
 IMPLEMENTED_OPERATORS_TEST = [
     'test_split_equal'
-    'test_random_',
     'test_add',
     'test_sub',
     'test_mul',
@@ -37,7 +36,7 @@
     'test_constant_pad',
     'test_edge_pad',
     'test_reflect_pad',
-    'test_squeeze_',
+    'test_squeeze',
     'test_unsqueeze',
     'test_softmax_example',
     'test_softmax_large_number',
@@ -81,26 +80,27 @@
     'test_cast',
     'test_instancenorm',
     #pytorch operator tests
-    'test_operator_exp',
+    'test_exp',
     'test_operator_maxpool',
     'test_operator_params',
     'test_operator_permute2',
     'test_depthtospace',
-    'test_size'
+    'test_size',
+    'test_log',
+    'test_gather',
+    'test_dropout',
+    'test_leakyrelu',
+    'test_elu'
     ]
 
 BASIC_MODEL_TESTS = [
     'test_AvgPool2D',
     'test_BatchNorm',
     'test_ConstantPad2d'
     'test_Conv2d',
-    'test_ELU',
-    'test_LeakyReLU',
     'test_MaxPool',
     'test_PReLU',
-    'test_ReLU',
     'test_selu_default',
-    'test_Sigmoid',
     'test_Softmax',
     'test_softmax_functional',
     'test_softmax_lastdim',