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
                
                 |
                  Source
                 | Raw
                 |Copy content
                 |Copy permalink

        Thomas Kluyver
    
Add test for storemagic

              r11146
            
      import tempfile, os

        Thomas Kluyver
    
Add test for StoreMagics.autorestore option

              r12333
            
      from IPython.config.loader import Config

        Thomas Kluyver
    
Add test for storemagic

              r11146
            
      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)

        Thomas Kluyver
    
Fix storemagic test for new alias API

              r12603
            
          assert ip.alias_manager.is_alias('bar')

        Jens Hedegaard Nielsen
    
Store magic test. On MacOSX the temp dir in /var is symlinked into /private/var thus making this comparison fail. This is solved by using os.path.realpath to expand the tempdir into is's real directory.

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

        Thomas Kluyver
    
Add test for storemagic

              r11146
            
          os.rmdir(tmpd)

        Thomas Kluyver
    
Add test for StoreMagics.autorestore option

              r12333
            
      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

Thomas Kluyver Add test for storemagic	r11146	import tempfile, os

Thomas Kluyver Add test for StoreMagics.autorestore option	r12333	from IPython.config.loader import Config
Thomas Kluyver Add test for storemagic	r11146	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)
Thomas Kluyver Fix storemagic test for new alias API	r12603	assert ip.alias_manager.is_alias('bar')
Jens Hedegaard Nielsen Store magic test. On MacOSX the temp dir in /var is symlinked into /private/var thus making this comparison fail. This is solved by using os.path.realpath to expand the tempdir into is's real directory.	r11361	nt.assert_in(os.path.realpath(tmpd), ip.user_ns['_dh'])
Thomas Kluyver Add test for storemagic	r11146
		os.rmdir(tmpd)
Thomas Kluyver Add test for StoreMagics.autorestore option	r12333
		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