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
|
#! /usr/bin/env python
import openturns as ot
import sys
import os
import shutil
from glob import glob
from tempfile import mkdtemp
import doctest
use_matplotlib = False
try:
import matplotlib
matplotlib.use('Agg')
use_matplotlib = True
except:
pass
ot_path = os.path.dirname(ot.__file__)
py_files = sorted(glob(ot_path + '/*.py'))
total_failure_count = 0
total_test_count = 0
# Temporary working directory for IO tests
work_dir = mkdtemp()
os.chdir(work_dir)
for py_file in py_files:
if (not use_matplotlib) and (os.path.basename(py_file) == 'viewer.py'):
failure_count, test_count = 0, 0
else:
py_basename = os.path.splitext(os.path.basename(py_file))[0]
module = __import__('openturns.' + py_basename, fromlist=[py_basename])
failure_count, test_count = doctest.testmod(
module, verbose=False, optionflags=doctest.ELLIPSIS)
total_failure_count += failure_count
total_test_count += test_count
print(('%s %5d tests failed'
% ((py_basename + ' ').ljust(60, '.'), failure_count)))
print(('-' * 79))
print(('%s %5d tests failed'
% ('TOTAL '.ljust(60, '.'), total_failure_count)))
# Delete temporary working directory for IO tests
os.chdir('..')
shutil.rmtree(work_dir)
|