##// END OF EJS Templates
Add test code for history session offset.
Thomas Kluyver -
Show More
@@ -30,11 +30,11 b' def test_history():'
30 30 ip.history_manager = HistoryManager(ip)
31 31 ip.history_manager.hist_file = histfile
32 32 print 'test',histfile
33 hist = ['a=1\n', 'def f():\n test = 1\n return test\n', 'b=2\n']
33 hist = ['a=1', 'def f():\n test = 1\n return test', 'b=2']
34 34 # test save and load
35 35 ip.history_manager.input_hist_raw[:] = []
36 36 for h in hist:
37 ip.history_manager.input_hist_raw.append(h)
37 ip.history_manager.store_inputs(h)
38 38 ip.save_history()
39 39 ip.history_manager.input_hist_raw[:] = []
40 40 ip.reload_history()
@@ -43,6 +43,23 b' def test_history():'
43 43 nt.assert_equal(len(ip.history_manager.input_hist_raw), len(hist))
44 44 for i,h in enumerate(hist):
45 45 nt.assert_equal(hist[i], ip.history_manager.input_hist_raw[i])
46
47 # Test that session offset works.
48 ip.history_manager.session_offset = \
49 len(ip.history_manager.input_hist_raw) -1
50 newcmds = ["z=5","class X(object):\n pass", "k='p'"]
51 for cmd in newcmds:
52 ip.history_manager.store_inputs(cmd)
53 gothist = ip.history_manager.get_history((1,4),
54 raw=True, output=False)
55 nt.assert_equal(gothist, dict(zip([1,2,3], newcmds)))
56
57 # Cross testing: check that magic %save picks up on the session
58 # offset.
59 testfilename = os.path.realpath(os.path.join(tmpdir, "test.py"))
60 ip.magic_save(testfilename + " 1-3")
61 testfile = open(testfilename, "r")
62 nt.assert_equal(testfile.read(), "\n".join(newcmds))
46 63 finally:
47 64 # Restore history manager
48 65 ip.history_manager = hist_manager_ori
General Comments 0
You need to be logged in to leave comments. Login now