forked from statsmodels/statsmodels
-
Notifications
You must be signed in to change notification settings - Fork 0
/
__init__.py
82 lines (68 loc) · 2.55 KB
/
__init__.py
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
#
# models - Statistical Models
#
from __future__ import with_statement
__docformat__ = 'restructuredtext'
#from version import __version__
#from info import __doc__
#from regression import *
#from genmod.glm import *
#from robust.rlm import *
#from discrete.discretemod import *
#import tsa
#from tools.tools import add_constant, chain_dot
#import base.model
#import tools.tools
#import datasets
#import glm.families
#import stats.stattools
#import iolib
from numpy import errstate
#__all__ = filter(lambda s:not s.startswith('_'),dir())
from numpy.testing import Tester
class NoseWrapper(Tester):
'''
This is simply a monkey patch for numpy.testing.Tester.
It allows extra_argv to be changed from its default None to ['--exe'] so
that the tests can be run the same across platforms. It also takes kwargs
that are passed to numpy.errstate to suppress floating point warnings.
'''
def test(self, label='fast', verbose=1, extra_argv=['--exe'], doctests=False,
coverage=False, **kwargs):
''' Run tests for module using nose
%(test_header)s
doctests : boolean
If True, run doctests in module, default False
coverage : boolean
If True, report coverage of NumPy code, default False
(Requires the coverage module:
http://nedbatchelder.com/code/modules/coverage.html)
kwargs
Passed to numpy.errstate. See its documentation for details.
'''
# cap verbosity at 3 because nose becomes *very* verbose beyond that
verbose = min(verbose, 3)
from numpy.testing import utils
utils.verbose = verbose
if doctests:
print("Running unit tests and doctests for %s" % self.package_name)
else:
print("Running unit tests for %s" % self.package_name)
self._show_system_info()
# reset doctest state on every run
import doctest
doctest.master = None
argv, plugins = self.prepare_test_args(label, verbose, extra_argv,
doctests, coverage)
from numpy.testing.noseclasses import NumpyTestProgram
from warnings import simplefilter #, catch_warnings
with errstate(**kwargs):
## with catch_warnings():
simplefilter('ignore', category=DeprecationWarning)
t = NumpyTestProgram(argv=argv, exit=False, plugins=plugins)
return t.result
test = NoseWrapper().test
try:
from .version import version as __version__
except ImportError:
__version__ = 'not-yet-built'