if(ITK_WRAP_PYTHON)
  itk_python_add_test(
    NAME
    itkGPUNeighborhoodOperatorImageFilterPythonTest2D
    TEST_DRIVER_ARGS
    COMMAND
    ${CMAKE_CURRENT_SOURCE_DIR}/itkGPUNeighborhoodOperatorImageFilterTest.py
    DATA{../../test/Input/BrainProtonDensitySlice.png}
    ${ITK_TEST_OUTPUT_DIR}/gpuNeighborhoodOperatorImageFilterTest2DPython.mha
    2)

  itk_python_add_test(
    NAME
    itkGPUNeighborhoodOperatorImageFilterPythonTest3D
    TEST_DRIVER_ARGS
    COMMAND
    ${CMAKE_CURRENT_SOURCE_DIR}/itkGPUNeighborhoodOperatorImageFilterTest.py
    DATA{../../test/Input/HeadMRVolume.mha}
    ${ITK_TEST_OUTPUT_DIR}/gpuNeighborhoodOperatorImageFilterTest3DPython.mha
    3)
endif()
