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
|
"""Elementary tests for wrapper_gen.py.
This test suite is not comprehensive because it is very hard to test
wrapper_gen directly.
"""
# Author: Prabhu Ramachandran <prabhu_r@users.sf.net>
# Copyright (c) 2004, Prabhu Ramachandran, Enthought, Inc.
# License: BSD Style.
import unittest
import vtk
from tvtk import wrapper_gen
_cache = wrapper_gen.WrapperGenerator()
class TestWrapperGenerator(unittest.TestCase):
def setUp(self):
self.wg = _cache
def test_find_type(self):
wg = self.wg
sigs = ['int', 'vtkOpenGLVolumeMapper',
('int', 'int', 'float', 'list'),
('int', 'vtkActor', 'vtkXMLReader'),
['vtkImageActor', 'vtkExporter'],
['int', 'vtkDataArray', 'vtkCellArray',
'vtkIdTypeArray']
]
expect = ['basic', 'vtk', 'basic', 'vtk', 'vtk', 'array']
for i, sig in enumerate(sigs):
self.assertEqual(expect[i], wg._find_type(sig))
def test_sig_types(self):
wg = self.wg
meths = [vtk.vtkProperty.GetColor,
vtk.vtkProperty.GetRepresentation,
vtk.vtkStructuredPointsReader.GetOutput,
vtk.vtkPolyData.SetPoints,
vtk.vtkPolyData.SetPolys,
vtk.vtkQuad.CellBoundary,
vtk.vtkContourFilter.SetLocator
]
expect = [('basic', 'basic'),
('basic', None),
('vtk', 'basic'),
('basic', 'array'),
('basic', 'array'),
('basic', 'array'),
('basic', 'vtk'),
]
for i, meth in enumerate(meths):
sig = wg.parser.get_method_signature(meth)
self.assertEqual(expect[i], wg._find_sig_type(sig))
sig = [(['int'], ['int']),
(['vtkStructuredPoints'], ['vtkFooClass'])]
self.assertEqual(('vtk', 'vtk'), wg._find_sig_type(sig))
def test_unicode_return_value(self):
wg = self.wg
try:
meth = vtk.vtkDelimitedTextReader.GetUTF8RecordDelimiters
except AttributeError: # VTK < 9.2.2
meth = vtk.vtkDelimitedTextReader.GetUnicodeRecordDelimiters
expect = 'unicode' if '-> unicode' in meth.__doc__ else 'string'
sig = wg.parser.get_method_signature(meth)
self.assertEqual(sig[0][0][0], expect)
if __name__ == "__main__":
unittest.main()
|
