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
|
"""Test the vtk.reference() type and test pass-by-reference.
Created on Sept 19, 2010 by David Gobbi
"""
import sys
from vtkmodules.vtkCommonCore import reference
from vtkmodules.vtkCommonDataModel import (
vtkCellArray,
vtkPlane,
)
from vtkmodules.test import Testing
class TestPassByReference(Testing.vtkTest):
def testFloatReference(self):
m = reference(3.0)
n = reference(4.0)
m *= 2
self.assertEqual(m, 6.0)
self.assertEqual(str(m), str(m.get()))
o = n + m
self.assertEqual(o, 10.0)
def testIntReference(self):
m = reference(3)
n = reference(4)
m |= n
self.assertEqual(m, 7.0)
self.assertEqual(str(m), str(m.get()))
def testStringReference(self):
m = reference("%s %s!")
m %= ("hello", "world")
self.assertEqual(m, "hello world!")
def testTupleReference(self):
m = reference((0,))
self.assertEqual(m, (0,))
self.assertEqual(len(m), 1)
self.assertEqual(m[0], 0)
def testPassByReferenceerence(self):
t = reference(0.0)
p0 = (0.5, 0.0, 0.0)
n = (1.0, 0.0, 0.0)
p1 = (0.0, 0.0, 0.0)
p2 = (1.0, 1.0, 1.0)
x = [0.0, 0.0, 0.0]
vtkPlane.IntersectWithLine(p1, p2, n, p0, t, x)
self.assertEqual(round(t,6), 0.5)
self.assertEqual(round(x[0],6), 0.5)
self.assertEqual(round(x[1],6), 0.5)
self.assertEqual(round(x[2],6), 0.5)
vtkPlane().IntersectWithLine(p1, p2, n, p0, t, x)
self.assertEqual(round(t,6), 0.5)
self.assertEqual(round(x[0],6), 0.5)
self.assertEqual(round(x[1],6), 0.5)
self.assertEqual(round(x[2],6), 0.5)
t.set(0)
p = vtkPlane()
p.SetOrigin(0.5, 0.0, 0.0)
p.SetNormal(1.0, 0.0, 0.0)
p.IntersectWithLine(p1, p2, t, x)
self.assertEqual(round(t,6), 0.5)
self.assertEqual(round(x[0],6), 0.5)
self.assertEqual(round(x[1],6), 0.5)
self.assertEqual(round(x[2],6), 0.5)
vtkPlane.IntersectWithLine(p, p1, p2, t, x)
self.assertEqual(round(t,6), 0.5)
self.assertEqual(round(x[0],6), 0.5)
self.assertEqual(round(x[1],6), 0.5)
self.assertEqual(round(x[2],6), 0.5)
def testPassTupleByReference(self):
n = reference(0)
t = reference((0,))
ca = vtkCellArray()
ca.InsertNextCell(3, (1, 3, 0))
ca.GetCell(0, n, t)
self.assertEqual(n, 3)
self.assertEqual(tuple(t), (1, 3, 0))
if __name__ == "__main__":
Testing.main([(TestPassByReference, 'test')])
|
