test argspec painting

thomasballinger · thomasballinger · commit 1eaac2c746a8 · 2015-01-24T00:13:12.000-05:00
diff --git a/bpython/test/test_curtsies_painting.py b/bpython/test/test_curtsies_painting.py
@@ -13,8 +13,9 @@
 from curtsies.fmtfuncs import cyan, bold, green, yellow, on_magenta, red
 from bpython.curtsiesfrontend.events import RefreshRequestEvent
 
-from bpython import config
+from bpython import config, inspection
 from bpython.curtsiesfrontend.repl import Repl
+from bpython.curtsiesfrontend import replpainter
 from bpython.repl import History
 from bpython.curtsiesfrontend.repl import INCONSISTENT_HISTORY_MSG, CONTIGUITY_BROKEN_MSG
 
@@ -79,6 +80,20 @@ def test_completion(self):
                   u'Welcome to bpython! Press <F1> f']
         self.assert_paint_ignoring_formatting(screen, (0, 4))
 
+    def test_argspec(self):
+        def foo(x, y, z=10):
+            "docstring!"
+            pass
+        argspec = inspection.getargspec('foo', foo) + [1]
+        array = replpainter.formatted_argspec(argspec, 30, setup_config())
+        screen = [(bold(cyan(u'foo'))+cyan(':')+cyan(' ')+cyan('(')+cyan('x') +
+                  yellow(',')+yellow(' ')+bold(cyan('y'))+yellow(',') +
+                  yellow(' ')+cyan('z')+yellow('=')+bold(cyan('10'))+yellow(')'))]
+        print screen
+        print array
+        self.assertFSArraysEqual(fsarray(array), fsarray(screen))
+
+
 @contextmanager
 def output_to_repl(repl):
     old_out, old_err = sys.stdout, sys.stderr
