# test_jax skips itself cleanly when JAX is not installed.
foreach(name basic numpy dlpack jax)
    add_test(NAME python_test_${name}
        COMMAND ${Python_EXECUTABLE} ${CMAKE_CURRENT_SOURCE_DIR}/test_${name}.py
    )
    set_tests_properties(python_test_${name} PROPERTIES
        ENVIRONMENT "PYTHONPATH=${PYTHON_OUTPUT_DIR}"
    )
endforeach()
