Update xmlrunner.py for Python 3.1 compat

rtyler · Feb 21, 2010 · e8f702e · e8f702e
1 parent 454db0f
commit e8f702e
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/tests/xmlrunner.py b/tests/xmlrunner.py
@@ -13,11 +13,12 @@
 import time
 import traceback
 import unittest
-from StringIO import StringIO
+try:
+    from io import StringIO
+except ImportError:
+    from StringIO import StringIO
 from xml.sax.saxutils import escape
 
-from StringIO import StringIO
-
 
 
 class _TestInfo(object):
@@ -171,7 +172,7 @@ def run(self, test):
             filename = "TEST-%s.xml" % classname
             if self._filename:
                 filename = self._filename
-            stream = file(os.path.join(self._path, filename), "w")
+            stream = open(os.path.join(self._path, filename), "w")
             stream.write('<?xml version="1.0" encoding="utf-8"?>\n')
         else:
             stream = self._stream