57 lines
1.8 KiB
CMake
57 lines
1.8 KiB
CMake
enable_testing()
|
|
|
|
# 查找依赖包
|
|
find_package(GTest REQUIRED)
|
|
find_package(OpenCV REQUIRED)
|
|
find_package(yaml-cpp REQUIRED)
|
|
find_package(CUDA REQUIRED)
|
|
|
|
# 设置测试配置文件路径
|
|
set(TEST_CONFIG_PATH ${CMAKE_SOURCE_DIR}/pipeline/configs/pipeline.yaml)
|
|
|
|
function(add_test_target target_name source_file)
|
|
add_executable(${target_name} ${source_file})
|
|
target_link_libraries(${target_name}
|
|
PRIVATE
|
|
pipeline
|
|
GTest::gtest_main
|
|
${OpenCV_LIBS}
|
|
${CUDA_LIBRARIES}
|
|
${NVINFER_LIB}
|
|
${NVONNXPARSER_LIB}
|
|
${NVINFER_PLUGIN_LIB}
|
|
avcodec
|
|
avformat
|
|
avutil
|
|
swscale
|
|
pthread
|
|
)
|
|
target_include_directories(${target_name}
|
|
PRIVATE
|
|
${CMAKE_SOURCE_DIR}
|
|
${TENSORRT_INCLUDE_DIRS}
|
|
)
|
|
target_compile_definitions(${target_name}
|
|
PRIVATE
|
|
TEST_CONFIG_PATH="${TEST_CONFIG_PATH}"
|
|
)
|
|
add_test(NAME ${target_name}
|
|
COMMAND ${target_name}
|
|
WORKING_DIRECTORY ${CMAKE_SOURCE_DIR})
|
|
endfunction()
|
|
|
|
# 添加测试可执行文件
|
|
add_test_target(cuda_helper_test test_cuda_helper.cpp)
|
|
add_test_target(frame_queue_test test_frame_queue.cpp)
|
|
add_test_target(config_parser_test test_yaml_config.cpp)
|
|
add_test_target(video_reader_test test_video_reader.cpp)
|
|
add_test_target(rtsp_reader_test test_rtsp_reader.cpp)
|
|
add_test_target(video_writer_test test_video_writer.cpp)
|
|
add_test_target(rtsp_writer_test test_rtsp_writer.cpp)
|
|
add_test_target(input_manager_test test_input_manager.cpp)
|
|
add_test_target(output_manager_test test_output_manager.cpp)
|
|
add_test_target(trt_inference_test test_trt_inference.cpp)
|
|
add_test_target(renderer_test test_renderer.cpp)
|
|
add_test_target(frame_drawer_test test_frame_drawer.cpp)
|
|
add_test_target(pipeline_test test_pipeline.cpp)
|
|
|