Test suggestion box behavior before/after #466.

thomasballinger · thomasballinger · commit aa4ff4273736 · 2016-08-05T19:05:11.000-04:00
diff --git a/bpython/curtsiesfrontend/repl.py b/bpython/curtsiesfrontend/repl.py
@@ -1252,8 +1252,8 @@ def current_output_line(self, value):
         self.stdin.current_line = '\n'
 
     def paint(self, about_to_exit=False, user_quit=False,
-              try_preserve_history_height=40,
-              min_infobox_height=4):
+              try_preserve_history_height=30,
+              min_infobox_height=5):
         """Returns an array of min_height or more rows and width columns, plus
         cursor position
 
diff --git a/bpython/test/test_curtsies_painting.py b/bpython/test/test_curtsies_painting.py
@@ -696,7 +696,19 @@ def test_lower_on_screen(self):
                   '| k                |',
                   '| l                |',
                   '+------------------+']
-        self.assert_paint_ignoring_formatting(screen, (0, 8))
+        # behavior before issue #466
+        self.assert_paint_ignoring_formatting(
+                screen, try_preserve_history_height=0)
+        self.assert_paint_ignoring_formatting(
+                screen, min_infobox_height=100)
+        # behavior after issue #466
+        screen = ['>>> abc.',
+                  '+------------------+',
+                  '| aaaaaaaaaaaaaaaa |',
+                  '| b                |',
+                  '| c                |',
+                  '+------------------+']
+        self.assert_paint_ignoring_formatting(screen)
 
     def test_at_bottom_of_screen(self):
         self.repl.get_top_usable_line = lambda: 17  # two lines from bottom
@@ -720,4 +732,16 @@ def test_at_bottom_of_screen(self):
                   '| k                |',
                   '| l                |',
                   '+------------------+']
-        self.assert_paint_ignoring_formatting(screen, (0, 8))
+        # behavior before issue #466
+        self.assert_paint_ignoring_formatting(
+                screen, try_preserve_history_height=0)
+        self.assert_paint_ignoring_formatting(
+                screen, min_infobox_height=100)
+        # behavior after issue #466
+        screen = ['>>> abc.',
+                  '+------------------+',
+                  '| aaaaaaaaaaaaaaaa |',
+                  '| b                |',
+                  '| c                |',
+                  '+------------------+']
+        self.assert_paint_ignoring_formatting(screen)
