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
|
"""Coverage-test Coverage.py itself."""
import os, shutil, sys
import nose
HTML_DIR = "htmlcov"
def run_tests_with_coverage():
"""Run the test suite with coverage measuring itself."""
import coverage
tracer = os.environ.get('COVERAGE_TEST_TRACER', 'c')
version = "%s%s" % sys.version_info[:2]
suffix = ".%s_%s" % (version, tracer)
cov = coverage.coverage(branch=True, data_suffix=suffix)
# Cheap trick: the coverage code itself is excluded from measurement, but
# if we clobber the cover_prefix in the coverage object, we can defeat the
# self-detection.
cov.cover_prefix = "Please measure coverage.py!"
cov.erase()
cov.start()
# Re-import coverage to get it coverage tested! I don't understand all the
# mechanics here, but if I don't carry over the imported modules (in
# covmods), then things go haywire (os == None, eventually).
covmods = {}
covdir = os.path.split(coverage.__file__)[0]
# We have to make a list since we'll be deleting in the loop.
modules = list(sys.modules.items())
for name, mod in modules:
if name.startswith('coverage'):
if hasattr(mod, '__file__') and mod.__file__.startswith(covdir):
covmods[name] = mod
del sys.modules[name]
import coverage # don't warn about re-import: pylint: disable-msg=W0404
sys.modules.update(covmods)
# Run nosetests, with the arguments from our command line.
print(":: Running nosetests %s" % " ".join(sys.argv[1:]))
nose.run()
cov.stop()
print(":: Saving .coverage%s" % suffix)
cov.save()
def report_on_combined_files():
"""Combine all the .coverage files and make an HTML report."""
if os.path.exists(HTML_DIR):
shutil.rmtree(HTML_DIR)
print(":: Writing HTML report to %s/index.html" % HTML_DIR)
import coverage
cov = coverage.coverage()
cov.combine()
cov.save()
cov.clear_exclude()
cov.exclude("#pragma: no cover")
cov.exclude("def __repr__")
cov.exclude("if __name__ == .__main__.:")
cov.exclude("raise AssertionError")
cov.html_report(
directory=HTML_DIR, ignore_errors=True, omit_prefixes=["mock"]
)
try:
cmd = sys.argv[1]
except IndexError:
cmd = ''
if cmd == 'run':
# Ugly hack: nose.run reads sys.argv directly, so here I delete my command
# argument so that sys.argv is left as just nose arguments.
del sys.argv[1]
run_tests_with_coverage()
elif cmd == 'report':
report_on_combined_files()
else:
print("Need 'run' or 'report'")
|
