Move the context determiner to its own file

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.