| 1 | n/a | import os |
|---|
| 2 | n/a | import sys |
|---|
| 3 | n/a | import unittest |
|---|
| 4 | n/a | |
|---|
| 5 | n/a | def test_suite(package=__package__, directory=os.path.dirname(__file__)): |
|---|
| 6 | n/a | suite = unittest.TestSuite() |
|---|
| 7 | n/a | for name in os.listdir(directory): |
|---|
| 8 | n/a | if name.startswith(('.', '__')): |
|---|
| 9 | n/a | continue |
|---|
| 10 | n/a | path = os.path.join(directory, name) |
|---|
| 11 | n/a | if (os.path.isfile(path) and name.startswith('test_') and |
|---|
| 12 | n/a | name.endswith('.py')): |
|---|
| 13 | n/a | submodule_name = os.path.splitext(name)[0] |
|---|
| 14 | n/a | module_name = "{0}.{1}".format(package, submodule_name) |
|---|
| 15 | n/a | __import__(module_name, level=0) |
|---|
| 16 | n/a | module_tests = unittest.findTestCases(sys.modules[module_name]) |
|---|
| 17 | n/a | suite.addTest(module_tests) |
|---|
| 18 | n/a | elif os.path.isdir(path): |
|---|
| 19 | n/a | package_name = "{0}.{1}".format(package, name) |
|---|
| 20 | n/a | __import__(package_name, level=0) |
|---|
| 21 | n/a | package_tests = getattr(sys.modules[package_name], 'test_suite')() |
|---|
| 22 | n/a | suite.addTest(package_tests) |
|---|
| 23 | n/a | else: |
|---|
| 24 | n/a | continue |
|---|
| 25 | n/a | return suite |
|---|
