Merge pull request #29243 from mrodozov/mxnet-py-unittest

Add test for mxnet python interface
cms-sw · Mar 24, 2020 · 62c289f · 62c289f
2 parents 7047313 + 0f39e07
commit 62c289f
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/PhysicsTools/MXNet/test/BuildFile.xml b/PhysicsTools/MXNet/test/BuildFile.xml
@@ -1,7 +1,9 @@
+<test name="test_py_mxnet" command="test_mxnet.py">
+  <use name="mxnet-predict"/>
+</test>
 <bin name="testMXNetCppPredictor" file="testRunner.cpp, testMXNetCppPredictor.cc">
     <use name="boost_filesystem" />
     <use name="cppunit" />
-
     <use name="PhysicsTools/MXNet" />
     <use name="FWCore/ParameterSet" />
     <use name="FWCore/Utilities" />

diff --git a/PhysicsTools/MXNet/test/test_mxnet.py b/PhysicsTools/MXNet/test/test_mxnet.py
@@ -0,0 +1,5 @@
+#!/usr/bin/env python
+import mxnet as mx
+a = mx.nd.ones((2, 3))
+b = a * 2 + 1
+b.asnumpy()