ENH: add additional assertions for ttype parameter

InsightSoftwareConsortium · Dec 9, 2020 · aac01f1 · aac01f1
1 parent 25a71e3
commit aac01f1
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/Wrapping/Generators/Python/Tests/extras.py b/Wrapping/Generators/Python/Tests/extras.py
@@ -344,6 +344,8 @@ def custom_callback(name, progress):
 numpyImage = np.random.randint(0, 256, (8, 12, 5)).astype(np.uint8)
 image = itk.image_from_array(numpyImage, is_vector=False)
 assert type(image) == type(itk.image_from_array(numpyImage, ttype=(type(image),)))
+assert type(image) == type(itk.image_from_array(numpyImage, ttype=[type(image)]))
+assert type(image) == type(itk.image_from_array(numpyImage, ttype=type(image)))
 cast = image.astype(np.uint8)
 assert cast == image
 (input_image_template, (input_pixel_type, input_image_dimension)) = itk.template(image)
@@ -369,6 +371,8 @@ def custom_callback(name, progress):
 numpyImage = np.random.randint(0, 256, (8, 5, 3)).astype(np.float32)
 image = itk.image_from_array(numpyImage, is_vector=True)
 assert type(image) == type(itk.image_from_array(numpyImage, ttype=(type(image),)))
+assert type(image) == type(itk.image_from_array(numpyImage, ttype=[type(image)]))
+assert type(image) == type(itk.image_from_array(numpyImage, ttype=type(image)))
 vectorimage = itk.cast_image_filter(
     Input=image, ttype=(type(image), itk.VectorImage[itk.F, 2])
 )