test_prompts.py
110 lines
| 3.4 KiB
| text/x-python
|
PythonLexer
|
r7570 | # -*- coding: utf-8 | ||
|
r5658 | """Tests for prompt generation.""" | ||
import unittest | ||||
|
r7570 | import os | ||
|
r5658 | |||
from IPython.testing import tools as tt, decorators as dec | ||||
|
r7577 | from IPython.core.prompts import PromptManager, LazyEvaluate | ||
|
r5658 | from IPython.testing.globalipapp import get_ipython | ||
|
r7575 | from IPython.utils.tempdir import TemporaryDirectory | ||
|
r5658 | |||
ip = get_ipython() | ||||
class PromptTests(unittest.TestCase): | ||||
def setUp(self): | ||||
self.pm = PromptManager(shell=ip, config=ip.config) | ||||
def test_multiline_prompt(self): | ||||
self.pm.in_template = "[In]\n>>>" | ||||
self.pm.render('in') | ||||
self.assertEqual(self.pm.width, 3) | ||||
self.assertEqual(self.pm.txtwidth, 3) | ||||
self.pm.in_template = '[In]\n' | ||||
self.pm.render('in') | ||||
self.assertEqual(self.pm.width, 0) | ||||
self.assertEqual(self.pm.txtwidth, 0) | ||||
def test_translate_abbreviations(self): | ||||
def do_translate(template): | ||||
self.pm.in_template = template | ||||
return self.pm.templates['in'] | ||||
pairs = [(r'%n>', '{color.number}{count}{color.prompt}>'), | ||||
(r'\T', '{time}'), | ||||
(r'\n', '\n') | ||||
] | ||||
tt.check_pairs(do_translate, pairs) | ||||
|
r5724 | def test_user_ns(self): | ||
self.pm.color_scheme = 'NoColor' | ||||
ip.ex("foo='bar'") | ||||
self.pm.in_template = "In [{foo}]" | ||||
prompt = self.pm.render('in') | ||||
|
r7874 | self.assertEqual(prompt, u'In [bar]') | ||
|
r5724 | |||
def test_builtins(self): | ||||
self.pm.color_scheme = 'NoColor' | ||||
self.pm.in_template = "In [{int}]" | ||||
prompt = self.pm.render('in') | ||||
|
r7874 | self.assertEqual(prompt, u"In [%r]" % int) | ||
|
r5724 | |||
def test_undefined(self): | ||||
self.pm.color_scheme = 'NoColor' | ||||
self.pm.in_template = "In [{foo_dne}]" | ||||
prompt = self.pm.render('in') | ||||
|
r7874 | self.assertEqual(prompt, u"In [<ERROR: 'foo_dne' not found>]") | ||
|
r5724 | |||
|
r5658 | def test_render(self): | ||
self.pm.in_template = r'\#>' | ||||
self.assertEqual(self.pm.render('in',color=False), '%d>' % ip.execution_count) | ||||
|
r7570 | |||
|
r12205 | @dec.onlyif_unicode_paths | ||
|
r7575 | def test_render_unicode_cwd(self): | ||
|
r7570 | save = os.getcwdu() | ||
|
r7575 | with TemporaryDirectory(u'ünicødé') as td: | ||
os.chdir(td) | ||||
self.pm.in_template = r'\w [\#]' | ||||
|
r7570 | p = self.pm.render('in', color=False) | ||
|
r7874 | self.assertEqual(p, u"%s [%i]" % (os.getcwdu(), ip.execution_count)) | ||
|
r7575 | os.chdir(save) | ||
|
r7577 | |||
def test_lazy_eval_unicode(self): | ||||
u = u'ünicødé' | ||||
lz = LazyEvaluate(lambda : u) | ||||
|
r7581 | # str(lz) would fail | ||
|
r7874 | self.assertEqual(unicode(lz), u) | ||
self.assertEqual(format(lz), u) | ||||
|
r7577 | |||
def test_lazy_eval_nonascii_bytes(self): | ||||
u = u'ünicødé' | ||||
b = u.encode('utf8') | ||||
lz = LazyEvaluate(lambda : b) | ||||
|
r7581 | # unicode(lz) would fail | ||
|
r7874 | self.assertEqual(str(lz), str(b)) | ||
self.assertEqual(format(lz), str(b)) | ||||
|
r7579 | |||
def test_lazy_eval_float(self): | ||||
f = 0.503 | ||||
lz = LazyEvaluate(lambda : f) | ||||
|
r7874 | self.assertEqual(str(lz), str(f)) | ||
self.assertEqual(unicode(lz), unicode(f)) | ||||
self.assertEqual(format(lz), str(f)) | ||||
self.assertEqual(format(lz, '.1'), '0.5') | ||||
|
r7577 | |||
|
r7773 | @dec.skip_win32 | ||
|
r7586 | def test_cwd_x(self): | ||
self.pm.in_template = r"\X0" | ||||
save = os.getcwdu() | ||||
os.chdir(os.path.expanduser('~')) | ||||
p = self.pm.render('in', color=False) | ||||
try: | ||||
|
r7874 | self.assertEqual(p, '~') | ||
|
r7586 | finally: | ||
os.chdir(save) | ||||