4.8 并行测试

NOTE:此示例代码可以在 https://github.com/dev-cafe/cmake-cookbook/tree/v1.0/chapter-04/recipe-08 中找到。该示例在CMake 3.5版(或更高版本)中是有效的,并且已经在GNU/Linux、macOS和Windows上进行过测试。

大多数现代计算机都有4个或更多个CPU核芯。CTest有个非常棒的特性,能够并行运行测试,如果您有多个可用的核。这可以减少测试的总时间,而减少总测试时间才是真正重要的,从而开发人员频繁地进行测试。本示例中,我们将演示这个特性,并讨论如何优化测试以获得最大的性能。

其他测试可以进行相应地表示,我们把这些测试脚本放在CMakeLists.txt同目录下面的test目录中。

准备工作

我们假设测试集包含标记为a, b,…,j的测试用例,每一个都有特定的持续时间:

测试用例 该单元的耗时
a, b, c, d 0.5
e, f, g 1.5
h 2.5
i 3.5
j 4.5

时间单位可以是分钟,但是为了保持简单和简短,我们将使用秒。为简单起见,我们可以用Python脚本表示test a,它消耗0.5个时间单位:

  1. import sys
  2. import time
  3. # wait for 0.5 seconds
  4. time.sleep(0.5)
  5. # finally report success
  6. sys.exit(0)

其他测试同理。我们将把这些脚本放在CMakeLists.txt下面,一个名为test的目录中。

具体实施

对于这个示例,我们需要声明一个测试列表,如下:

  1. CMakeLists.txt非常简单:

    1. # set minimum cmake version
    2. cmake_minimum_required(VERSION 3.5 FATAL_ERROR)
    3. # project name
    4. project(recipe-08 LANGUAGES NONE)
    5. # detect python
    6. find_package(PythonInterp REQUIRED)
    7. # define tests
    8. enable_testing()
    9. add_test(a ${PYTHON_EXECUTABLE} ${CMAKE_CURRENT_SOURCE_DIR}/test/a.py)
    10. add_test(b ${PYTHON_EXECUTABLE} ${CMAKE_CURRENT_SOURCE_DIR}/test/b.py)
    11. add_test(c ${PYTHON_EXECUTABLE} ${CMAKE_CURRENT_SOURCE_DIR}/test/c.py)
    12. add_test(d ${PYTHON_EXECUTABLE} ${CMAKE_CURRENT_SOURCE_DIR}/test/d.py)
    13. add_test(e ${PYTHON_EXECUTABLE} ${CMAKE_CURRENT_SOURCE_DIR}/test/e.py)
    14. add_test(f ${PYTHON_EXECUTABLE} ${CMAKE_CURRENT_SOURCE_DIR}/test/f.py)
    15. add_test(g ${PYTHON_EXECUTABLE} ${CMAKE_CURRENT_SOURCE_DIR}/test/g.py)
    16. add_test(h ${PYTHON_EXECUTABLE} ${CMAKE_CURRENT_SOURCE_DIR}/test/h.py)
    17. add_test(i ${PYTHON_EXECUTABLE} ${CMAKE_CURRENT_SOURCE_DIR}/test/i.py)
    18. add_test(j ${PYTHON_EXECUTABLE} ${CMAKE_CURRENT_SOURCE_DIR}/test/j.py)
  2. 我们可以配置项目,使用ctest运行测试,总共需要17秒:

    1. $ mkdir -p build
    2. $ cd build
    3. $ cmake ..
    4. $ ctest
    5. Start 1: a
    6. 1/10 Test #1: a ................................ Passed 0.51 sec
    7. Start 2: b
    8. 2/10 Test #2: b ................................ Passed 0.51 sec
    9. Start 3: c
    10. 3/10 Test #3: c ................................ Passed 0.51 sec
    11. Start 4: d
    12. 4/10 Test #4: d ................................ Passed 0.51 sec
    13. Start 5: e
    14. 5/10 Test #5: e ................................ Passed 1.51 sec
    15. Start 6: f
    16. 6/10 Test #6: f ................................ Passed 1.51 sec
    17. Start 7: g
    18. 7/10 Test #7: g ................................ Passed 1.51 sec
    19. Start 8: h
    20. 8/10 Test #8: h ................................ Passed 2.51 sec
    21. Start 9: i
    22. 9/10 Test #9: i ................................ Passed 3.51 sec
    23. Start 10: j
    24. 10/10 Test #10: j ................................ Passed 4.51 sec
    25. 100% tests passed, 0 tests failed out of 10
    26. Total Test time (real) = 17.11 sec
  3. 现在,如果机器有4个内核可用,我们可以在不到5秒的时间内在4个内核上运行测试集:

    1. $ ctest --parallel 4
    2. Start 10: j
    3. Start 9: i
    4. Start 8: h
    5. Start 5: e
    6. 1/10 Test #5: e ................................ Passed 1.51 sec
    7. Start 7: g
    8. 2/10 Test #8: h ................................ Passed 2.51 sec
    9. Start 6: f
    10. 3/10 Test #7: g ................................ Passed 1.51 sec
    11. Start 3: c
    12. 4/10 Test #9: i ................................ Passed 3.63 sec
    13. 5/10 Test #3: c ................................ Passed 0.60 sec
    14. Start 2: b
    15. Start 4: d
    16. 6/10 Test #6: f ................................ Passed 1.51 sec
    17. 7/10 Test #4: d ................................ Passed 0.59 sec
    18. 8/10 Test #2: b ................................ Passed 0.59 sec
    19. Start 1: a
    20. 9/10 Test #10: j ................................ Passed 4.51 sec
    21. 10/10 Test #1: a ................................ Passed 0.51 sec
    22. 100% tests passed, 0 tests failed out of 10
    23. Total Test time (real) = 4.74 sec

