diff options
Diffstat (limited to 'coverage/data.py')
| -rw-r--r-- | coverage/data.py | 12 |
1 files changed, 8 insertions, 4 deletions
diff --git a/coverage/data.py b/coverage/data.py index 1b883750..fe076bef 100644 --- a/coverage/data.py +++ b/coverage/data.py @@ -56,6 +56,10 @@ class CoverageData(object): # self.arcs = {} + self.os = os + self.sorted = sorted + self.pickle = pickle + def usefile(self, use_file=True): """Set whether or not to use a disk file for data.""" self.use_file = use_file @@ -93,13 +97,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 self.lines.items()] + [(f, self.sorted(lmap.keys())) for f, lmap in self.lines.items()] ) 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 self.arcs.items()] + [(f, self.sorted(amap.keys())) for f, amap in self.arcs.items()] ) def write_file(self, filename): @@ -119,7 +123,7 @@ class CoverageData(object): # Write the pickle to the file. fdata = open(filename, 'wb') try: - pickle.dump(data, fdata, 2) + self.pickle.dump(data, fdata, 2) finally: fdata.close() @@ -229,7 +233,7 @@ class CoverageData(object): if fullpath: filename_fn = lambda f: f else: - filename_fn = os.path.basename + filename_fn = self.os.path.basename for filename, lines in self.lines.items(): summ[filename_fn(filename)] = len(lines) return summ |