upstream/ipython Files · IPython/core/tests/test_prompts.py

Reset the interactive namespace __warningregistry__ before executing code...

Reset the interactive namespace __warningregistry__ before executing code Fixes . Idea: Right now, people often don't see important warnings when running code in IPython, because (to a first approximation) any given warning will only issue once per session. Blink and you'll miss it! This is a very common contributor to confused emails to numpy-discussion. E.g.: In [5]: 1 / my_array_with_random_contents /home/njs/.user-python2.7-64bit-3/bin/ipython:1: RuntimeWarning: divide by zero encountered in divide #!/home/njs/.user-python2.7-64bit-3/bin/python Out[5]: array([ 1.77073316, -2.29765021, -2.01800811, ..., 1.13871243, -1.08302964, -8.6185091 ]) Oo, right, guess I gotta be careful of those zeros -- thanks, numpy, for giving me that warning! A few days later: In [592]: 1 / some_other_array Out[592]: array([ 3.07735763, 0.50769289, 0.83984078, ..., -0.67563917, -0.85736257, -1.36511271]) Oops, it turns out that this array had a zero in it too, and that's going to bite me later. But no warning this time! The effect of this commit is to make it so that warnings triggered by the code in cell 5 do *not* suppress warnings triggered by the code in cell 592. Note that this only applies to warnings triggered *directly* by code entered interactively -- if somepkg.foo() calls anotherpkg.bad_func() which issues a warning, then this warning will still only be displayed once, even if multiple cells call somepkg.foo(). But if cell 5 and cell 592 both call anotherpkg.bad_func() directly, then both will get warnings. (Important exception: if foo() is defined *interactively*, and calls anotherpkg.bad_func(), then every cell that calls foo() will display the warning again. This is unavoidable without fixes to CPython upstream.) Explanation: Python's warning system has some weird quirks. By default, it tries to suppress duplicate warnings, where "duplicate" means the same warning message triggered twice by the same line of code. This requires determining which line of code is responsible for triggering a warning, and this is controlled by the stacklevel= argument to warnings.warn. Basically, though, the idea is that if foo() calls bar() which calls baz() which calls some_deprecated_api(), then baz() will get counted as being "responsible", and the warning system will make a note that the usage of some_deprecated_api() inside baz() has already been warned about and doesn't need to be warned about again. So far so good. To accomplish this, obviously, there has to be a record of somewhere which line this was. You might think that this would be done by recording the filename:linenumber pair in a dict inside the warnings module, or something like that. You would be wrong. What actually happens is that the warnings module will use stack introspection to reach into baz()'s execution environment, create a global (module-level) variable there named __warningregistry__, and then, inside this dictionary, record just the line number. Basically, it assumes that any given module contains only one line 1, only one line 2, etc., so storing the filename is irrelevant. Obviously for interactive code this is totally wrong -- all cells share the same execution environment and global namespace, and they all contain a new line 1. Currently the warnings module treats these as if they were all the same line. In fact they are not the same line; once we have executed a given chunk of code, we will never see those particular lines again. As soon as a given chunk of code finishes executing, its line number labels become meaningless, and the corresponding warning registry entries become meaningless as well. Therefore, with this patch we delete the __warningregistry__ each time we execute a new block of code.

Thomas Kluyver - - Load All Authors

File last commit:

r16767:90c8f7be


                r18548:61431d7d

Download file

             test_prompts.py
        
                    109 lines
            
             | 3.4 KiB
            
                | text/x-python
            
             |
                PythonLexer
            
             / IPython / core / tests / test_prompts.py
          
                    History
                
                 |
                  Source
                 | Raw
                 |Copy content
                 |Copy permalink

        MinRK
    
add failing test for unicode cwd in prompts

              r7570
            
      # -*- coding: utf-8

        Thomas Kluyver
    
Add tests for prompt system.

              r5658
            
      """Tests for prompt generation."""

      import unittest

        MinRK
    
add failing test for unicode cwd in prompts

              r7570
            
      import os

        Thomas Kluyver
    
Add tests for prompt system.

              r5658
            
      from IPython.testing import tools as tt, decorators as dec

        MinRK
    
test LazyEvaluate with non-ascii input

              r7577
            
      from IPython.core.prompts import PromptManager, LazyEvaluate

        Thomas Kluyver
    
Add tests for prompt system.

              r5658
            
      from IPython.testing.globalipapp import get_ipython

        Thomas Kluyver
    
