10 lines
482 B
CMake
10 lines
482 B
CMake
if(ENABLE_TESTING)
|
|
set(TEST_MAIN "unit_tests")
|
|
include_directories(${BayesNet_SOURCE_DIR}/src/BayesNet)
|
|
include_directories(${BayesNet_SOURCE_DIR}/src/Platform)
|
|
set(TEST_SOURCES BayesModels.cc BayesNetwork.cc ${BayesNet_SOURCES} ${Platform_SOURCES})
|
|
add_executable(${TEST_MAIN} ${TEST_SOURCES})
|
|
target_link_libraries(${TEST_MAIN} PUBLIC "${TORCH_LIBRARIES}" Catch2::Catch2WithMain)
|
|
add_test(NAME ${TEST_MAIN} COMMAND ${TEST_MAIN})
|
|
endif(ENABLE_TESTING)
|