Fixed some of the compatiblity issues

1 files changed, 5 insertions, 5 deletions

diff --git a/mox_test.py b/mox_test.py
index 793f640..bed58b9 100755
--- a/mox_test.py
+++ b/mox_test.py
@@ -18,7 +18,7 @@
 #
 # This file was mofified by quermit@gmail.com
 
-import cStringIO
+import io
 import unittest
 import re
 
@@ -394,9 +394,9 @@ class IsATest(unittest.TestCase):
     self.failIf(isa_list == mixed_list)
 
   def testSpecialTypes(self):
-    """Verify that IsA can handle objects like cStringIO.StringIO."""
-    isA = mox.IsA(cStringIO.StringIO())
-    stringIO = cStringIO.StringIO()
+    """Verify that IsA can handle objects like io.StringIO."""
+    isA = mox.IsA(io.StringIO())
+    stringIO = io.StringIO()
     self.assert_(isA == stringIO)
 
 
@@ -1999,7 +1999,7 @@ class MoxTest(unittest.TestCase):
     # Forgot to replay!
     try:
       foo.GetBar().ShowMeTheMoney()
-    except AttributeError, e:
+    except AttributeError as e:
       self.assertEquals('MockMethod has no attribute "ShowMeTheMoney". '
           'Did you remember to put your mocks in replay mode?', str(e))