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

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 PythonCurvatureAnisotropicDiffusionImageFilterTest
    COMMAND
      ${CMAKE_CURRENT_SOURCE_DIR}/CurvatureAnisotropicDiffusionImageFilterTest.py
      DATA{${ITK_DATA_ROOT}/Input/cthead1.png}
      ${ITK_TEST_OUTPUT_DIR}/CurvatureAnisotropicDiffusionImageFilterTest.png
      5
      0.125
      3
  )
  itk_python_add_test(
    NAME PythonSmoothingRecursiveGaussianImageFilterTest
    COMMAND
      ${CMAKE_CURRENT_SOURCE_DIR}/SmoothingRecursiveGaussianImageFilterTest.py
      DATA{${ITK_DATA_ROOT}/Input/cthead1.png}
      ${ITK_TEST_OUTPUT_DIR}/SmoothingRecursiveGaussianImageFilterTest.png
      3
  )
  itk_python_add_test(
    NAME PythonGradientAnisotropicDiffusionImageFilterTest
    TEST_DRIVER_ARGS
      --compare
      ${ITK_TEST_OUTPUT_DIR}/PythonGradientAnisotropicDiffusionImageFilterTest.png
      DATA{Baseline/PythonGradientAnisotropicDiffusionImageFilterTest.png}
    COMMAND
      ${CMAKE_CURRENT_SOURCE_DIR}/GradientAnisotropicDiffusionImageFilterTest.py
      DATA{${test_input_dir}/BrainProtonDensitySlice.png}
      ${ITK_TEST_OUTPUT_DIR}/PythonGradientAnisotropicDiffusionImageFilterTest.png
      5
      0.125
      1.0
  )
endif()
