diff options
Diffstat (limited to 'Lib/lib2to3/tests/test_util.py')
| -rw-r--r-- | Lib/lib2to3/tests/test_util.py | 13 |
1 files changed, 8 insertions, 5 deletions
diff --git a/Lib/lib2to3/tests/test_util.py b/Lib/lib2to3/tests/test_util.py index c6c6139..2fab8b9 100644 --- a/Lib/lib2to3/tests/test_util.py +++ b/Lib/lib2to3/tests/test_util.py @@ -3,6 +3,9 @@ # Testing imports from . import support +# Python imports +import os.path + # Local imports from lib2to3.pytree import Node, Leaf from lib2to3 import fixer_util @@ -578,14 +581,14 @@ class Test_find_indentation(support.TestCase): def test_nothing(self): fi = fixer_util.find_indentation node = parse("node()") - self.assertEqual(fi(node), "") + self.assertEqual(fi(node), u"") node = parse("") - self.assertEqual(fi(node), "") + self.assertEqual(fi(node), u"") def test_simple(self): fi = fixer_util.find_indentation node = parse("def f():\n x()") - self.assertEqual(fi(node), "") - self.assertEqual(fi(node.children[0].children[4].children[2]), " ") + self.assertEqual(fi(node), u"") + self.assertEqual(fi(node.children[0].children[4].children[2]), u" ") node = parse("def f():\n x()\n y()") - self.assertEqual(fi(node.children[0].children[4].children[4]), " ") + self.assertEqual(fi(node.children[0].children[4].children[4]), u" ") |