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

  itk_python_add_test(
    NAME
    itkGradientVectorFlowImageFilterPythonTest
    TEST_DRIVER_ARGS
    --compare
    ${ITK_TEST_OUTPUT_DIR}/itkGradientVectorFlowImageFilterPythonTestX.png
    DATA{${test_baseline_dir}/itkGradientVectorFlowImageFilterPythonTestX.png}
    --compare
    ${ITK_TEST_OUTPUT_DIR}/itkGradientVectorFlowImageFilterPythonTestY.png
    DATA{${test_baseline_dir}/itkGradientVectorFlowImageFilterPythonTestY.png}
    COMMAND
    ${CMAKE_CURRENT_SOURCE_DIR}/itkGradientVectorFlowImageFilterTest.py
    DATA{${test_input_dir}/Contours.png}
    ${ITK_TEST_OUTPUT_DIR}/itkGradientVectorFlowImageFilterPythonTestX.png
    ${ITK_TEST_OUTPUT_DIR}/itkGradientVectorFlowImageFilterPythonTestY.png
    50
    50000.0
    0.001)

  list(
    FIND
    ITK_WRAP_IMAGE_DIMS
    2
    wrap_2_index)
  if(ITK_WRAP_float AND wrap_2_index GREATER -1)
    itk_python_add_test(
      NAME
      PythonLaplacianImageFilterTest
      COMMAND
      ${CMAKE_CURRENT_SOURCE_DIR}/LaplacianImageFilterTest.py
      DATA{${ITK_DATA_ROOT}/Input/cthead1.png}
      ${ITK_TEST_OUTPUT_DIR}/LaplacianImageFilterTest.png)
    itk_python_add_test(
      NAME
      PythonCannyEdgeDetectionImageFilterTest
      TEST_DRIVER_ARGS
      --compareNumberOfPixelsTolerance
      1
      --compare
      ${ITK_TEST_OUTPUT_DIR}/PythonCannyEdgeDetectionImageFilterTest.png
      DATA{Baseline/PythonCannyEdgeDetectionImageFilterTest.png}
      COMMAND
      ${CMAKE_CURRENT_SOURCE_DIR}/CannyEdgeDetectionImageFilterTest.py
      DATA{${ITK_DATA_ROOT}/Input/cthead1.png}
      ${ITK_TEST_OUTPUT_DIR}/PythonCannyEdgeDetectionImageFilterTest.png
      9)
    itk_python_add_test(
      NAME
      PythonHoughTransform2DLinesImageFilterTest
      COMMAND
      ${CMAKE_CURRENT_SOURCE_DIR}/HoughTransform2DLinesImageFilterTest.py
      DATA{Baseline/itkHoughTransform2DLinesImageFilterTest.png})

  endif()
endif()
