/usr/src/WrapITK/Python/Tests/template.py is in libinsighttoolkit3-dev 3.20.1-1.
This file is owned by root:root, with mode 0o755.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 | #!/usr/bin/env python
import itk
dim = 2
PType = itk.US
# check the repr string
assert "<itkTemplate itk::Image>" == repr(itk.Image)
# template should work with CType instance and with numbers
IType = itk.Image[PType, dim]
# template should return the same class with a class as parameter
# or with an object of this class, and should also be the same
# with the attribute
# create instances of image for the next tests
im = IType.New()
im2 = IType.New()
readerType = itk.ImageFileReader[IType]
readerType2 = itk.ImageFileReader[im]
readerType3 = itk.ImageFileReader.IUS2
assert readerType == readerType2 == readerType3
# we should be able to get the template and its parameters from the class
(tpl, parameters) = itk.template( IType )
assert tpl == itk.Image
assert parameters == (PType, dim)
# the template must raise a KeyError exception if the template parameter
# is unknown
try :
itk.ImageFileReader['unknown parameter']
raise Exception('no exception sent for unknown parameter')
except KeyError:
pass
# TODO: test the rest of the dict interface
# TODO: test __eq__, __ne__ and __hash__
# something else ?
# now test the New method
# without parameter
reader = readerType.New()
reader2 = readerType.New()
# with an attribute parameter
reader = readerType.New(FileName='test.png')
assert reader.GetFileName() == 'test.png'
# wwith a wrong attribute name
try :
reader = readerType.New(WrongName='test.png')
raise Exception('no exception sent for wrong attribute name')
except AttributeError:
pass
# wwith a wrong attribute type
try :
reader = readerType.New(FileName=1)
raise Exception('no exception sent for wrong attribute type')
except TypeError:
pass
# pass filter as argument for input
# to a filter with SetInput method
median = itk.MedianImageFilter[IType, IType].New(reader)
assert repr(reader.GetOutput().GetPointer()) == repr(median.GetInput().GetPointer())
# to a filter with a SetImage method
calculator = itk.MinimumMaximumImageCalculator[IType].New(reader)
# not GetImage() method here to verify it's the right image
# to a filter with several inputs
sub = itk.SubtractImageFilter[IType, IType, IType].New(reader, reader2)
assert repr(reader.GetOutput().GetPointer()) == repr(sub.GetInput(0).GetPointer())
assert repr(reader2.GetOutput().GetPointer()) == repr(sub.GetInput(1).GetPointer())
# pass image as argument for input
# to a filter with SetInput method
median = itk.MedianImageFilter[IType, IType].New(im)
assert repr(im.GetPointer()) == repr(median.GetInput().GetPointer())
# to a filter with a SetImage method
calculator = itk.MinimumMaximumImageCalculator[IType].New(im)
# not GetImage() method here to verify it's the right image
# to a filter with several inputs
sub = itk.SubtractImageFilter[IType, IType, IType].New(im, im2)
assert repr(im.GetPointer()) == repr(sub.GetInput(0).GetPointer())
assert repr(im2.GetPointer()) == repr(sub.GetInput(1).GetPointer())
# pass invalid input
try:
itk.MedianImageFilter[IType, IType].New(1)
raise Exception('no exception sent for wrong input type')
except TypeError:
pass
try:
itk.SubtractImageFilter[IType, IType, IType].New(im, 1)
raise Exception('no exception sent for wrong 2nd input type')
except TypeError:
pass
# pass both input and attribute
recons = itk.ReconstructionByDilationImageFilter[IType, IType].New(reader.GetOutput(), im, FullyConnected=True)
assert repr(reader.GetOutput().GetPointer()) == repr(recons.GetInput(0).GetPointer())
assert repr(im.GetPointer()) == repr(recons.GetInput(1).GetPointer())
assert recons.GetFullyConnected() == True
# pass input to object which do not take one
try:
IType.New(im)
raise Exception('no exception sent for object without input')
except AttributeError:
pass
# TODO: test auto_progress
# but how ?
# something else ?
|