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

  itk_python_add_test(
    NAME
    PythonLazyLoadingImage
    COMMAND
    LazyLoadingImageTest.py)

  list(
    FIND
    ITK_WRAP_IMAGE_DIMS
    2
    wrap_2_index)
  if(ITK_WRAP_float AND wrap_2_index GREATER -1)
    # TODO: reenable the geodesic test once we get why the result is not the
    # same than with c++
    # keep just one to be sure it run, but don't compare the images

    itk_python_add_test(
      NAME
      PythonGeodesicActiveContourLeftVentricleTest
      #     --compare GeodesicActiveContourLeftVentricleTest.png
      #             DATA{${WrapITK_SOURCE_DIR}/images/GeodesicActiveContourLeftVentricleTest.png}
      COMMAND
      GeodesicActiveContourImageFilterTest.py
      DATA{${test_input_dir}/BrainProtonDensitySlice.png}
      ${ITK_TEST_OUTPUT_DIR}/GeodesicActiveContourLeftVentricleTest.png
      81
      114
      5.0
      1.0
      -0.5
      3.0
      2.0)

    itk_python_add_test(
      NAME
      PythonThresholdSegmentationLevelSetVentricleTest
      TEST_DRIVER_ARGS
      --compare
      ${ITK_TEST_OUTPUT_DIR}/PythonThresholdSegmentationLevelSetVentricleTest.png
      DATA{Baseline/PythonThresholdSegmentationLevelSetVentricleTest.png}
      COMMAND
      ThresholdSegmentationLevelSetImageFilterTest.py
      DATA{${test_input_dir}/BrainProtonDensitySlice.png}
      ${ITK_TEST_OUTPUT_DIR}/PythonThresholdSegmentationLevelSetVentricleTest.png
      81
      112
      5
      210
      250)

    itk_python_add_test(
      NAME
      PythonThresholdSegmentationLevelSetWhiteMatterTest
      TEST_DRIVER_ARGS
      --compare
      ${ITK_TEST_OUTPUT_DIR}/ThresholdSegmentationLevelSetWhiteMatterTest.png
      DATA{Baseline/PythonThresholdSegmentationLevelSetWhiteMatterTest.png}
      COMMAND
      ThresholdSegmentationLevelSetImageFilterTest.py
      DATA{${test_input_dir}/BrainProtonDensitySlice.png}
      ${ITK_TEST_OUTPUT_DIR}/ThresholdSegmentationLevelSetWhiteMatterTest.png
      60
      116
      5
      150
      180)

    itk_python_add_test(
      NAME
      PythonThresholdSegmentationLevelSetGrayMatterTest
      TEST_DRIVER_ARGS
      --compare
      ${ITK_TEST_OUTPUT_DIR}/PythonThresholdSegmentationLevelSetGrayMatterTest.png
      DATA{Baseline/PythonThresholdSegmentationLevelSetGrayMatterTest.png}
      COMMAND
      ThresholdSegmentationLevelSetImageFilterTest.py
      DATA{${WrapITK_SOURCE_DIR}/images/BrainProtonDensitySlice.png}
      ${ITK_TEST_OUTPUT_DIR}/PythonThresholdSegmentationLevelSetGrayMatterTest.png
      107
      69
      5
      180
      210)

    itk_python_expression_add_test(
      NAME
      itkBinaryMaskToNarrowBandPointSetFilterPythonTest
      EXPRESSION
      "filt = itk.BinaryMaskToNarrowBandPointSetFilter.New()")
  endif()
endif()