test painting docstring

thomasballinger · thomasballinger · commit 343fddce5d20 · 2015-01-24T11:41:34.000-05:00
diff --git a/bpython/test/test_curtsies_painting.py b/bpython/test/test_curtsies_painting.py
@@ -89,10 +89,16 @@ def foo(x, y, z=10):
         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))
 
+    def test_formatted_docstring(self):
+        actual = replpainter.formatted_docstring(
+                'Returns the results\n\n' 'Also has side effects',
+                40, config=setup_config())
+        expected = fsarray(['Returns the results', '', 'Also has side effects'])
+        self.assertFSArraysEqualIgnoringFormatting(actual, expected)
+
+
 
 @contextmanager
 def output_to_repl(repl):
