diff --git a/IPython/extensions/tests/test_storemagic.py b/IPython/extensions/tests/test_storemagic.py index 6f8371d..acf1abf 100644 --- a/IPython/extensions/tests/test_storemagic.py +++ b/IPython/extensions/tests/test_storemagic.py @@ -1,7 +1,6 @@ import tempfile, os from traitlets.config.loader import Config -import nose.tools as nt def setup_module(): @@ -23,10 +22,10 @@ def test_store_restore(): ip.magic('store foobar foobaz') # Check storing - nt.assert_equal(ip.db['autorestore/foo'], 78) - nt.assert_in('bar', ip.db['stored_aliases']) - nt.assert_equal(ip.db['autorestore/foobar'], 79) - nt.assert_equal(ip.db['autorestore/foobaz'], '80') + assert ip.db["autorestore/foo"] == 78 + assert "bar" in ip.db["stored_aliases"] + assert ip.db["autorestore/foobar"] == 79 + assert ip.db["autorestore/foobaz"] == "80" # Remove those items ip.user_ns.pop('foo', None) @@ -37,14 +36,14 @@ def test_store_restore(): ip.user_ns['_dh'][:] = [] # Check restoring - ip.magic('store -r foo bar foobar foobaz') - nt.assert_equal(ip.user_ns['foo'], 78) - assert ip.alias_manager.is_alias('bar') - nt.assert_equal(ip.user_ns['foobar'], 79) - nt.assert_equal(ip.user_ns['foobaz'], '80') + ip.magic("store -r foo bar foobar foobaz") + assert ip.user_ns["foo"] == 78 + assert ip.alias_manager.is_alias("bar") + assert ip.user_ns["foobar"] == 79 + assert ip.user_ns["foobaz"] == "80" - ip.magic('store -r') # restores _dh too - nt.assert_in(os.path.realpath(tmpd), ip.user_ns['_dh']) + ip.magic("store -r") # restores _dh too + assert os.path.realpath(tmpd) in ip.user_ns["_dh"] os.rmdir(tmpd) @@ -57,10 +56,10 @@ def test_autorestore(): orig_config = ip.config try: ip.config = c - ip.extension_manager.reload_extension('storemagic') - nt.assert_not_in('foo', ip.user_ns) + ip.extension_manager.reload_extension("storemagic") + assert "foo" not in ip.user_ns c.StoreMagics.autorestore = True - ip.extension_manager.reload_extension('storemagic') - nt.assert_equal(ip.user_ns['foo'], 95) + ip.extension_manager.reload_extension("storemagic") + assert ip.user_ns["foo"] == 95 finally: ip.config = orig_config