[Python-checkins] distutils2: Fix ImportError in test

[jason.coombs]

http://hg.python.org/distutils2/rev/9a84abc0e879
changeset:   1305:9a84abc0e879
user:        Jason R. Coombs <jaraco at jaraco.com>
date:        Tue Mar 13 17:01:26 2012 -0700
summary:
  Fix ImportError in test

files:
  distutils2/tests/test_msvc9compiler.py |  5 ++++-
  1 files changed, 4 insertions(+), 1 deletions(-)


diff --git a/distutils2/tests/test_msvc9compiler.py b/distutils2/tests/test_msvc9compiler.py
--- a/distutils2/tests/test_msvc9compiler.py
+++ b/distutils2/tests/test_msvc9compiler.py
@@ -101,7 +101,10 @@
         v = Reg.get_value(path, 'dragfullwindows')
         self.assertIn(v, ('0', '1', '2'))
 
-        import winreg
+        try:
+            import winreg
+        except ImportError:
+            import _winreg as winreg
         HKCU = winreg.HKEY_CURRENT_USER
         keys = Reg.read_keys(HKCU, 'xxxx')
         self.assertEqual(keys, None)

-- 
Repository URL: http://hg.python.org/distutils2