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
94
95
96
97
|
#!/usr/bin/env python
"""Script to run all the tests.
"""
# Author: Prabhu Ramachandran <prabhu [at] aero . iitb . ac . in>
# Copyright (c) 2008-2020, Prabhu Ramachandran
# License: BSD Style.
import sys
import os
from os.path import splitext
import glob
from common import TestCase
def get_tests():
"""Get all the tests to run.
"""
files = glob.glob('test_*.py')
return files
def run_all(tests):
"""Run the given tests.
"""
args = ' '.join(sys.argv[1:])
success = []
fail = []
for test in tests:
cmd = 'python %s %s'%(test, args)
print(cmd)
status = os.system(cmd)
if status == 0:
print("OK")
success.append(test)
else:
print("FAIL: %s"%test)
fail.append(test)
print('-'*70)
print("%d successful tests, %d failures"%(len(success), len(fail)))
for test in fail:
print(test)
print('-'*70)
return len(fail) != 0
class RunAllTests(TestCase):
"""Runs all the tests in one go, instead of running each test
separately. This speeds up the testing.
"""
def get_tests(self):
tests = get_tests()
tests = [splitext(t)[0] for t in tests]
klasses = []
for test in tests:
# Find test.
m = __import__(test)
m.mayavi = self.script
m.application = self.application
for name in dir(m):
klass = getattr(m, name)
try:
if issubclass(klass, TestCase) and klass is not TestCase:
mod_name = '%s.%s'%(test, name)
klasses.append((mod_name, klass))
break
except TypeError:
continue
return klasses
def do(self):
klasses = self.get_tests()
for name, klass in klasses:
# Close existing scenes.
e = self.script.engine
for scene in e.scenes:
e.close_scene(scene)
print('*'*80)
print(name)
obj = klass()
obj.trait_set(script=self.script)
obj.test()
def main():
argv = ' '.join(sys.argv)
if '--one-shot' in argv:
argv = argv.replace('--one-shot', '')
sys.argv = argv.split()
t = RunAllTests()
t.main()
else:
tests = get_tests()
status = run_all(tests)
sys.exit(status)
if __name__ == "__main__":
main()
|
