Show More
@@ -1,129 +1,129 b'' | |||
|
1 | 1 | # -*- coding: utf-8 |
|
2 | 2 | """Tests for prompt generation.""" |
|
3 | 3 | |
|
4 | 4 | import unittest |
|
5 | 5 | |
|
6 | 6 | import os |
|
7 | 7 | |
|
8 | 8 | from IPython.testing import tools as tt, decorators as dec |
|
9 | 9 | from IPython.core.prompts import PromptManager, LazyEvaluate, _invisible_characters |
|
10 | 10 | from IPython.testing.globalipapp import get_ipython |
|
11 | 11 | from IPython.utils.tempdir import TemporaryWorkingDirectory |
|
12 | 12 | from IPython.utils import py3compat |
|
13 | 13 | from IPython.utils.py3compat import unicode_type |
|
14 | 14 | |
|
15 | 15 | ip = get_ipython() |
|
16 | 16 | |
|
17 | 17 | |
|
18 | 18 | class PromptTests(unittest.TestCase): |
|
19 | 19 | def setUp(self): |
|
20 | 20 | self.pm = PromptManager(shell=ip, config=ip.config) |
|
21 | 21 | |
|
22 | 22 | def test_multiline_prompt(self): |
|
23 | 23 | self.pm.in_template = "[In]\n>>>" |
|
24 | 24 | self.pm.render('in') |
|
25 | 25 | self.assertEqual(self.pm.width, 3) |
|
26 | 26 | self.assertEqual(self.pm.txtwidth, 3) |
|
27 | 27 | |
|
28 | 28 | self.pm.in_template = '[In]\n' |
|
29 | 29 | self.pm.render('in') |
|
30 | 30 | self.assertEqual(self.pm.width, 0) |
|
31 | 31 | self.assertEqual(self.pm.txtwidth, 0) |
|
32 | 32 | |
|
33 | 33 | def test_translate_abbreviations(self): |
|
34 | 34 | def do_translate(template): |
|
35 | 35 | self.pm.in_template = template |
|
36 | 36 | return self.pm.templates['in'] |
|
37 | 37 | |
|
38 | 38 | pairs = [(r'%n>', '{color.number}{count}{color.prompt}>'), |
|
39 | 39 | (r'\T', '{time}'), |
|
40 | 40 | (r'\n', '\n') |
|
41 | 41 | ] |
|
42 | 42 | |
|
43 | 43 | tt.check_pairs(do_translate, pairs) |
|
44 | 44 | |
|
45 | 45 | def test_user_ns(self): |
|
46 | 46 | self.pm.color_scheme = 'NoColor' |
|
47 | 47 | ip.ex("foo='bar'") |
|
48 | 48 | self.pm.in_template = "In [{foo}]" |
|
49 | 49 | prompt = self.pm.render('in') |
|
50 | 50 | self.assertEqual(prompt, u'In [bar]') |
|
51 | 51 | |
|
52 | 52 | def test_builtins(self): |
|
53 | 53 | self.pm.color_scheme = 'NoColor' |
|
54 | 54 | self.pm.in_template = "In [{int}]" |
|
55 | 55 | prompt = self.pm.render('in') |
|
56 | 56 | self.assertEqual(prompt, u"In [%r]" % int) |
|
57 | 57 | |
|
58 | 58 | def test_undefined(self): |
|
59 | 59 | self.pm.color_scheme = 'NoColor' |
|
60 | 60 | self.pm.in_template = "In [{foo_dne}]" |
|
61 | 61 | prompt = self.pm.render('in') |
|
62 | 62 | self.assertEqual(prompt, u"In [<ERROR: 'foo_dne' not found>]") |
|
63 | 63 | |
|
64 | 64 | def test_render(self): |
|
65 | 65 | self.pm.in_template = r'\#>' |
|
66 | 66 | self.assertEqual(self.pm.render('in',color=False), '%d>' % ip.execution_count) |
|
67 | 67 | |
|
68 | 68 | @dec.onlyif_unicode_paths |
|
69 | 69 | def test_render_unicode_cwd(self): |
|
70 | 70 | with TemporaryWorkingDirectory(u'ΓΌnicΓΈdΓ©'): |
|
71 | 71 | self.pm.in_template = r'\w [\#]' |
|
72 | 72 | p = self.pm.render('in', color=False) |
|
73 | 73 | self.assertEqual(p, u"%s [%i]" % (py3compat.getcwd(), ip.execution_count)) |
|
74 | 74 | |
|
75 | 75 | def test_lazy_eval_unicode(self): |
|
76 | 76 | u = u'ΓΌnicΓΈdΓ©' |
|
77 | 77 | lz = LazyEvaluate(lambda : u) |
|
78 | 78 | # str(lz) would fail |
|
79 | 79 | self.assertEqual(unicode_type(lz), u) |
|
80 | 80 | self.assertEqual(format(lz), u) |
|
81 | 81 | |
|
82 | 82 | def test_lazy_eval_nonascii_bytes(self): |
|
83 | 83 | u = u'ΓΌnicΓΈdΓ©' |
|
84 | 84 | b = u.encode('utf8') |
|
85 | 85 | lz = LazyEvaluate(lambda : b) |
|
86 | 86 | # unicode(lz) would fail |
|
87 | 87 | self.assertEqual(str(lz), str(b)) |
|
88 | 88 | self.assertEqual(format(lz), str(b)) |
|
89 | 89 | |
|
90 | 90 | def test_lazy_eval_float(self): |
|
91 | 91 | f = 0.503 |
|
92 | 92 | lz = LazyEvaluate(lambda : f) |
|
93 | 93 | |
|
94 | 94 | self.assertEqual(str(lz), str(f)) |
|
95 | 95 | self.assertEqual(unicode_type(lz), unicode_type(f)) |
|
96 | 96 | self.assertEqual(format(lz), str(f)) |
|
97 | 97 | self.assertEqual(format(lz, '.1'), '0.5') |
|
98 | 98 | |
|
99 | 99 | @dec.skip_win32 |
|
100 | 100 | def test_cwd_x(self): |
|
101 | 101 | self.pm.in_template = r"\X0" |
|
102 | 102 | save = py3compat.getcwd() |
|
103 | 103 | os.chdir(os.path.expanduser('~')) |
|
104 | 104 | p = self.pm.render('in', color=False) |
|
105 | 105 | try: |
|
106 | 106 | self.assertEqual(p, '~') |
|
107 | 107 | finally: |
|
108 | 108 | os.chdir(save) |
|
109 | 109 | |
|
110 | 110 | def test_invisible_chars(self): |
|
111 | 111 | self.assertEqual(_invisible_characters('abc'), 0) |
|
112 | 112 | self.assertEqual(_invisible_characters('\001\033[1;37m\002'), 9) |
|
113 | 113 | # Sequences must be between \001 and \002 to be counted |
|
114 | 114 | self.assertEqual(_invisible_characters('\033[1;37m'), 0) |
|
115 | 115 | # Test custom escape sequences |
|
116 | 116 | self.assertEqual(_invisible_characters('\001\033]133;A\a\002'), 10) |
|
117 | 117 | |
|
118 | 118 | def test_width(self): |
|
119 |
default_in = '\x01\x1b]133;A\x07\x02In [ |
|
|
119 | default_in = '\x01\x1b]133;A\x07\x02In [1]: \x01\x1b]133;B\x07\x02' | |
|
120 | 120 | self.pm.in_template = default_in |
|
121 | 121 | self.pm.render('in') |
|
122 | 122 | self.assertEqual(self.pm.width, 8) |
|
123 | 123 | self.assertEqual(self.pm.txtwidth, 8) |
|
124 | 124 | |
|
125 | 125 | # Test custom escape sequences |
|
126 | 126 | self.pm.in_template = '\001\033]133;A\a\002' + default_in + '\001\033]133;B\a\002' |
|
127 | 127 | self.pm.render('in') |
|
128 | 128 | self.assertEqual(self.pm.width, 8) |
|
129 | 129 | self.assertEqual(self.pm.txtwidth, 8) |
General Comments 0
You need to be logged in to leave comments.
Login now