Show More
@@ -1,109 +1,114 b'' | |||
|
1 | 1 | """Tests for the IPython tab-completion machinery. |
|
2 | 2 | """ |
|
3 | 3 | #----------------------------------------------------------------------------- |
|
4 | 4 | # Module imports |
|
5 | 5 | #----------------------------------------------------------------------------- |
|
6 | 6 | |
|
7 | 7 | # stdlib |
|
8 | 8 | import os |
|
9 | 9 | import sys |
|
10 | 10 | import unittest |
|
11 | 11 | |
|
12 | 12 | # third party |
|
13 | 13 | import nose.tools as nt |
|
14 | 14 | |
|
15 | 15 | # our own packages |
|
16 | 16 | from IPython.utils.tempdir import TemporaryDirectory |
|
17 | 17 | from IPython.core.history import HistoryManager, extract_hist_ranges |
|
18 | 18 | |
|
19 | 19 | def test_history(): |
|
20 | 20 | |
|
21 | 21 | ip = get_ipython() |
|
22 | 22 | with TemporaryDirectory() as tmpdir: |
|
23 | 23 | #tmpdir = '/software/temp' |
|
24 | 24 | histfile = os.path.realpath(os.path.join(tmpdir, 'history.sqlite')) |
|
25 | 25 | # Ensure that we restore the history management that we mess with in |
|
26 | 26 | # this test doesn't affect the IPython instance used by the test suite |
|
27 | 27 | # beyond this test. |
|
28 | 28 | hist_manager_ori = ip.history_manager |
|
29 | 29 | try: |
|
30 | 30 | ip.history_manager = HistoryManager(shell=ip) |
|
31 | 31 | ip.history_manager.hist_file = histfile |
|
32 | 32 | ip.history_manager.init_db() # Has to be called after changing file |
|
33 | 33 | ip.history_manager.reset() |
|
34 | 34 | print 'test',histfile |
|
35 | 35 | hist = ['a=1', 'def f():\n test = 1\n return test', 'b=2'] |
|
36 | 36 | for i, h in enumerate(hist, start=1): |
|
37 | 37 | ip.history_manager.store_inputs(i, h) |
|
38 | 38 | |
|
39 | 39 | ip.history_manager.db_log_output = True |
|
40 | 40 | # Doesn't match the input, but we'll just check it's stored. |
|
41 | 41 | ip.history_manager.output_hist_reprs[3].append("spam") |
|
42 | 42 | ip.history_manager.store_output(3) |
|
43 | 43 | |
|
44 | 44 | nt.assert_equal(ip.history_manager.input_hist_raw, [''] + hist) |
|
45 | 45 | |
|
46 | 46 | # Check lines were written to DB |
|
47 | 47 | c = ip.history_manager.db.execute("SELECT source_raw FROM history") |
|
48 | 48 | nt.assert_equal([x for x, in c], hist) |
|
49 | 49 | |
|
50 | 50 | # New session |
|
51 | 51 | ip.history_manager.reset() |
|
52 | 52 | newcmds = ["z=5","class X(object):\n pass", "k='p'"] |
|
53 | 53 | for i, cmd in enumerate(newcmds, start=1): |
|
54 | 54 | ip.history_manager.store_inputs(i, cmd) |
|
55 | 55 | gothist = ip.history_manager.get_range(start=1, stop=4) |
|
56 | 56 | nt.assert_equal(list(gothist), zip([0,0,0],[1,2,3], newcmds)) |
|
57 | 57 | # Previous session: |
|
58 | 58 | gothist = ip.history_manager.get_range(-1, 1, 4) |
|
59 | 59 | nt.assert_equal(list(gothist), zip([1,1,1],[1,2,3], hist)) |
|
60 | 60 | |
|
61 | 61 | # Check get_hist_tail |
|
62 | 62 | gothist = ip.history_manager.get_tail(4, output=True, |
|
63 | 63 | include_latest=True) |
|
64 | 64 | expected = [(1, 3, (hist[-1], ["spam"])), |
|
65 | 65 | (2, 1, (newcmds[0], None)), |
|
66 | 66 | (2, 2, (newcmds[1], None)), |
|
67 | 67 | (2, 3, (newcmds[2], None)),] |
|
68 | 68 | nt.assert_equal(list(gothist), expected) |
|
69 | 69 | |
|
70 | 70 | gothist = ip.history_manager.get_tail(2) |
|
71 | 71 | expected = [(2, 1, newcmds[0]), |
|
72 | 72 | (2, 2, newcmds[1])] |
|
73 | 73 | nt.assert_equal(list(gothist), expected) |
|
74 | 74 | |
|
75 | 75 | # Check get_hist_search |
|
76 | 76 | gothist = ip.history_manager.search("*test*") |
|
77 | 77 | nt.assert_equal(list(gothist), [(1,2,hist[1])] ) |
|
78 | 78 | gothist = ip.history_manager.search("b*", output=True) |
|
79 | 79 | nt.assert_equal(list(gothist), [(1,3,(hist[2],["spam"]))] ) |
|
80 | 80 | |
|
81 | 81 | # Cross testing: check that magic %save can get previous session. |
|
82 | 82 | testfilename = os.path.realpath(os.path.join(tmpdir, "test.py")) |
|
83 | 83 | ip.magic_save(testfilename + " ~1/1-3") |
|
84 | 84 | testfile = open(testfilename, "r") |
|
85 | 85 | nt.assert_equal(testfile.read(), "\n".join(hist)) |
|
86 | ||
|
87 | # Duplicate line numbers - check that it doesn't crash, and | |
|
88 | # gets a new session | |
|
89 | ip.history_manager.store_inputs(1, "rogue") | |
|
90 | nt.assert_equal(ip.history_manager.session_number, 3) | |
|
86 | 91 | finally: |
|
87 | 92 | # Restore history manager |
|
88 | 93 | ip.history_manager = hist_manager_ori |
|
89 | 94 | |
|
90 | 95 | def test_extract_hist_ranges(): |
|
91 | 96 | instr = "1 2/3 ~4/5-6 ~4/7-~4/9 ~9/2-~7/5" |
|
92 | 97 | expected = [(0, 1, 2), # 0 == current session |
|
93 | 98 | (2, 3, 4), |
|
94 | 99 | (-4, 5, 7), |
|
95 | 100 | (-4, 7, 10), |
|
96 | 101 | (-9, 2, None), # None == to end |
|
97 | 102 | (-8, 1, None), |
|
98 | 103 | (-7, 1, 6)] |
|
99 | 104 | actual = list(extract_hist_ranges(instr)) |
|
100 | 105 | nt.assert_equal(actual, expected) |
|
101 | 106 | |
|
102 | 107 | def test_magic_rerun(): |
|
103 | 108 | """Simple test for %rerun (no args -> rerun last line)""" |
|
104 | 109 | ip = get_ipython() |
|
105 | 110 | ip.run_cell("a = 10") |
|
106 | 111 | ip.run_cell("a += 1") |
|
107 | 112 | nt.assert_equal(ip.user_ns["a"], 11) |
|
108 | 113 | ip.run_cell("%rerun") |
|
109 | 114 | nt.assert_equal(ip.user_ns["a"], 12) |
General Comments 0
You need to be logged in to leave comments.
Login now