upstream/ipython Files · IPython/extensions/tests/test_storemagic.py

Catch interrupted poll() in terminal console...

Catch interrupted poll() in terminal console Alternative to my own PR - catch ZMQError in run_cell, and if it's caused by an interrupt, ignore it. more complex, especially if we want to handle the timeout nicely as proposed in the comments, but it's possibly also more convenient for other users of that API. Or perhaps not - I'm not sure what makes sense for other API consumers in this case. Fixes gh-8105

Thomas Kluyver - - Load All Authors

File last commit:

r12603:2ac0509e


                r20836:0b3b28de

Download file

             test_storemagic.py
        
                    50 lines
            
             | 1.3 KiB
            
                | text/x-python
            
             |
                PythonLexer
            
             / IPython / extensions / tests / test_storemagic.py
          
                    History
                
                 |
                  Annotation
                 | Raw
                 |Copy content
                 |Copy permalink

      import tempfile, os

      from IPython.config.loader import Config

      import nose.tools as nt

      ip = get_ipython()

      ip.magic('load_ext storemagic')

      def test_store_restore():

          ip.user_ns['foo'] = 78

          ip.magic('alias bar echo "hello"')

          tmpd = tempfile.mkdtemp()

          ip.magic('cd ' + tmpd)

          ip.magic('store foo')

          ip.magic('store bar')

          # Check storing

          nt.assert_equal(ip.db['autorestore/foo'], 78)

          nt.assert_in('bar', ip.db['stored_aliases'])

          # Remove those items

          ip.user_ns.pop('foo', None)

          ip.alias_manager.undefine_alias('bar')

          ip.magic('cd -')

          ip.user_ns['_dh'][:] = []

          # Check restoring

          ip.magic('store -r')

          nt.assert_equal(ip.user_ns['foo'], 78)

          assert ip.alias_manager.is_alias('bar')

          nt.assert_in(os.path.realpath(tmpd), ip.user_ns['_dh'])

          os.rmdir(tmpd)

      def test_autorestore():

          ip.user_ns['foo'] = 95

          ip.magic('store foo')

          del ip.user_ns['foo']

          c = Config()

          c.StoreMagics.autorestore = False

          orig_config = ip.config

          try:

              ip.config = c

              ip.extension_manager.reload_extension('storemagic')

              nt.assert_not_in('foo', ip.user_ns)

              c.StoreMagics.autorestore = True

              ip.extension_manager.reload_extension('storemagic')

              nt.assert_equal(ip.user_ns['foo'], 95)

          finally:

              ip.config = orig_config

	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

				import tempfile, os

				from IPython.config.loader import Config
				import nose.tools as nt

				ip = get_ipython()
				ip.magic('load_ext storemagic')

				def test_store_restore():
				ip.user_ns['foo'] = 78
				ip.magic('alias bar echo "hello"')
				tmpd = tempfile.mkdtemp()
				ip.magic('cd ' + tmpd)
				ip.magic('store foo')
				ip.magic('store bar')

				# Check storing
				nt.assert_equal(ip.db['autorestore/foo'], 78)
				nt.assert_in('bar', ip.db['stored_aliases'])

				# Remove those items
				ip.user_ns.pop('foo', None)
				ip.alias_manager.undefine_alias('bar')
				ip.magic('cd -')
				ip.user_ns['_dh'][:] = []

				# Check restoring
				ip.magic('store -r')
				nt.assert_equal(ip.user_ns['foo'], 78)
				assert ip.alias_manager.is_alias('bar')
				nt.assert_in(os.path.realpath(tmpd), ip.user_ns['_dh'])

				os.rmdir(tmpd)

				def test_autorestore():
				ip.user_ns['foo'] = 95
				ip.magic('store foo')
				del ip.user_ns['foo']
				c = Config()
				c.StoreMagics.autorestore = False
				orig_config = ip.config
				try:
				ip.config = c
				ip.extension_manager.reload_extension('storemagic')
				nt.assert_not_in('foo', ip.user_ns)
				c.StoreMagics.autorestore = True
				ip.extension_manager.reload_extension('storemagic')
				nt.assert_equal(ip.user_ns['foo'], 95)
				finally:
				ip.config = orig_config