| import imp |
| import unittest |
| from test import support |
| |
| |
| class LockTests(unittest.TestCase): |
| |
| """Very basic test of import lock functions.""" |
| |
| def verify_lock_state(self, expected): |
| self.failUnlessEqual(imp.lock_held(), expected, |
| "expected imp.lock_held() to be %r" % expected) |
| def testLock(self): |
| LOOPS = 50 |
| |
| # The import lock may already be held, e.g. if the test suite is run |
| # via "import test.autotest". |
| lock_held_at_start = imp.lock_held() |
| self.verify_lock_state(lock_held_at_start) |
| |
| for i in range(LOOPS): |
| imp.acquire_lock() |
| self.verify_lock_state(True) |
| |
| for i in range(LOOPS): |
| imp.release_lock() |
| |
| # The original state should be restored now. |
| self.verify_lock_state(lock_held_at_start) |
| |
| if not lock_held_at_start: |
| try: |
| imp.release_lock() |
| except RuntimeError: |
| pass |
| else: |
| self.fail("release_lock() without lock should raise " |
| "RuntimeError") |
| |
| class ImportTests(unittest.TestCase): |
| |
| def test_find_module_encoding(self): |
| fd = imp.find_module("heapq")[0] |
| self.assertEqual(fd.encoding, "iso-8859-1") |
| |
| def test_issue1267(self): |
| fp, filename, info = imp.find_module("pydoc") |
| self.assertNotEqual(fp, None) |
| self.assertEqual(fp.encoding, "iso-8859-1") |
| self.assertEqual(fp.tell(), 0) |
| self.assertEqual(fp.readline(), '#!/usr/bin/env python\n') |
| fp.close() |
| |
| fp, filename, info = imp.find_module("tokenize") |
| self.assertNotEqual(fp, None) |
| self.assertEqual(fp.encoding, "utf-8") |
| self.assertEqual(fp.tell(), 0) |
| self.assertEqual(fp.readline(), |
| '"""Tokenization help for Python programs.\n') |
| fp.close() |
| |
| def test_reload(self): |
| import marshal |
| imp.reload(marshal) |
| import string |
| imp.reload(string) |
| ## import sys |
| ## self.assertRaises(ImportError, reload, sys) |
| |
| |
| def test_main(): |
| support.run_unittest( |
| LockTests, |
| ImportTests, |
| ) |
| |
| if __name__ == "__main__": |
| test_main() |