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
|
"""Test methods that use default parameter values.
Created on Feb 9, 2016 by David Gobbi
"""
import sys
import vtk
from vtk.test import Testing
class TestDefaultArgs(Testing.vtkTest):
def testDefaultInt(self):
"""Simple test of an integer arg with default value."""
image = vtk.vtkImageData()
image.SetExtent(0,9,0,9,0,9)
image.AllocateScalars(vtk.VTK_UNSIGNED_CHAR, 1)
ipi = vtk.vtkImagePointIterator()
# call this method with the threadId parameter set to 0
ipi.Initialize(image, (0,9,0,9,0,9), None, None, 0)
# call this method without the threadId parameter
ipi.Initialize(image, (0,9,0,9,0,9), None, None)
def testDefaultObjectPointer(self):
"""Test a vtkObject pointer arg with default value of 0."""
image = vtk.vtkImageData()
image.SetExtent(0,9,0,9,0,9)
image.AllocateScalars(vtk.VTK_UNSIGNED_CHAR, 1)
ipi = vtk.vtkImagePointIterator()
# call this method with the stencil parameter set to None
ipi.Initialize(image, (0,9,0,9,0,9), None)
# call this method without the stencil parameter
ipi.Initialize(image, (0,9,0,9,0,9))
def testDefaultArray(self):
"""Test an array arg with default value of 0."""
image = vtk.vtkImageData()
image.SetExtent(0,9,0,9,0,9)
image.AllocateScalars(vtk.VTK_UNSIGNED_CHAR, 1)
ipi = vtk.vtkImagePointIterator()
# call this method with the parameter set
ipi.Initialize(image, (0,9,0,9,0,9))
# call this method without extent parameter
ipi.Initialize(image)
def testDefaultPointer(self):
"""Test a POD pointer arg with default value of 0."""
a = vtk.vtkIntArray()
a.SetNumberOfComponents(3)
# pass an int pointer arg, expect something back
inc = [0]
vtk.vtkImagePointDataIterator.GetVoidPointer(a, 0, inc)
self.assertEqual(inc, [3])
# do not pass the pointer arg, default value 0 is passed
vtk.vtkImagePointDataIterator.GetVoidPointer(a, 0)
if __name__ == "__main__":
Testing.main([(TestDefaultArgs, 'test')])
|