ENH: Added WriteAnImage Python script

Author: kian-weimer

src/IO/ImageBase/WriteAnImage/CMakeLists.txt

@@ -21,3 +21,8 @@ install( FILES Code.cxx CMakeLists.txt
 enable_testing()
 add_test( NAME WriteAnImageTest
   COMMAND ${CMAKE_RUNTIME_OUTPUT_DIRECTORY}/WriteAnImage )
+
+if(ITK_WRAP_PYTHON)
+  add_test(NAME WriteAnImageTestPython
+    COMMAND ${PYTHON_EXECUTABLE} ${CMAKE_CURRENT_SOURCE_DIR}/Code.py)
+endif()

src/IO/ImageBase/WriteAnImage/Code.py

@@ -0,0 +1,29 @@
+#!/usr/bin/env python
+
+import sys
+import itk
+
+if len(sys.argv) > 1:
+    output_filename = sys.argv[1]
+else:
+    output_filename = "testPython.png"
+
+pixel_type = itk.UC
+dimension = 2
+image_type = itk.Image[pixel_type, dimension]
+
+start = itk.Index[dimension]()
+start.Fill(0)
+
+size = itk.Size[dimension]()
+size[0] = 200
+size[1] = 300
+
+region = itk.ImageRegion[dimension]()
+region.SetIndex(start)
+region.SetSize(size)
+
+image = image_type.New(Regions=region)
+image.Allocate()
+
+itk.imwrite(image, output_filename)

src/IO/ImageBase/WriteAnImage/Documentation.rst

@@ -26,6 +26,13 @@ Results
 Code
 ----
 
+Python
+......
+
+.. literalinclude:: Code.py
+   :language: python
+   :lines: 1, 16-
+
 C++
 ...