From 6395e5ca5120712b36269c549ea1ac82ca4b8e60 Mon Sep 17 00:00:00 2001 From: Ned Batchelder Date: Sun, 31 Jan 2016 18:39:49 -0500 Subject: Yield statements that are not resumed shouldn't be marked as missing. #440 --- coverage/data.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) (limited to 'coverage/data.py') diff --git a/coverage/data.py b/coverage/data.py index 17cf73c..60e104d 100644 --- a/coverage/data.py +++ b/coverage/data.py @@ -4,6 +4,7 @@ """Coverage data for coverage.py.""" import glob +import itertools import json import optparse import os @@ -178,11 +179,12 @@ class CoverageData(object): """ if self._arcs is not None: - if filename in self._arcs: - return [s for s, __ in self._arcs[filename] if s > 0] + arcs = self._arcs.get(filename) + if arcs is not None: + all_lines = itertools.chain.from_iterable(arcs) + return list(set(l for l in all_lines if l > 0)) elif self._lines is not None: - if filename in self._lines: - return self._lines[filename] + return self._lines.get(filename) return None def arcs(self, filename): -- cgit v1.2.1