test_storemagic.py
71 lines
| 2.2 KiB
| text/x-python
|
PythonLexer
Thomas Kluyver
|
r11146 | import tempfile, os | ||
Nikita Kniazev
|
r27126 | from pathlib import Path | ||
Thomas Kluyver
|
r11146 | |||
Min RK
|
r21253 | from traitlets.config.loader import Config | ||
Thomas Kluyver
|
r11146 | |||
Matthias Bussonnier
|
r25073 | |||
def setup_module(): | ||||
M Bussonnier
|
r28954 | ip.run_line_magic("load_ext", "storemagic") | ||
Thomas Kluyver
|
r11146 | |||
def test_store_restore(): | ||||
M Bussonnier
|
r28954 | 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" | ||||
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" | ||||
ip.user_ns["foo"] = 78 | ||||
ip.run_line_magic("alias", 'bar echo "hello"') | ||||
ip.user_ns["foobar"] = 79 | ||||
ip.user_ns["foobaz"] = "80" | ||||
Thomas Kluyver
|
r11146 | tmpd = tempfile.mkdtemp() | ||
M Bussonnier
|
r28954 | ip.run_line_magic("cd", tmpd) | ||
ip.run_line_magic("store", "foo") | ||||
ip.run_line_magic("store", "bar") | ||||
ip.run_line_magic("store", "foobar foobaz") | ||||
Gökcen Eraslan
|
r25209 | |||
Thomas Kluyver
|
r11146 | # Check storing | ||
Samuel Gaist
|
r26911 | 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" | ||||
Gökcen Eraslan
|
r25209 | |||
Thomas Kluyver
|
r11146 | # Remove those items | ||
M Bussonnier
|
r28954 | ip.user_ns.pop("foo", None) | ||
ip.user_ns.pop("foobar", None) | ||||
ip.user_ns.pop("foobaz", None) | ||||
ip.alias_manager.undefine_alias("bar") | ||||
ip.run_line_magic("cd", "-") | ||||
ip.user_ns["_dh"][:] = [] | ||||
Gökcen Eraslan
|
r25209 | |||
Thomas Kluyver
|
r11146 | # Check restoring | ||
M Bussonnier
|
r28954 | ip.run_line_magic("store", "-r foo bar foobar foobaz") | ||
Samuel Gaist
|
r26911 | 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" | ||||
Gökcen Eraslan
|
r25209 | |||
M Bussonnier
|
r28954 | ip.run_line_magic("store", "-r") # restores _dh too | ||
Nikita Kniazev
|
r27126 | assert any(Path(tmpd).samefile(p) for p in ip.user_ns["_dh"]) | ||
Gökcen Eraslan
|
r25209 | |||
Thomas Kluyver
|
r11146 | os.rmdir(tmpd) | ||
Thomas Kluyver
|
r12333 | |||
def test_autorestore(): | ||||
M Bussonnier
|
r28954 | ip.user_ns["foo"] = 95 | ||
ip.run_line_magic("store", "foo") | ||||
del ip.user_ns["foo"] | ||||
Thomas Kluyver
|
r12333 | c = Config() | ||
c.StoreMagics.autorestore = False | ||||
orig_config = ip.config | ||||
try: | ||||
ip.config = c | ||||
Samuel Gaist
|
r26911 | ip.extension_manager.reload_extension("storemagic") | ||
assert "foo" not in ip.user_ns | ||||
Thomas Kluyver
|
r12333 | c.StoreMagics.autorestore = True | ||
Samuel Gaist
|
r26911 | ip.extension_manager.reload_extension("storemagic") | ||
assert ip.user_ns["foo"] == 95 | ||||
Thomas Kluyver
|
r12333 | finally: | ||
ip.config = orig_config | ||||