Attachment #321322 for bug #431388




    ip.run_line_magic('alias_magic', '--line env_alias env')
    nt.assert_equal(ip.run_line_magic('env', ''),
                    ip.run_line_magic('env_alias', ''))


def test_store():
    """Test %store."""
    ip = get_ipython()
    ip.run_line_magic('load_ext', 'storemagic')

    # make sure the storage is empty
    ip.run_line_magic('store', '-z')
    ip.user_ns['var'] = 42
    ip.run_line_magic('store', 'var')
    ip.user_ns['var'] = 39
    ip.run_line_magic('store', '-r')
    nt.assert_equal(ip.user_ns['var'], 42)

    ip.run_line_magic('store', '-d var')
    ip.user_ns['var'] = 39
    ip.run_line_magic('store' , '-r')
    nt.assert_equal(ip.user_ns['var'], 39)





        # run without arguments -> list variables & values
        elif not args:
            vars = db.keys('autorestore/*')
            vars.sort()
            if vars:
                size = max(map(len, vars))

            except KeyError:
                # it might be an alias
                # This needs to be refactored to use the new AliasManager stuff.
                if args[0] in ip.alias_manager:
                    name = args[0]
                    nargs, cmd = ip.alias_manager.alias_table[ name ]
                    staliases = db.get('stored_aliases',{})
                    staliases[ name ] = cmd
                    db['stored_aliases'] = staliases

                    """ % (args[0], obj) )
                    return
                #pickled = pickle.dumps(obj)
                db[ 'autorestore/' + args[0] ] = obj
                print "Stored '%s' (%s)" % (args[0], obj.__class__.__name__)


- 

Return to bug 431388

Lines 741-743 def test_alias_magic(): Link Here

(-)a/IPython/core/tests/test_magic.py (+19 lines)
741	ip.run_line_magic('alias_magic', '--line env_alias env')	741	ip.run_line_magic('alias_magic', '--line env_alias env')
742	nt.assert_equal(ip.run_line_magic('env', ''),	742	nt.assert_equal(ip.run_line_magic('env', ''),
743	ip.run_line_magic('env_alias', ''))	743	ip.run_line_magic('env_alias', ''))
		744
		745
		746	def test_store():
		747	"""Test %store."""
		748	ip = get_ipython()
		749	ip.run_line_magic('load_ext', 'storemagic')
		750
		751	# make sure the storage is empty
		752	ip.run_line_magic('store', '-z')
		753	ip.user_ns['var'] = 42
		754	ip.run_line_magic('store', 'var')
		755	ip.user_ns['var'] = 39
		756	ip.run_line_magic('store', '-r')
		757	nt.assert_equal(ip.user_ns['var'], 42)
		758
		759	ip.run_line_magic('store', '-d var')
		760	ip.user_ns['var'] = 39
		761	ip.run_line_magic('store' , '-r')
		762	nt.assert_equal(ip.user_ns['var'], 39)

Lines 140-146 def store(self, parameter_s=''): Link Here

(-)a/IPython/extensions/storemagic.py (-5 / +4 lines)
140		140
141	# run without arguments -> list variables & values	141	# run without arguments -> list variables & values
142	elif not args:	142	elif not args:
143	vars = self.db.keys('autorestore/*')	143	vars = db.keys('autorestore/*')
144	vars.sort()	144	vars.sort()
145	if vars:	145	if vars:
146	size = max(map(len, vars))	146	size = max(map(len, vars))
Lines 186-194 def store(self, parameter_s=''): Link Here
186	except KeyError:	186	except KeyError:
187	# it might be an alias	187	# it might be an alias
188	# This needs to be refactored to use the new AliasManager stuff.	188	# This needs to be refactored to use the new AliasManager stuff.
189	if args[0] in self.alias_manager:	189	if args[0] in ip.alias_manager:
190	name = args[0]	190	name = args[0]
191	nargs, cmd = self.alias_manager.alias_table[ name ]	191	nargs, cmd = ip.alias_manager.alias_table[ name ]
192	staliases = db.get('stored_aliases',{})	192	staliases = db.get('stored_aliases',{})
193	staliases[ name ] = cmd	193	staliases[ name ] = cmd
194	db['stored_aliases'] = staliases	194	db['stored_aliases'] = staliases
Lines 207-213 def store(self, parameter_s=''): Link Here
207	""" % (args[0], obj) )	207	""" % (args[0], obj) )
208	return	208	return
209	#pickled = pickle.dumps(obj)	209	#pickled = pickle.dumps(obj)
210	self.db[ 'autorestore/' + args[0] ] = obj	210	db[ 'autorestore/' + args[0] ] = obj
211	print "Stored '%s' (%s)" % (args[0], obj.__class__.__name__)	211	print "Stored '%s' (%s)" % (args[0], obj.__class__.__name__)
212		212
213		213
214	-