From 6ec3302004a2293535e5c3814c1c19e2f141e972 Mon Sep 17 00:00:00 2001 From: Ned Batchelder Date: Mon, 7 Feb 2011 08:37:14 -0500 Subject: Don't warn about not collecting data if we never ran any code in the first place. --- coverage/control.py | 21 ++++++++++++++++----- 1 file changed, 16 insertions(+), 5 deletions(-) (limited to 'coverage/control.py') diff --git a/coverage/control.py b/coverage/control.py index 514f23d..6fc1b98 100644 --- a/coverage/control.py +++ b/coverage/control.py @@ -172,6 +172,10 @@ class coverage(object): # Only _harvest_data once per measurement cycle. self._harvested = False + + # When stop() is called, we can tell it that in fact no product code + # was run, to make the warnings more reasonable. + self._never_run = False # Set the reporting precision. Numbers.set_precision(self.config.precision) @@ -361,8 +365,14 @@ class coverage(object): self._harvested = False self.collector.start() - def stop(self): - """Stop measuring code coverage.""" + def stop(self, never_run=False): + """Stop measuring code coverage. + + Set `never_run` to True to indicate that no product code was run, so + we don't warn unnecessarily. + + """ + self._never_run = never_run self.collector.stop() self._harvest_data() @@ -444,9 +454,10 @@ class coverage(object): self._warn("Module %s was never imported." % pkg) # Find out if we got any data. - summary = self.data.summary() - if not summary: - self._warn("No data was collected.") + if not self._never_run: + summary = self.data.summary() + if not summary: + self._warn("No data was collected.") # Find files that were never executed at all. for src in self.source: -- cgit v1.2.1