Use TemporaryWorkingDirectory context manager in some tests

              r16767
            
      from IPython.utils.tempdir import TemporaryWorkingDirectory

        Thomas Kluyver
    
Python 3 compatibility for os.getcwdu()

              r13447
            
      from IPython.utils import py3compat

        Thomas Kluyver
    
Replace references to unicode and basestring

              r13353
            
      from IPython.utils.py3compat import unicode_type

        Thomas Kluyver
    
Add tests for prompt system.

              r5658
            
      ip = get_ipython()

      class PromptTests(unittest.TestCase):

          def setUp(self):

              self.pm = PromptManager(shell=ip, config=ip.config)

          def test_multiline_prompt(self):

              self.pm.in_template = "[In]\n>>>"

              self.pm.render('in')

              self.assertEqual(self.pm.width, 3)

              self.assertEqual(self.pm.txtwidth, 3)

              self.pm.in_template = '[In]\n'

              self.pm.render('in')

              self.assertEqual(self.pm.width, 0)

              self.assertEqual(self.pm.txtwidth, 0)

          def test_translate_abbreviations(self):

              def do_translate(template):

                  self.pm.in_template = template

                  return self.pm.templates['in']

              pairs = [(r'%n>', '{color.number}{count}{color.prompt}>'),

                       (r'\T', '{time}'),

                       (r'\n', '\n')

                      ]

              tt.check_pairs(do_translate, pairs)

        MinRK
    
allow access to user_ns in prompt_manager...

              r5724
            
          def test_user_ns(self):

              self.pm.color_scheme = 'NoColor'

              ip.ex("foo='bar'")

              self.pm.in_template = "In [{foo}]"

              prompt = self.pm.render('in')

        Bradley M. Froehle
    
s/assertEquals/assertEqual/

              r7874
            
              self.assertEqual(prompt, u'In [bar]')

        MinRK
    
allow access to user_ns in prompt_manager...

              r5724
            
          def test_builtins(self):

              self.pm.color_scheme = 'NoColor'

              self.pm.in_template = "In [{int}]"

              prompt = self.pm.render('in')

        Bradley M. Froehle
    
s/assertEquals/assertEqual/

              r7874
            
              self.assertEqual(prompt, u"In [%r]" % int)

        MinRK
    
allow access to user_ns in prompt_manager...

              r5724
            
          def test_undefined(self):

              self.pm.color_scheme = 'NoColor'

              self.pm.in_template = "In [{foo_dne}]"

              prompt = self.pm.render('in')

        Bradley M. Froehle
    
s/assertEquals/assertEqual/

              r7874
            
              self.assertEqual(prompt, u"In [<ERROR: 'foo_dne' not found>]")

        MinRK
    
allow access to user_ns in prompt_manager...

              r5724
            
        Thomas Kluyver
    
Add tests for prompt system.

              r5658
            
          def test_render(self):

              self.pm.in_template = r'\#>'

              self.assertEqual(self.pm.render('in',color=False), '%d>' % ip.execution_count)

        MinRK
    
add failing test for unicode cwd in prompts

              r7570
            
        Thomas Kluyver
    
Skip some more tests that require unicode paths

              r12168
            
          @dec.onlyif_unicode_paths

        MinRK
    
use TemporaryDirectory ctx manager in test_render_unicode_cwd

              r7575
            
          def test_render_unicode_cwd(self):

        Thomas Kluyver
    
Use TemporaryWorkingDirectory context manager in some tests

              r16767
            
              with TemporaryWorkingDirectory(u'ünicødé'):

        MinRK
    
use TemporaryDirectory ctx manager in test_render_unicode_cwd

              r7575
            
                  self.pm.in_template = r'\w [\#]'

        MinRK
    
add failing test for unicode cwd in prompts

              r7570
            
                  p = self.pm.render('in', color=False)

        Thomas Kluyver
    
Python 3 compatibility for os.getcwdu()

              r13447
            
                  self.assertEqual(p, u"%s [%i]" % (py3compat.getcwd(), ip.execution_count))

        MinRK
    
test LazyEvaluate with non-ascii input

              r7577
            
          def test_lazy_eval_unicode(self):

              u = u'ünicødé'

              lz = LazyEvaluate(lambda : u)

        MinRK
    
use cleaner, less safe, unicode/str in LazyEvaluate

              r7581
            
              # str(lz) would fail

        Thomas Kluyver
    
Replace references to unicode and basestring

              r13353
            
              self.assertEqual(unicode_type(lz), u)

        Bradley M. Froehle
    
