Test for #19 ('import six' can be after its first use). refs #19

Signed-off-by: Daira Hopwood <[email protected]>

Author: daira

tests/test_fix_unicode.py

@@ -7,6 +7,7 @@
 b = U"\\u2041"
 c = ur'''blah
 foo'''
+import sys
 """
 
 UNICODE_LITERALS_six = """\
@@ -16,16 +17,26 @@
 b = six.u("\\u2041")
 c = six.u(r'''blah
 foo''')
+import sys
 """
 
-UNICODE_LITERALS_compat = UNICODE_LITERALS
+UNICODE_LITERALS_compat = """\
+from __future__ import absolute_import
+a = u''
+b = U"\\u2041"
+c = ur'''blah
+foo'''
+import sys
+"""
 
 UNICODE_LITERALS_future = """\
+from __future__ import absolute_import
 from __future__ import unicode_literals
 a = ''
 b = "\\u2041"
 c = r'''blah
 foo'''
+import sys
 """
 
 def test_unicode_six():