upstream/ipython Files · IPython/utils/tests/test_io.py

Refuse to install event loop hooks when not using `prompt_toolkit` (#14132)...

Refuse to install event loop hooks when not using `prompt_toolkit` (#14132) Without this, `%gui` is effectively a no-op but the user thinks it works. For example. If running `ipython`: ``` In [1]: import matplotlib; matplotlib.use('QtAgg'); from matplotlib import pyplot; pyplot.ion(); pyplot.plot([1, 2, 3, 4]) Installed qt6 event loop hook. Out[1]: [<matplotlib.lines.Line2D at 0x1ba2f59d2a0>] ``` The window appears and responds as expected. If running `ipython --simple-prompt`, the user would see the same output, when in fact no event loop hook was installed since it's not supported without `prompt_toolkit`. The resulting Qt window is unresponsive because the event loop is not running, i.e. with `--simple-prompt`, Qt windows should block (but `pyplot` doesn't/can't know to do that) With this PR, the user will see: ``` In [1]: import matplotlib; matplotlib.use('QtAgg'); from matplotlib import pyplot; pyplot.ion(); pyplot.plot([1, 2, 3, 4]) Cannot install event loop hook for "qt" when running with `--simple-prompt`. NOTE: Tk is supported natively; use Tk apps and Tk backends with `--simple-prompt`. Out[1]: [<matplotlib.lines.Line2D at 0x170be0c0310>] ``` They'll still get an unresponsive Qt window, but they'll at least be told this can't work (while anything using Tk will work just fine).

Matthias Bussonnier - - Load All Authors

File last commit:

r27764:aefe51c6


                r28368:22fc5ab5

Download file

             test_io.py
        
                    60 lines
            
             | 1.4 KiB
            
                | text/x-python
            
             |
                PythonLexer
            
             / IPython / utils / tests / test_io.py
          
                    History
                
                 |
                  Annotation
                 | Raw
                 |Copy content
                 |Copy permalink

      # encoding: utf-8

      """Tests for io.py"""

      # Copyright (c) IPython Development Team.

      # Distributed under the terms of the Modified BSD License.

      import sys

      from io import StringIO

      import unittest

      from IPython.utils.io import Tee, capture_output

      def test_tee_simple():

          "Very simple check with stdout only"

          chan = StringIO()

          text = 'Hello'

          tee = Tee(chan, channel='stdout')

          print(text, file=chan)

          assert chan.getvalue() == text + "\n"

      class TeeTestCase(unittest.TestCase):

          def tchan(self, channel):

              trap = StringIO()

              chan = StringIO()

              text = 'Hello'

              std_ori = getattr(sys, channel)

              setattr(sys, channel, trap)

              tee = Tee(chan, channel=channel)

              print(text, end='', file=chan)

              trap_val = trap.getvalue()

              self.assertEqual(chan.getvalue(), text)

              tee.close()

              setattr(sys, channel, std_ori)

              assert getattr(sys, channel) == std_ori

          def test(self):

              for chan in ['stdout', 'stderr']:

                  self.tchan(chan)

      class TestIOStream(unittest.TestCase):

          def test_capture_output(self):

              """capture_output() context works"""

              with capture_output() as io:

                  print("hi, stdout")

                  print("hi, stderr", file=sys.stderr)

              self.assertEqual(io.stdout, "hi, stdout\n")

              self.assertEqual(io.stderr, "hi, stderr\n")

	Site-wide shortcuts
/	Use quick search box
g h	Goto home page
g g	Goto my private gists page
g G	Goto my public gists page
g 0-9	Goto bookmarked items from 0-9
n r	New repository page
n g	New gist page

	Repositories
g s	Goto summary page
g c	Goto changelog page
g f	Goto files page
g F	Goto files page with file search activated
g p	Goto pull requests page
g o	Goto repository settings
g O	Goto repository access permissions settings
t s	Toggle sidebar on some pages

				# encoding: utf-8
				"""Tests for io.py"""

				# Copyright (c) IPython Development Team.
				# Distributed under the terms of the Modified BSD License.


				import sys
				from io import StringIO

				import unittest

				from IPython.utils.io import Tee, capture_output


				def test_tee_simple():
				"Very simple check with stdout only"
				chan = StringIO()
				text = 'Hello'
				tee = Tee(chan, channel='stdout')
				print(text, file=chan)
				assert chan.getvalue() == text + "\n"


				class TeeTestCase(unittest.TestCase):

				def tchan(self, channel):
				trap = StringIO()
				chan = StringIO()
				text = 'Hello'

				std_ori = getattr(sys, channel)
				setattr(sys, channel, trap)

				tee = Tee(chan, channel=channel)

				print(text, end='', file=chan)
				trap_val = trap.getvalue()
				self.assertEqual(chan.getvalue(), text)

				tee.close()

				setattr(sys, channel, std_ori)
				assert getattr(sys, channel) == std_ori

				def test(self):
				for chan in ['stdout', 'stderr']:
				self.tchan(chan)

				class TestIOStream(unittest.TestCase):

				def test_capture_output(self):
				"""capture_output() context works"""

				with capture_output() as io:
				print("hi, stdout")
				print("hi, stderr", file=sys.stderr)

				self.assertEqual(io.stdout, "hi, stdout\n")
				self.assertEqual(io.stderr, "hi, stderr\n")