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
|
#!/usr/bin/env python
# -*- coding: utf-8 -*-
'''
=========================================================================
Program: Visualization Toolkit
Module: TestNamedColorsIntegration.py
Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
All rights reserved.
See Copyright.txt or http://www.kitware.com/Copyright.htm for details.
This software is distributed WITHOUT ANY WARRANTY; without even
the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
PURPOSE. See the above copyright notice for more information.
=========================================================================
'''
# Run this test like so:
# vtkpython TestGlobFileNames.py -D $VTK_DATA_ROOT
import re
import vtk
import vtk.test.Testing
from vtk.util.misc import vtkGetDataRoot
VTK_DATA_ROOT = vtkGetDataRoot()
class TestGlobFileNames(vtk.test.Testing.vtkTest):
def testGlobFileNames(self):
globFileNames = vtk.vtkGlobFileNames()
globFileNames.SetDirectory(VTK_DATA_ROOT + "/Data/")
# globs do not include Kleene star support for pattern repetitions thus
# we insert a pattern for both single and double digit file extensions.
globFileNames.AddFileNames("headsq/quarter.[1-9]")
globFileNames.AddFileNames("headsq/quarter.[1-9][0-9]")
fileNames = globFileNames.GetFileNames()
n = globFileNames.GetNumberOfFileNames()
if n != 93:
for i in range(0, n):
print("File:", i, " ", fileNames.GetValue(i))
print("GetNumberOfValues should return 93, returned", n)
print("Listing of ", VTK_DATA_ROOT, "/Data/headsq")
directory = vtk.vtkDirectory()
directory.Open(VTK_DATA_ROOT + "/Data/headsq")
m = directory.GetNumberOfFiles()
for j in range(0, m):
print(directory.GetFile(j))
exit(1)
for i in range(0, n):
filename = fileNames.GetValue(i)
if filename != globFileNames.GetNthFileName(i):
print("mismatched filename for pattern quarter.*:", filename)
exit(1)
m = re.search("[\w|\W]*quarter.*", filename)
if m == None:
print("string does not match pattern quarter.*:", filename)
# check that we can re-use the Glob object
globFileNames.Reset()
globFileNames.SetDirectory(VTK_DATA_ROOT + "/Data/")
globFileNames.AddFileNames(VTK_DATA_ROOT + "/Data/financial.*")
fileNames = globFileNames.GetFileNames()
n = fileNames.GetNumberOfValues()
for i in range(0, n):
filename = fileNames.GetValue(i)
if filename != globFileNames.GetNthFileName(i):
print("mismatched filename for pattern financial.*: ", filename)
exit(1)
m = re.search("[\w|\W]*financial.*", filename)
if m == None:
print("string does not match pattern financial.*:", filename)
exit(1)
vtk.test.Testing.interact()
if __name__ == "__main__":
vtk.test.Testing.main([(TestGlobFileNames, 'test')])
|
