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
98
99
100
101
102
103
104
105
|
##############################################################################
#
# Copyright (c) 2014-2018 by The University of Queensland
# http://www.uq.edu.au
#
# Primary Business: Queensland, Australia
# Licensed under the Apache License, version 2.0
# http://www.apache.org/licenses/LICENSE-2.0
#
# Development until 2012 by Earth Systems Science Computational Center (ESSCC)
# Development 2012-2013 by School of Earth Sciences
# Development from 2014 by Centre for Geoscience Computing (GeoComp)
# Development from 2019 by School of Earth and Environmental Sciences
#
##############################################################################
"""
A small set of functions to automate test discovery and running and also
allow specific tests to be run without modifying the test scripts themselves.
When imported into a test file, any external script can import the test file
module and run tests as they wish.
Both specific and general forms of the functions return the result state,
allowing access to list all skipped tests and other data kept by the
`TestTextResults` returned.
Examples:
Running all test classes::
import your_test_module_or_file as tests
tests.run_tests("your_test_module_or_file")
Running specific test classes::
import your_test_module_or_file as tests
tests.run_tests("your_test_module_or_file", [tests.Test_classA, tests.Test_classB])
Running a specific test within a class::
import your_test_module_or_file as tests
tests.run_single_test(tests.Test_classname("test_functionname"))
Printing the list of skipped tests::
results = tests.run_tests("name")
for skipped_test in results.skipped:
print(skipped_test)
"""
from __future__ import print_function, division
__copyright__="""Copyright (c) 2014-2018 by The University of Queensland
http://www.uq.edu.au
Primary Business: Queensland, Australia"""
__license__="""Licensed under the Apache License, version 2.0
http://www.apache.org/licenses/LICENSE-2.0"""
__url__="https://launchpad.net/escript-finley"
import esys.escriptcore.utestselect as unittest
from .util import getMPIRankWorld
from .escriptcpp import MPIBarrierWorld
import os
import sys
def __add_tests(suite, modulename):
import inspect
clsmembers = inspect.getmembers(sys.modules[modulename], inspect.isclass)
for name, cls in clsmembers:
if modulename == cls.__module__ and name.startswith("Test") \
and issubclass(cls, unittest.TestCase):
suite.addTest(unittest.makeSuite(cls))
def run_single_test(test, exit_on_failure=False):
suite = unittest.TestSuite()
suite.addTest(test)
s=unittest.TextTestRunner(verbosity=2).run(suite)
if exit_on_failure and not s.wasSuccessful():
sys.exit(1)
return s
def run_tests(modulename, classes = [], exit_on_failure = False):
rank = getMPIRankWorld()
stream=sys.stderr #default
verb=2
if rank > 0:
stream=open(os.devnull,'w')
verb=0
suite = unittest.TestSuite()
if len(classes) == 0:
__add_tests(suite, modulename)
else:
for test_class in classes:
suite.addTest(unittest.makeSuite(test_class))
s=unittest.TextTestRunner(stream=stream,verbosity=verb).run(suite)
if exit_on_failure and not s.wasSuccessful():
sys.stderr.flush()
MPIBarrierWorld()
sys.exit(1)
return s
|
