From 8acd852de3faab4406bc05abf312ed18ed2985ca Mon Sep 17 00:00:00 2001 From: Ned Batchelder Date: Fri, 28 Sep 2018 19:27:41 -0400 Subject: Move the context determiner to its own file --- coverage/context.py | 11 +++++++++++ coverage/control.py | 9 +-------- 2 files changed, 12 insertions(+), 8 deletions(-) create mode 100644 coverage/context.py (limited to 'coverage') 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. -- cgit v1.2.1