set(test_input_dir ${itk-module_SOURCE_DIR}/test/Input)
set(test_baseline_dir ${itk-module_SOURCE_DIR}/test/Baseline)

list(
  FIND
  ITK_WRAP_IMAGE_DIMS
  2
  wrap_2_index)
if(ITK_WRAP_PYTHON
   AND ITK_WRAP_float
   AND wrap_2_index GREATER -1)
  itk_python_add_test(
    NAME
    PythonFastMarchingLeftVentricleTest
    TEST_DRIVER_ARGS
    --compare
    ${ITK_TEST_OUTPUT_DIR}/FastMarchingLeftVentricleTest.png
    DATA{${test_baseline_dir}/FastMarchingLeftVentricleTest.png}
    COMMAND
    ${CMAKE_CURRENT_SOURCE_DIR}/FastMarchingImageFilterTest.py
    DATA{${test_input_dir}/BrainProtonDensitySlice.png}
    ${ITK_TEST_OUTPUT_DIR}/FastMarchingLeftVentricleTest.png
    81
    114
    1.0
    -0.5
    3.0
    100
    100)

  itk_python_add_test(
    NAME
    PythonFastMarchingRightVentricleTest
    TEST_DRIVER_ARGS
    --compare
    ${ITK_TEST_OUTPUT_DIR}/FastMarchingRightVentricleTest.png
    DATA{${test_baseline_dir}/FastMarchingRightVentricleTest.png}
    COMMAND
    ${CMAKE_CURRENT_SOURCE_DIR}/FastMarchingImageFilterTest.py
    DATA{${test_input_dir}/BrainProtonDensitySlice.png}
    ${ITK_TEST_OUTPUT_DIR}/FastMarchingRightVentricleTest.png
    99
    114
    1.0
    -0.5
    3.0
    100
    100)

  itk_python_add_test(
    NAME
    PythonFastMarchingWhiteMatterTest
    TEST_DRIVER_ARGS
    --compare
    ${ITK_TEST_OUTPUT_DIR}/FastMarchingWhiteMatterTest.png
    DATA{${test_baseline_dir}/FastMarchingWhiteMatterTest.png}
    COMMAND
    ${CMAKE_CURRENT_SOURCE_DIR}/FastMarchingImageFilterTest.py
    DATA{${test_input_dir}/BrainProtonDensitySlice.png}
    ${ITK_TEST_OUTPUT_DIR}/FastMarchingWhiteMatterTest.png
    56
    92
    1.0
    -0.3
    2.0
    200
    100)

  itk_python_add_test(
    NAME
    PythonFastMarchingGrayMatterTest
    TEST_DRIVER_ARGS
    --compare
    ${ITK_TEST_OUTPUT_DIR}/FastMarchingGrayMatterTest.png
    DATA{${test_baseline_dir}/FastMarchingGrayMatterTest.png}
    COMMAND
    ${CMAKE_CURRENT_SOURCE_DIR}/FastMarchingImageFilterTest.py
    DATA{${test_input_dir}/BrainProtonDensitySlice.png}
    ${ITK_TEST_OUTPUT_DIR}/FastMarchingGrayMatterTest.png
    40
    90
    .5
    -0.3
    2.0
    200
    100)
endif()
