mirror of
https://github.com/Kitware/CMake.git
synced 2025-10-14 19:08:07 +08:00
38 lines
1.1 KiB
CMake
38 lines
1.1 KiB
CMake
enable_language(C)
|
|
|
|
include(CTest)
|
|
|
|
cmake_policy(GET CMP0201 numpy_policy)
|
|
|
|
if(numpy_policy STREQUAL NEW)
|
|
set(components "Development.Module")
|
|
endif()
|
|
|
|
if(CMake_TEST_FindPython2_NumPy)
|
|
|
|
find_package (Python2 REQUIRED COMPONENTS ${components} NumPy)
|
|
|
|
Python2_add_library (arraytest2 MODULE arraytest.c)
|
|
target_compile_definitions (arraytest2 PRIVATE PYTHON2)
|
|
target_link_libraries (arraytest2 PRIVATE Python2::NumPy)
|
|
|
|
add_test (NAME python2_arraytest
|
|
COMMAND "${CMAKE_COMMAND}" -E env "PYTHONPATH=$<TARGET_FILE_DIR:arraytest2>"
|
|
"${Python2_INTERPRETER}" -c "import numpy; import arraytest2; arraytest2.vecsq(numpy.array([1, 2, 3]));")
|
|
|
|
endif()
|
|
|
|
if(CMake_TEST_FindPython3_NumPy)
|
|
|
|
find_package (Python3 REQUIRED COMPONENTS ${components} NumPy)
|
|
|
|
Python3_add_library (arraytest3 MODULE arraytest.c)
|
|
target_compile_definitions (arraytest3 PRIVATE PYTHON3)
|
|
target_link_libraries (arraytest3 PRIVATE Python3::NumPy)
|
|
|
|
add_test (NAME python3_arraytest
|
|
COMMAND "${CMAKE_COMMAND}" -E env "PYTHONPATH=$<TARGET_FILE_DIR:arraytest3>"
|
|
"${Python3_INTERPRETER}" -c "import numpy; import arraytest3; arraytest3.vecsq(numpy.array([1, 2, 3]));")
|
|
|
|
endif()
|