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
|
#!@PYTHON_EXECUTABLE@
# set environment variables
from __future__ import print_function
import os, subprocess, sys
from optparse import OptionParser
# the environment needed to find the lib depend on the system
# I'm really not sure that's needed on all the systems...
# TODO: can we replace os.name by os.uname()[0] for all systems ?
env_lib_path = {"posix": "LD_LIBRARY_PATH", "mac": "DYLD_LIBRARY_PATH", "nt": "PATH"}[os.name]
# the environment needed to find the lib depend on the system
if os.name == "posix":
if os.uname()[0] == "Darwin" :
env_lib_path = "DYLD_LIBRARY_PATH"
else :
env_lib_path = "LD_LIBRARY_PATH"
elif os.name == "nt":
env_lib_path = "PATH"
else:
raise "Please add the library path for your system."
lib_path_list = "@PYTHON_TEST_LIB_PATH@".split(';')
if os.environ.has_key(env_lib_path) :
os.environ[env_lib_path] = os.pathsep.join(lib_path_list + [os.environ[env_lib_path]])
else:
os.environ[env_lib_path] = os.pathsep.join(lib_path_list)
env_python_path = "PYTHONPATH"
python_path_list = "@PYTHON_TEST_PYTHON_PATH@".split(';')
if os.environ.has_key(env_python_path) :
os.environ[env_python_path] = os.pathsep.join(python_path_list + [os.environ[env_python_path]])
else:
os.environ[env_python_path] = os.pathsep.join(python_path_list)
env_wrapitk_python_path = "WRAPITK_PYTHON_PATH"
wrapitk_python_path_list = "@PYTHON_TEST_WRAPITK_PYTHON_PATH@".split(';')
if os.environ.has_key(env_wrapitk_python_path) :
os.environ[env_wrapitk_python_path] = os.pathsep.join(wrapitk_python_path_list + [os.environ[env_wrapitk_python_path]])
else:
os.environ[env_wrapitk_python_path] = os.pathsep.join(wrapitk_python_path_list)
# setup the option manager to be able to compare the resulting images in the same test
optionParser = OptionParser(usage="usage: %prog [--compare image1 image2] test.py [arg1 [arg2 [...]]]")
optionParser.add_option("-c", "--compare", action="append", dest="compare", default=[], metavar="IMAGE1 IMAGE2", nargs=2, help="compare IMAGE1 and IMAGE2 and exit with an error if the images are different. This options can appear several times to compare several images.")
optionParser.add_option("-v", "--verbose", action="store_true", dest="verbose", default=False, help="display the command line before running the programs.")
(options, args) = optionParser.parse_args()
if len(args) < 1:
optionParser.print_usage(sys.stderr)
sys.exit(1)
# call the program, and exit if it fail
command = ["@PYTHON_EXECUTABLE@"] + args
if options.verbose:
print(sys.stderr, "+", " ".join(command))
returnValue = subprocess.call( command )
if returnValue != 0 :
sys.exit( returnValue )
# compare the images if needed
for imageFile1, imageFile2 in options.compare :
command = ["@IMAGE_COMPARE@", imageFile1, imageFile2]
if options.verbose:
print(sys.stderr, "+", " ".join(command))
returnValue = subprocess.call( command )
if returnValue != 0 :
sys.exit( returnValue )
# everything suceed - just exit
sys.exit(0)
|
