summaryrefslogtreecommitdiffstats
path: root/Tests/RunCMake/FindPython/NumPyOnly.cmake
blob: 30e55e9a5cc9d2e326bd214ac3d3d3c5f722270e (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
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()