summaryrefslogtreecommitdiff
path: root/coverage/data.py
diff options
context:
space:
mode:
authorNed Batchelder <ned@nedbatchelder.com>2013-12-13 22:45:10 -0500
committerNed Batchelder <ned@nedbatchelder.com>2013-12-13 22:45:10 -0500
commit2df9b1c35cbb5c92204fc5923368a3d619a34f6d (patch)
treed1ede8ffef812ba4e345b08f698f001ebe69cb56 /coverage/data.py
parent84221611890880b749dbb650e8d07ac8918dba46 (diff)
parent7c66441eab3af17539c478a2cb4e19cd93ba0cf4 (diff)
downloadpython-coveragepy-git-2df9b1c35cbb5c92204fc5923368a3d619a34f6d.tar.gz
Merged 4.0 to default
Diffstat (limited to 'coverage/data.py')
-rw-r--r--coverage/data.py16
1 files changed, 5 insertions, 11 deletions
diff --git a/coverage/data.py b/coverage/data.py
index fb88c5b1..042b6405 100644
--- a/coverage/data.py
+++ b/coverage/data.py
@@ -2,7 +2,7 @@
import os
-from coverage.backward import iitems, pickle, sorted # pylint: disable=W0622
+from coverage.backward import iitems, pickle
from coverage.files import PathAliases
from coverage.misc import file_be_gone
@@ -101,13 +101,13 @@ class CoverageData(object):
def line_data(self):
"""Return the map from filenames to lists of line numbers executed."""
return dict(
- [(f, sorted(lmap.keys())) for f, lmap in iitems(self.lines)]
+ (f, sorted(lmap.keys())) for f, lmap in iitems(self.lines)
)
def arc_data(self):
"""Return the map from filenames to lists of line number pairs."""
return dict(
- [(f, sorted(amap.keys())) for f, amap in iitems(self.arcs)]
+ (f, sorted(amap.keys())) for f, amap in iitems(self.arcs)
)
def write_file(self, filename):
@@ -128,11 +128,8 @@ class CoverageData(object):
self.debug.write("Writing data to %r" % (filename,))
# Write the pickle to the file.
- fdata = open(filename, 'wb')
- try:
+ with open(filename, 'wb') as fdata:
pickle.dump(data, fdata, 2)
- finally:
- fdata.close()
def read_file(self, filename):
"""Read the coverage data from `filename`."""
@@ -142,11 +139,8 @@ class CoverageData(object):
"""Return the raw pickled data from `filename`."""
if self.debug and self.debug.should('dataio'):
self.debug.write("Reading data from %r" % (filename,))
- fdata = open(filename, 'rb')
- try:
+ with open(filename, 'rb') as fdata:
data = pickle.load(fdata)
- finally:
- fdata.close()
return data
def _read_file(self, filename):