Replace PySys_WriteStdout with PySys_FormatStdout to ensure no trunca…

…tion for long lines. Fixes numba#3811. Signed-off-by: HE, Tao <sighingnow@gmail.com>
sighingnow · Mar 3, 2019 · b01a590 · b01a590
1 parent ae88a6f
commit b01a590
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/numba/pythonapi.py b/numba/pythonapi.py
@@ -1213,7 +1213,7 @@ def borrow_none(self):
 
     def sys_write_stdout(self, fmt, *args):
         fnty = Type.function(Type.void(), [self.cstring], var_arg=True)
-        fn = self._get_function(fnty, name="PySys_WriteStdout")
+        fn = self._get_function(fnty, name="PySys_FormatStdout")
         return self.builder.call(fn, (fmt,) + args)
 
     def object_dump(self, obj):

diff --git a/numba/tests/test_print.py b/numba/tests/test_print.py
@@ -189,5 +189,15 @@ def print_kwarg():
                     "keyword arguments.")
         self.assertIn(raises.exception.msg, expected)
 
+    def test_print_no_truncation(self):
+        ''' See: https://github.com/numba/numba/issues/3811
+        '''
+        @jit(nopython=True)
+        def foo():
+            print(''.join(['a'] * 10000))
+        with captured_stdout():
+            foo()
+            self.assertEqual(sys.stdout.getvalue(), ''.join(['a'] * 10000) + '\n')
+
 if __name__ == '__main__':
     unittest.main()