/
test.py
42 lines (35 loc) · 1.23 KB
/
test.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
#!/usr/bin/env python
import sys
from os import path
from os import getcwd
def run(verbosity=1, doctest=False, numpy=True):
"""Run causalimpact tests.
Parameters
----------
verbosity: integer, optional
Level of detail in test reports. Higher numbers provide more detail.
doctest: bool, optional
True to run doctests in code modules
numpy: bool, optional
True to test modules dependent on numpy
"""
try:
import nose
except ImportError:
raise ImportError("The nose package is needed to run the causalimpact tests.")
sys.stderr.write("Running causalimpact tests:")
ci_install_dir = path.join(path.dirname(__file__), path.pardir)
# stop if running from source directory
if getcwd() == path.abspath(path.join(ci_install_dir, path.pardir)):
raise RuntimeError(
"Can't run tests from source directory.\n"
"Run 'nosetests' from the command line."
)
argv = [" ", "--verbosity = %d" % verbosity, "-w", ci_install_dir, "-exe"]
if doctest:
argv.extend(["--with-doctest", "--doctest-extension=txt"])
if not numpy:
argv.extend(["-A not numpy"])
nose.run(argv=argv)
if __name__ == "__main__":
run()