s/assertEquals/assertEqual/

              r7874
            
              self.assertEqual(format(lz), u)

        MinRK
    
test LazyEvaluate with non-ascii input

              r7577
            
          def test_lazy_eval_nonascii_bytes(self):

              u = u'ünicødé'

              b = u.encode('utf8')

              lz = LazyEvaluate(lambda : b)

        MinRK
    
use cleaner, less safe, unicode/str in LazyEvaluate

              r7581
            
              # unicode(lz) would fail

        Bradley M. Froehle
    
s/assertEquals/assertEqual/

              r7874
            
              self.assertEqual(str(lz), str(b))

              self.assertEqual(format(lz), str(b))

        MinRK
    
test LazyEval with float and format string

              r7579
            
          def test_lazy_eval_float(self):

              f = 0.503

              lz = LazyEvaluate(lambda : f)

        Bradley M. Froehle
    
s/assertEquals/assertEqual/

              r7874
            
              self.assertEqual(str(lz), str(f))

        Thomas Kluyver
    
Replace references to unicode and basestring

              r13353
            
              self.assertEqual(unicode_type(lz), unicode_type(f))

        Bradley M. Froehle
    
s/assertEquals/assertEqual/

              r7874
            
              self.assertEqual(format(lz), str(f))

              self.assertEqual(format(lz, '.1'), '0.5')

        MinRK
    
test LazyEvaluate with non-ascii input

              r7577
            
        MinRK
    
skip test_cwdx on Windows...

              r7773
            
          @dec.skip_win32

        MinRK
    
fix&test missing HOME in cwd_x

              r7586
            
          def test_cwd_x(self):

              self.pm.in_template = r"\X0"

        Thomas Kluyver
    
Python 3 compatibility for os.getcwdu()

              r13447
            
              save = py3compat.getcwd()

        MinRK
    
fix&test missing HOME in cwd_x

              r7586
            
              os.chdir(os.path.expanduser('~'))

              p = self.pm.render('in', color=False)

              try:

        Bradley M. Froehle
    
s/assertEquals/assertEqual/

              r7874
            
                  self.assertEqual(p, '~')

        MinRK
    
fix&test missing HOME in cwd_x

              r7586
            
              finally:

                  os.chdir(save)

	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

MinRK add failing test for unicode cwd in prompts	r7570	# -*- coding: utf-8
Thomas Kluyver Add tests for prompt system.	r5658	"""Tests for prompt generation."""

		import unittest

MinRK add failing test for unicode cwd in prompts	r7570	import os
Thomas Kluyver Add tests for prompt system.	r5658
		from IPython.testing import tools as tt, decorators as dec
MinRK test LazyEvaluate with non-ascii input	r7577	from IPython.core.prompts import PromptManager, LazyEvaluate
Thomas Kluyver Add tests for prompt system.	r5658	from IPython.testing.globalipapp import get_ipython
Thomas Kluyver Use TemporaryWorkingDirectory context manager in some tests	r16767	from IPython.utils.tempdir import TemporaryWorkingDirectory
Thomas Kluyver Python 3 compatibility for os.getcwdu()	r13447	from IPython.utils import py3compat
Thomas Kluyver Replace references to unicode and basestring	r13353	from IPython.utils.py3compat import unicode_type
Thomas Kluyver Add tests for prompt system.	r5658
		ip = get_ipython()


		class PromptTests(unittest.TestCase):
		def setUp(self):
		self.pm = PromptManager(shell=ip, config=ip.config)

		def test_multiline_prompt(self):
		self.pm.in_template = "[In]\n>>>"
		self.pm.render('in')
		self.assertEqual(self.pm.width, 3)
		self.assertEqual(self.pm.txtwidth, 3)

		self.pm.in_template = '[In]\n'
		self.pm.render('in')
		self.assertEqual(self.pm.width, 0)
		self.assertEqual(self.pm.txtwidth, 0)

		def test_translate_abbreviations(self):
		def do_translate(template):
		self.pm.in_template = template
		return self.pm.templates['in']

		pairs = [(r'%n>', '{color.number}{count}{color.prompt}>'),
		(r'\T', '{time}'),
		(r'\n', '\n')
		]

		tt.check_pairs(do_translate, pairs)

MinRK allow access to user_ns in prompt_manager...	r5724	def test_user_ns(self):
		self.pm.color_scheme = 'NoColor'
		ip.ex("foo='bar'")
		self.pm.in_template = "In [{foo}]"
		prompt = self.pm.render('in')
