test_storemagic.py
66 lines
| 2.1 KiB
| text/x-python
|
PythonLexer
Thomas Kluyver
|
r11146 | import tempfile, os | ||
Min RK
|
r21253 | from traitlets.config.loader import Config | ||
Thomas Kluyver
|
r11146 | import nose.tools as nt | ||
Matthias Bussonnier
|
r25073 | |||
def setup_module(): | ||||
ip.magic('load_ext storemagic') | ||||
Thomas Kluyver
|
r11146 | |||
def test_store_restore(): | ||||
Matthias Bussonnier
|
r25073 | assert 'bar' not in ip.user_ns, "Error: some other test leaked `bar` in user_ns" | ||
assert 'foo' not in ip.user_ns, "Error: some other test leaked `foo` in user_ns" | ||||
Gökcen Eraslan
|
r25209 | assert 'foobar' not in ip.user_ns, "Error: some other test leaked `foobar` in user_ns" | ||
assert 'foobaz' not in ip.user_ns, "Error: some other test leaked `foobaz` in user_ns" | ||||
Thomas Kluyver
|
r11146 | ip.user_ns['foo'] = 78 | ||
ip.magic('alias bar echo "hello"') | ||||
Gökcen Eraslan
|
r25209 | ip.user_ns['foobar'] = 79 | ||
ip.user_ns['foobaz'] = '80' | ||||
Thomas Kluyver
|
r11146 | tmpd = tempfile.mkdtemp() | ||
ip.magic('cd ' + tmpd) | ||||
ip.magic('store foo') | ||||
ip.magic('store bar') | ||||
Gökcen Eraslan
|
r25209 | ip.magic('store foobar foobaz') | ||
Thomas Kluyver
|
r11146 | # Check storing | ||
nt.assert_equal(ip.db['autorestore/foo'], 78) | ||||
nt.assert_in('bar', ip.db['stored_aliases']) | ||||
Gökcen Eraslan
|
r25209 | nt.assert_equal(ip.db['autorestore/foobar'], 79) | ||
nt.assert_equal(ip.db['autorestore/foobaz'], '80') | ||||
Thomas Kluyver
|
r11146 | # Remove those items | ||
ip.user_ns.pop('foo', None) | ||||
Gökcen Eraslan
|
r25209 | ip.user_ns.pop('foobar', None) | ||
ip.user_ns.pop('foobaz', None) | ||||
Thomas Kluyver
|
r11146 | ip.alias_manager.undefine_alias('bar') | ||
ip.magic('cd -') | ||||
ip.user_ns['_dh'][:] = [] | ||||
Gökcen Eraslan
|
r25209 | |||
Thomas Kluyver
|
r11146 | # Check restoring | ||
Gökcen Eraslan
|
r25209 | ip.magic('store -r foo bar foobar foobaz') | ||
Thomas Kluyver
|
r11146 | nt.assert_equal(ip.user_ns['foo'], 78) | ||
Thomas Kluyver
|
r12603 | assert ip.alias_manager.is_alias('bar') | ||
Gökcen Eraslan
|
r25209 | nt.assert_equal(ip.user_ns['foobar'], 79) | ||
nt.assert_equal(ip.user_ns['foobaz'], '80') | ||||
ip.magic('store -r') # restores _dh too | ||||
Jens Hedegaard Nielsen
|
r11361 | nt.assert_in(os.path.realpath(tmpd), ip.user_ns['_dh']) | ||
Gökcen Eraslan
|
r25209 | |||
Thomas Kluyver
|
r11146 | os.rmdir(tmpd) | ||
Thomas Kluyver
|
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 | ||||