import unittest2 as unittest
from StringIO import StringIO

@@ -22,16 +22,25 @@
 nittest%0A
+try:%0A    
 from Str
@@ -61,8 +61,44 @@
 tringIO%0A
+except:%0A    from io import StringIO%0A

Fix test for py3