工作原理

可以观察到,在并行情况下,测试j、i、h和e同时开始。当并行运行时,总测试时间会有显著的减少。观察ctest --parallel 4的输出,我们可以看到并行测试运行从最长的测试开始,最后运行最短的测试。从最长的测试开始是一个非常好的策略。这就像打包移动的盒子:从较大的项目开始,然后用较小的项目填补空白。a-j测试在4个核上的叠加比较,从最长的开始,如下图所示:

  1. --> time
  2. core 1: jjjjjjjjj
  3. core 2: iiiiiiibd
  4. core 3: hhhhhggg
  5. core 4: eeefffac

按照定义测试的顺序运行,运行结果如下:

  1. --> time
  2. core 1: aeeeiiiiiii
  3. core 2: bfffjjjjjjjjj
  4. core 3: cggg
  5. core 4: dhhhhh

按照定义测试的顺序运行测试,总的来说需要更多的时间,因为这会让2个核大部分时间处于空闲状态(这里的核3和核4)。CMake知道每个测试的时间成本,是因为我们先顺序运行了测试,将每个测试的成本数据记录在test/Temporary/CTestCostData.txt文件中:

  1. a 1 0.506776
  2. b 1 0.507882
  3. c 1 0.508175
  4. d 1 0.504618
  5. e 1 1.51006
  6. f 1 1.50975
  7. g 1 1.50648
  8. h 1 2.51032
  9. i 1 3.50475
  10. j 1 4.51111

如果在配置项目之后立即开始并行测试,它将按照定义测试的顺序运行测试,在4个核上的总测试时间明显会更长。这意味着什么呢?这意味着,我们应该减少的时间成本来安排测试?这是一种决策,但事实证明还有另一种方法,我们可以自己表示每次测试的时间成本:

  1. add_test(a ${PYTHON_EXECUTABLE} ${CMAKE_CURRENT_SOURCE_DIR}/test/a.py)
  2. add_test(b ${PYTHON_EXECUTABLE} ${CMAKE_CURRENT_SOURCE_DIR}/test/b.py)
  3. add_test(c ${PYTHON_EXECUTABLE} ${CMAKE_CURRENT_SOURCE_DIR}/test/c.py)
  4. add_test(d ${PYTHON_EXECUTABLE} ${CMAKE_CURRENT_SOURCE_DIR}/test/d.py)
  5. set_tests_properties(a b c d PROPERTIES COST 0.5)
  6. add_test(e ${PYTHON_EXECUTABLE} ${CMAKE_CURRENT_SOURCE_DIR}/test/e.py)
  7. add_test(f ${PYTHON_EXECUTABLE} ${CMAKE_CURRENT_SOURCE_DIR}/test/f.py)
  8. add_test(g ${PYTHON_EXECUTABLE} ${CMAKE_CURRENT_SOURCE_DIR}/test/g.py)
  9. set_tests_properties(e f g PROPERTIES COST 1.5)
  10. add_test(h ${PYTHON_EXECUTABLE} ${CMAKE_CURRENT_SOURCE_DIR}/test/h.py)
  11. set_tests_properties(h PROPERTIES COST 2.5)
  12. add_test(i ${PYTHON_EXECUTABLE} ${CMAKE_CURRENT_SOURCE_DIR}/test/i.py)
  13. set_tests_properties(i PROPERTIES COST 3.5)
  14. add_test(j ${PYTHON_EXECUTABLE} ${CMAKE_CURRENT_SOURCE_DIR}/test/j.py)
  15. set_tests_properties(j PROPERTIES COST 4.5)

成本参数可以是一个估计值,也可以从test/Temporary/CTestCostData.txt中提取。

更多信息

除了使用ctest --parallel N,还可以使用环境变量CTEST_PARALLEL_LEVEL将其设置为所需的级别。