Use some strange self.os stuff to hold references to modules so we'll have them available when called from atexit(). Fixes issue #71, I think.

1 files changed, 8 insertions, 4 deletions

diff --git a/coverage/data.py b/coverage/data.py
index 1b88375..fe076be 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