PEP 3147 implementation

Lib/test/test_zipfile.py

 # We can test part of the module without zlib.
 try:
     import zlib
 except ImportError:
     zlib = None
 
 import io
 import os
+import imp
 import time
 import shutil
 import struct
 import zipfile
 import unittest
 
 
 from tempfile import TemporaryFile
 from random import randint, random
 from unittest import skipUnless
@@ skipping 561 matching lines @@
         zipfile.ZIP64_LIMIT = self._limit
         unlink(TESTFN)
         unlink(TESTFN2)
 
 
 class PyZipFileTests(unittest.TestCase):
     def test_write_pyfile(self):
         with zipfile.PyZipFile(TemporaryFile(), "w") as zipfp:
             fn = __file__
             if fn.endswith('.pyc') or fn.endswith('.pyo'):
-                fn = fn[:-1]
+                if '__pycache__' in fn:

[Georg, 2010/04/10 19:07:56]

Probably too nitpicky, but this could pick up an unrelated
"Margaret__pycache__i9" along the path.

[barry, 2010/04/12 21:59:05]

True.  Probably doesn't matter for the tests.  I did however split the pathname
along os.sep and checked for __pycache__ in there.  It's a bit nasty because you
also have to include os.altsep if it's not None.  Not sure I like the new code
any better, but see what you think.

+                    fn = imp.source_from_cache(fn)
+                else:
+                    fn = fn[:-1]
 
             zipfp.writepy(fn)
 
             bn = os.path.basename(fn)
             self.assertNotIn(bn, zipfp.namelist())
             self.assertTrue(bn + 'o' in zipfp.namelist() or
                             bn + 'c' in zipfp.namelist())
 
         with zipfile.PyZipFile(TemporaryFile(), "w") as zipfp:
             fn = __file__
@@ skipping 638 matching lines @@
 
 
 def test_main():
     run_unittest(TestsWithSourceFile, TestZip64InSmallFiles, OtherTests,
                  PyZipFileTests, DecryptionTests, TestsWithMultipleOpens,
                  TestWithDirectory, UniversalNewlineTests,
                  TestsWithRandomBinaryFiles)
 
 if __name__ == "__main__":
     test_main()