Even more tests.

Trundle · Trundle · commit bf8833dd30db · 2010-11-10T07:51:40.000+01:00
diff --git a/bpython/test/test_repl.py b/bpython/test/test_repl.py
@@ -151,6 +151,20 @@ def setInputLine(self, line):
         """Set current input line of the test REPL."""
         self.repl.input_line = line
 
+    def test_func_name(self):
+        for (line, expected_name) in [("spam(", "spam"),
+                                      ("spam(map([]", "map"),
+                                      ("spam((), ", "spam")]:
+            self.setInputLine(line)
+            self.assertTrue(self.repl.get_args())
+            self.assertEqual(self.repl.current_func.__name__, expected_name)
+
+    def test_syntax_error_parens(self):
+        for line in ["spam(]", "spam([)", "spam())"]:
+            self.setInputLine(line)
+            # Should not explode
+            self.repl.get_args()
+
     def test_kw_arg_position(self):
         self.setInputLine("spam(a=0")
         self.assertTrue(self.repl.get_args())
@@ -171,6 +185,16 @@ def test_lambda_position(self):
         # Argument position
         self.assertEqual(self.repl.argspec[3], 1)
 
+    def test_name_in_assignment_without_spaces(self):
+        # Issue #127
+        self.setInputLine("x=range(")
+        self.assertTrue(self.repl.get_args())
+        self.assertEqual(self.repl.current_func.__name__, "range")
+
+    def test_nonexistent_name(self):
+        self.setInputLine("spamspamspam(")
+        self.assertFalse(self.repl.get_args())
+
 
 if __name__ == '__main__':
     unittest.main()
