include(UsePythonTest)

# Value of PYTHONPATH variable for in-build-area testing.
set(TEST_PATH "${PyOpenMM_INSTALL_TEST_DIR}/lib/python")
file(TO_NATIVE_PATH TEST_PATH ${TEST_PATH})
set(TEST_PYTHON_PATH ${TEST_PATH} $ENV{PYTHONPATH})
# message("TEST_PYTHON_PATH = " ${TEST_PYTHON_PATH})

if(WIN32)
    # set(TEST_PYTHON_PATH "${TEST_PYTHON_PATH}/Lib/site-packages")
endif(WIN32)

# Run all the testWhatever.py programs
file(GLOB TEST_PROGS "test*.py")
foreach(TEST_PROG ${TEST_PROGS})
    get_filename_component(TEST_ROOT ${TEST_PROG} NAME_WE)
    set_source_files_properties(${TEST_PROG} PROPERTIES PYTHONPATH "${TEST_PYTHON_PATH}")
    ADD_PYTHON_TEST("${TEST_ROOT}" ${TEST_PROG})
endforeach(TEST_PROG ${TEST_PROGS})

# Run doctests for particular modules
set(TEST_PROG "${CMAKE_SOURCE_DIR}/simtk/unit/doctests.py")
set_source_files_properties(${TEST_PROG} PROPERTIES PYTHONPATH "${TEST_PYTHON_PATH}")
#ADD_PYTHON_TEST(test_units "${TEST_PROG}")

### Only run tests requiring numpy if numpy is installed ###

execute_process(
    COMMAND "${PYTHON_EXECUTABLE}" numpy_required.py
    WORKING_DIRECTORY "${CMAKE_CURRENT_SOURCE_DIR}"
    RESULT_VARIABLE PYOPENMM_LACK_NUMPY
)
if(NOT PYOPENMM_LACK_NUMPY)
    add_subdirectory(test_using_numpy)
endif()

execute_process(
    COMMAND "${PYTHON_EXECUTABLE}" scipy_required.py
    WORKING_DIRECTORY "${CMAKE_CURRENT_SOURCE_DIR}"
    RESULT_VARIABLE PYOPENMM_LACK_SCIPY
)
if(NOT PYOPENMM_LACK_SCIPY)
    add_subdirectory(test_using_scipy)
endif()