1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
|
import tempfile, os
from pathlib import Path
from traitlets.config.loader import Config
def setup_module():
ip.run_line_magic("load_ext", "storemagic")
def test_store_restore():
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"
tmpd = tempfile.mkdtemp()
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")
# Check storing
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)
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"][:] = []
# Check restoring
ip.run_line_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.run_line_magic("store", "-r") # restores _dh too
assert any(Path(tmpd).samefile(p) for p in ip.user_ns["_dh"])
os.rmdir(tmpd)
def test_autorestore():
ip.user_ns["foo"] = 95
ip.run_line_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")
assert "foo" not in ip.user_ns
c.StoreMagics.autorestore = True
ip.extension_manager.reload_extension("storemagic")
assert ip.user_ns["foo"] == 95
finally:
ip.config = orig_config
|
