ast.literal_eval can properly evaluate complex numbers now. This fixes issue4907.

1 files changed, 11 insertions, 0 deletions

diff --git a/Lib/test/test_ast.py b/Lib/test/test_ast.py
index 00a5aae593..994e10be7d 100644
--- a/Lib/test/test_ast.py
+++ b/Lib/test/test_ast.py
@@ -271,6 +271,17 @@ class ASTHelpers_Test(unittest.TestCase):
         self.assertEqual(ast.literal_eval('(True, False, None)'), (True, False, None))
         self.assertRaises(ValueError, ast.literal_eval, 'foo()')
 
+    def test_literal_eval_issue4907(self):
+        self.assertEqual(ast.literal_eval('2j'), 2j)
+        self.assertEqual(ast.literal_eval('10 + 2j'), 10 + 2j)
+        self.assertEqual(ast.literal_eval('1.5 - 2j'), 1.5 - 2j)
+        try:
+            ast.literal_eval('2 + (3 + 4j)')
+        except ValueError:
+            pass
+        else:
+            self.fail('expected value error')
+
 
 def test_main():
     test_support.run_unittest(AST_Tests, ASTHelpers_Test)