diff options
| -rw-r--r-- | coverage/context.py | 11 | ||||
| -rw-r--r-- | coverage/control.py | 9 |
2 files changed, 12 insertions, 8 deletions
diff --git a/coverage/context.py b/coverage/context.py new file mode 100644 index 00000000..24d01f2a --- /dev/null +++ b/coverage/context.py @@ -0,0 +1,11 @@ +# Licensed under the Apache License: http://www.apache.org/licenses/LICENSE-2.0 +# For details: https://github.com/nedbat/coveragepy/blob/master/NOTICE.txt + +"""Determine contexts for coverage.py""" + +def should_start_context_test_function(frame): + """Is this frame calling a test_* function?""" + fn_name = frame.f_code.co_name + if fn_name.startswith("test"): + return fn_name + return None diff --git a/coverage/control.py b/coverage/control.py index 83bcedb5..dd862ae9 100644 --- a/coverage/control.py +++ b/coverage/control.py @@ -14,6 +14,7 @@ from coverage.annotate import AnnotateReporter from coverage.backward import string_class, iitems from coverage.collector import Collector, CTracer from coverage.config import read_coverage_config +from coverage.context import should_start_context_test_function from coverage.data import CoverageData, combine_parallel_data from coverage.debug import DebugControl, write_formatted_info from coverage.disposition import disposition_debug_msg @@ -896,14 +897,6 @@ if int(os.environ.get("COVERAGE_DEBUG_CALLS", 0)): # pragma: debugg Coverage = decorate_methods(show_calls(show_args=True), butnot=['get_data'])(Coverage) -def should_start_context_test_function(frame): - """Who-Tests-What hack: Determine whether this frame begins a new who-context.""" - fn_name = frame.f_code.co_name - if fn_name.startswith("test"): - return fn_name - return None - - def process_startup(): """Call this at Python start-up to perhaps measure coverage. |