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

Fix Python 3 handling of urllib...

Fix Python 3 handling of urllib Added try/except blocks to properly import moved methods in the urllib and urllib2 modules. This would break things including `%load`ing scripts by url and installing a local copy of MathJax.

Thomas Kluyver - - Load All Authors

File last commit:

r12603:2ac0509e


                r13640:18bbdacf

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