[Python-checkins] r70154 - python/branches/py3k/Lib/importlib/test/test_api.py

[brett.cannon]

Author: brett.cannon
Date: Wed Mar  4 02:02:54 2009
New Revision: 70154

Log:
Add a test for importlib.import_module.


Modified:
   python/branches/py3k/Lib/importlib/test/test_api.py

Modified: python/branches/py3k/Lib/importlib/test/test_api.py
==============================================================================
--- python/branches/py3k/Lib/importlib/test/test_api.py	(original)
+++ python/branches/py3k/Lib/importlib/test/test_api.py	Wed Mar  4 02:02:54 2009
@@ -26,7 +26,7 @@
                 module = importlib.import_module(name)
                 self.assertEqual(module.__name__, name)
 
-    def test_relative_package_import(self):
+    def test_shallow_relative_package_import(self):
         # Test importing a module from a package through a relatve import.
         pkg_name = 'pkg'
         pkg_long_name = '{0}.__init__'.format(pkg_name)
@@ -39,6 +39,15 @@
                 module = importlib.import_module(relative_name, pkg_name)
                 self.assertEqual(module.__name__, absolute_name)
 
+    def test_deep_relative_package_import(self):
+        modules = ['a.__init__', 'a.b.__init__', 'a.c']
+        with util.mock_modules(*modules) as mock:
+            with util.import_state(meta_path=[mock]):
+                importlib.import_module('a')
+                importlib.import_module('a.b')
+                module = importlib.import_module('..c', 'a.b')
+                self.assertEqual(module.__name__, 'a.c')
+
     def test_absolute_import_with_package(self):
         # Test importing a module from a package with an absolute name with
         # the 'package' argument given.