Add python test

src/py/CMakeLists.txt

 
 find_package(pybind11 REQUIRED)
 pybind11_add_module(migraphx_py migraphx_py.cpp)
-target_link_libraries(migraphx_py migraphx migraphx_onnx migraphx_gpu migraphx_cpu)
+target_link_libraries(migraphx_py migraphx migraphx_onnx migraphx_cpu)
+set_target_properties(migraphx_py PROPERTIES 
+    OUTPUT_NAME migraphx
+)

src/py/migraphx_py.cpp

@@ -2,6 +2,8 @@
 #include <pybind11/pybind11.h>
 #include <pybind11/stl.h>
 #include <migraphx/program.hpp>
+#include <migraphx/generate.hpp>
+#include <migraphx/cpu/target.hpp>
 #include <migraphx/onnx.hpp>
 
 namespace py = pybind11;
@@ -71,6 +73,14 @@ PYBIND11_MODULE(migraphx, m)
 
     m.def("parse_onnx", &migraphx::parse_onnx);
 
+    m.def("target", [](const std::string& name) -> migraphx::target {
+        if (name == "cpu")
+            return migraphx::cpu::target{};
+        throw std::runtime_error("Target not found: " + name);
+    });
+
+    m.def("generate_argument", &migraphx::generate_argument, py::arg("s"), py::arg("seed") = 0);
+
 #ifdef VERSION_INFO
     m.attr("__version__") = VERSION_INFO;
 #else

test/CMakeLists.txt

@@ -120,6 +120,8 @@ add_test(NAME test_onnx COMMAND $<TARGET_FILE:test_onnx> WORKING_DIRECTORY ${CMA
 add_dependencies(tests test_onnx)
 add_dependencies(check test_onnx)
 
+add_subdirectory(py)
+
 function(test_header NAME HEADER)
 
     file(WRITE ${CMAKE_CURRENT_BINARY_DIR}/header-main-include-${NAME}.cpp 

test/py/CMakeLists.txt

+
+find_package(PythonInterp)
+
+function(add_py_test NAME SCRIPT)
+    set (ENV_COMMAND ${CMAKE_COMMAND} -E env 
+        "PYTHONPATH=$<TARGET_FILE_DIR:migraphx_py>"
+    )
+    add_test(
+        NAME test_py_${NAME} 
+        COMMAND ${ENV_COMMAND} ${PYTHON_EXECUTABLE} ${CMAKE_CURRENT_SOURCE_DIR}/${SCRIPT} ${ARGN})
+    add_custom_target(test_py_${NAME} 
+        COMMAND ${ENV_COMMAND} ${PYTHON_EXECUTABLE} ${CMAKE_CURRENT_SOURCE_DIR}/${SCRIPT} ${ARGN} 
+        COMMENT "${PYTHON_EXECUTABLE} ${SCRIPT}")
+endfunction()
+
+add_py_test(test test.py)

test/py/test.py

+import migraphx