38 lines
1.0 KiB
Python
38 lines
1.0 KiB
Python
import doctest
|
|
import importlib
|
|
import glob
|
|
|
|
|
|
TARGET_GLOB = 'aritprog*.py'
|
|
TEST_FILE = 'aritprog.rst'
|
|
TEST_MSG = '{0:16} {1.attempted:2} tests, {1.failed:2} failed - {2}'
|
|
|
|
|
|
def main(argv):
|
|
verbose = '-v' in argv
|
|
for module_file_name in sorted(glob.glob(TARGET_GLOB)):
|
|
module_name = module_file_name.replace('.py', '')
|
|
module = importlib.import_module(module_name)
|
|
gen_factory = getattr(module, 'ArithmeticProgression', None)
|
|
if gen_factory is None:
|
|
gen_factory = getattr(module, 'aritprog_gen', None)
|
|
if gen_factory is None:
|
|
continue
|
|
|
|
test(gen_factory, verbose)
|
|
|
|
|
|
def test(gen_factory, verbose=False):
|
|
res = doctest.testfile(
|
|
TEST_FILE,
|
|
globs={'aritprog_gen': gen_factory},
|
|
verbose=verbose,
|
|
optionflags=doctest.REPORT_ONLY_FIRST_FAILURE)
|
|
tag = 'FAIL' if res.failed else 'OK'
|
|
print(TEST_MSG.format(gen_factory.__module__, res, tag))
|
|
|
|
|
|
if __name__ == '__main__':
|
|
import sys
|
|
main(sys.argv)
|