Bradley M. Froehle s/assertEquals/assertEqual/	r7874	self.assertEqual(prompt, u'In [bar]')
MinRK allow access to user_ns in prompt_manager...	r5724
		def test_builtins(self):
		self.pm.color_scheme = 'NoColor'
		self.pm.in_template = "In [{int}]"
		prompt = self.pm.render('in')
Bradley M. Froehle s/assertEquals/assertEqual/	r7874	self.assertEqual(prompt, u"In [%r]" % int)
MinRK allow access to user_ns in prompt_manager...	r5724
		def test_undefined(self):
		self.pm.color_scheme = 'NoColor'
		self.pm.in_template = "In [{foo_dne}]"
		prompt = self.pm.render('in')
Bradley M. Froehle s/assertEquals/assertEqual/	r7874	self.assertEqual(prompt, u"In [<ERROR: 'foo_dne' not found>]")
MinRK allow access to user_ns in prompt_manager...	r5724
Thomas Kluyver Add tests for prompt system.	r5658	def test_render(self):
		self.pm.in_template = r'\#>'
		self.assertEqual(self.pm.render('in',color=False), '%d>' % ip.execution_count)
MinRK add failing test for unicode cwd in prompts	r7570
Thomas Kluyver Skip some more tests that require unicode paths	r12168	@dec.onlyif_unicode_paths
MinRK use TemporaryDirectory ctx manager in test_render_unicode_cwd	r7575	def test_render_unicode_cwd(self):
Thomas Kluyver Use TemporaryWorkingDirectory context manager in some tests	r16767	with TemporaryWorkingDirectory(u'ünicødé'):
MinRK use TemporaryDirectory ctx manager in test_render_unicode_cwd	r7575	self.pm.in_template = r'\w [\#]'
MinRK add failing test for unicode cwd in prompts	r7570	p = self.pm.render('in', color=False)
Thomas Kluyver Python 3 compatibility for os.getcwdu()	r13447	self.assertEqual(p, u"%s [%i]" % (py3compat.getcwd(), ip.execution_count))
MinRK test LazyEvaluate with non-ascii input	r7577
		def test_lazy_eval_unicode(self):
		u = u'ünicødé'
		lz = LazyEvaluate(lambda : u)
MinRK use cleaner, less safe, unicode/str in LazyEvaluate	r7581	# str(lz) would fail
Thomas Kluyver Replace references to unicode and basestring	r13353	self.assertEqual(unicode_type(lz), u)
Bradley M. Froehle s/assertEquals/assertEqual/	r7874	self.assertEqual(format(lz), u)
MinRK test LazyEvaluate with non-ascii input	r7577
		def test_lazy_eval_nonascii_bytes(self):
		u = u'ünicødé'
		b = u.encode('utf8')
		lz = LazyEvaluate(lambda : b)
MinRK use cleaner, less safe, unicode/str in LazyEvaluate	r7581	# unicode(lz) would fail
Bradley M. Froehle s/assertEquals/assertEqual/	r7874	self.assertEqual(str(lz), str(b))
		self.assertEqual(format(lz), str(b))
MinRK test LazyEval with float and format string	r7579
		def test_lazy_eval_float(self):
		f = 0.503
		lz = LazyEvaluate(lambda : f)

Bradley M. Froehle s/assertEquals/assertEqual/	r7874	self.assertEqual(str(lz), str(f))
Thomas Kluyver Replace references to unicode and basestring	r13353	self.assertEqual(unicode_type(lz), unicode_type(f))
Bradley M. Froehle s/assertEquals/assertEqual/	r7874	self.assertEqual(format(lz), str(f))
		self.assertEqual(format(lz, '.1'), '0.5')
MinRK test LazyEvaluate with non-ascii input	r7577
MinRK skip test_cwdx on Windows...	r7773	@dec.skip_win32
MinRK fix&test missing HOME in cwd_x	r7586	def test_cwd_x(self):
		self.pm.in_template = r"\X0"
Thomas Kluyver Python 3 compatibility for os.getcwdu()	r13447	save = py3compat.getcwd()
MinRK fix&test missing HOME in cwd_x	r7586	os.chdir(os.path.expanduser('~'))
		p = self.pm.render('in', color=False)
		try:
Bradley M. Froehle s/assertEquals/assertEqual/	r7874	self.assertEqual(p, '~')
MinRK fix&test missing HOME in cwd_x	r7586	finally:
		os.chdir(save)