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
|
"""Test overloaded method resolution in VTK-Python
The wrappers should call overloaded C++ methods using similar
overload resolution rules as C++. Python itself does not have
method overloading.
Created on Feb 15, 2015 by David Gobbi
"""
import sys
from vtkmodules.vtkCommonCore import (
VTK_DOUBLE,
VTK_INT,
VTK_OBJECT,
VTK_STRING,
reference,
vtkDenseArray,
vtkFloatArray,
vtkIntArray,
vtkObject,
vtkVariant,
vtkVariantArray,
)
from vtkmodules.vtkCommonDataModel import (
vtkFieldData,
vtkVector3d,
)
from vtkmodules.vtkCommonMath import vtkMatrix4x4
from vtkmodules.vtkCommonTransforms import vtkTransform
from vtkmodules.test import Testing
class TestOverloads(Testing.vtkTest):
def testMethods(self):
"""Test overloaded methods"""
# single-argument method vtkTransform::SetMatrix()
t = vtkTransform()
m = vtkMatrix4x4()
m.SetElement(0, 0, 2)
t.SetMatrix(m)
self.assertEqual(t.GetMatrix().GetElement(0, 0), 2)
t.SetMatrix([0,1,0,0, 1,0,0,0, 0,0,-1,0, 0,0,0,1])
self.assertEqual(t.GetMatrix().GetElement(0, 0), 0)
# mixed number of arguments
fd = vtkFieldData()
fa = vtkFloatArray()
fa.SetName("Real")
ia = vtkIntArray()
ia.SetName("Integer")
fd.AddArray(fa)
fd.AddArray(ia)
a = fd.GetArray("Real")
self.assertEqual(id(a), id(fa))
i = reference(0)
a = fd.GetArray("Integer", i)
self.assertEqual(id(a), id(ia))
self.assertEqual(i, 1)
def testConstructors(self):
"""Test overloaded constructors"""
# resolve by number of arguments
v = vtkVector3d(3, 4, 5)
self.assertEqual((v[0], v[1], v[2]), (3, 4, 5))
v = vtkVector3d(6)
self.assertEqual((v[0], v[1], v[2]), (6, 6, 6))
# resolve by argument type
v = vtkVariant(3.0)
self.assertEqual(v.GetType(), VTK_DOUBLE)
v = vtkVariant(1)
self.assertEqual(v.GetType(), VTK_INT)
v = vtkVariant("hello")
self.assertEqual(v.GetType(), VTK_STRING)
v = vtkVariant(vtkObject())
self.assertEqual(v.GetType(), VTK_OBJECT)
def testArgumentConversion(self):
"""Test argument conversion via implicit constructors"""
# automatic conversion to vtkVariant
a = vtkVariantArray()
a.InsertNextValue(2.5)
a.InsertNextValue(vtkObject())
self.assertEqual(a.GetValue(0), vtkVariant(2.5))
self.assertEqual(a.GetValue(1).GetType(), VTK_OBJECT)
# same, but this one is via "const vtkVariant&" argument
a = vtkDenseArray[float]()
a.Resize(1)
a.SetVariantValue(0, 2.5)
self.assertEqual(a.GetVariantValue(0).ToDouble(), 2.5)
if __name__ == "__main__":
Testing.main([(TestOverloads, 'test')])
|
