test_text.py
170 lines
| 5.5 KiB
| text/x-python
|
PythonLexer
|
r4538 | # encoding: utf-8 | ||
"""Tests for IPython.utils.text""" | ||||
#----------------------------------------------------------------------------- | ||||
# Copyright (C) 2011 The IPython Development Team | ||||
# | ||||
# Distributed under the terms of the BSD License. The full license is in | ||||
# the file COPYING, distributed as part of this software. | ||||
#----------------------------------------------------------------------------- | ||||
#----------------------------------------------------------------------------- | ||||
# Imports | ||||
#----------------------------------------------------------------------------- | ||||
import os | ||||
|
r4654 | import math | ||
|
r7387 | import random | ||
|
r4538 | |||
import nose.tools as nt | ||||
from nose import with_setup | ||||
from IPython.testing import decorators as dec | ||||
from IPython.utils import text | ||||
#----------------------------------------------------------------------------- | ||||
# Globals | ||||
#----------------------------------------------------------------------------- | ||||
def test_columnize(): | ||||
|
r4539 | """Basic columnize tests.""" | ||
size = 5 | ||||
items = [l*size for l in 'abc'] | ||||
out = text.columnize(items, displaywidth=80) | ||||
|
r7875 | nt.assert_equal(out, 'aaaaa bbbbb ccccc\n') | ||
|
r7387 | out = text.columnize(items, displaywidth=12) | ||
|
r7875 | nt.assert_equal(out, 'aaaaa ccccc\nbbbbb\n') | ||
|
r7387 | out = text.columnize(items, displaywidth=10) | ||
|
r7875 | nt.assert_equal(out, 'aaaaa\nbbbbb\nccccc\n') | ||
|
r7387 | |||
def test_columnize_random(): | ||||
"""Test with random input to hopfully catch edge case """ | ||||
for nitems in [random.randint(2,70) for i in range(2,20)]: | ||||
displaywidth = random.randint(20,200) | ||||
rand_len = [random.randint(2,displaywidth) for i in range(nitems)] | ||||
items = ['x'*l for l in rand_len] | ||||
out = text.columnize(items, displaywidth=displaywidth) | ||||
longer_line = max([len(x) for x in out.split('\n')]) | ||||
longer_element = max(rand_len) | ||||
if longer_line > displaywidth: | ||||
print "Columnize displayed something lager than displaywidth : %s " % longer_line | ||||
print "longer element : %s " % longer_element | ||||
print "displaywidth : %s " % displaywidth | ||||
print "number of element : %s " % nitems | ||||
print "size of each element :\n %s" % rand_len | ||||
assert False | ||||
|
r4539 | |||
|
r7387 | def test_columnize_medium(): | ||
"""Test with inputs than shouldn't be wider tahn 80 """ | ||||
size = 40 | ||||
items = [l*size for l in 'abc'] | ||||
out = text.columnize(items, displaywidth=80) | ||||
|
r7875 | nt.assert_equal(out, '\n'.join(items+[''])) | ||
|
r4539 | |||
def test_columnize_long(): | ||||
"""Test columnize with inputs longer than the display window""" | ||||
size = 11 | ||||
items = [l*size for l in 'abc'] | ||||
out = text.columnize(items, displaywidth=size-1) | ||||
|
r7875 | nt.assert_equal(out, '\n'.join(items+[''])) | ||
|
r4654 | |||
|
r5354 | def eval_formatter_check(f): | ||
|
r5355 | ns = dict(n=12, pi=math.pi, stuff='hello there', os=os, u=u"café", b="café") | ||
|
r4891 | s = f.format("{n} {n//4} {stuff.split()[0]}", **ns) | ||
|
r7875 | nt.assert_equal(s, "12 3 hello") | ||
|
r4654 | s = f.format(' '.join(['{n//%i}'%i for i in range(1,8)]), **ns) | ||
|
r7875 | nt.assert_equal(s, "12 6 4 3 2 2 1") | ||
|
r4654 | s = f.format('{[n//i for i in range(1,8)]}', **ns) | ||
|
r7875 | nt.assert_equal(s, "[12, 6, 4, 3, 2, 2, 1]") | ||
|
r4654 | s = f.format("{stuff!s}", **ns) | ||
|
r7875 | nt.assert_equal(s, ns['stuff']) | ||
|
r4654 | s = f.format("{stuff!r}", **ns) | ||
|
r7875 | nt.assert_equal(s, repr(ns['stuff'])) | ||
|
r4654 | |||
|
r5355 | # Check with unicode: | ||
s = f.format("{u}", **ns) | ||||
|
r7875 | nt.assert_equal(s, ns['u']) | ||
|
r5355 | # This decodes in a platform dependent manner, but it shouldn't error out | ||
s = f.format("{b}", **ns) | ||||
|
r4654 | nt.assert_raises(NameError, f.format, '{dne}', **ns) | ||
|
r5354 | def eval_formatter_slicing_check(f): | ||
|
r4654 | ns = dict(n=12, pi=math.pi, stuff='hello there', os=os) | ||
s = f.format(" {stuff.split()[:]} ", **ns) | ||||
|
r7875 | nt.assert_equal(s, " ['hello', 'there'] ") | ||
|
r4654 | s = f.format(" {stuff.split()[::-1]} ", **ns) | ||
|
r7875 | nt.assert_equal(s, " ['there', 'hello'] ") | ||
|
r4654 | s = f.format("{stuff[::2]}", **ns) | ||
|
r7875 | nt.assert_equal(s, ns['stuff'][::2]) | ||
|
r4654 | |||
nt.assert_raises(SyntaxError, f.format, "{n:x}", **ns) | ||||
|
r5354 | def eval_formatter_no_slicing_check(f): | ||
|
r4654 | ns = dict(n=12, pi=math.pi, stuff='hello there', os=os) | ||
s = f.format('{n:x} {pi**2:+f}', **ns) | ||||
|
r7875 | nt.assert_equal(s, "c +9.869604") | ||
|
r4654 | |||
|
r5501 | s = f.format('{stuff[slice(1,4)]}', **ns) | ||
|
r7875 | nt.assert_equal(s, 'ell') | ||
|
r5501 | |||
|
r4654 | nt.assert_raises(SyntaxError, f.format, "{a[:]}") | ||
|
r5354 | def test_eval_formatter(): | ||
f = text.EvalFormatter() | ||||
eval_formatter_check(f) | ||||
eval_formatter_no_slicing_check(f) | ||||
def test_full_eval_formatter(): | ||||
f = text.FullEvalFormatter() | ||||
eval_formatter_check(f) | ||||
eval_formatter_slicing_check(f) | ||||
def test_dollar_formatter(): | ||||
f = text.DollarFormatter() | ||||
eval_formatter_check(f) | ||||
eval_formatter_slicing_check(f) | ||||
ns = dict(n=12, pi=math.pi, stuff='hello there', os=os) | ||||
s = f.format("$n", **ns) | ||||
|
r7875 | nt.assert_equal(s, "12") | ||
|
r5354 | s = f.format("$n.real", **ns) | ||
|
r7875 | nt.assert_equal(s, "12") | ||
|
r5354 | s = f.format("$n/{stuff[:5]}", **ns) | ||
|
r7875 | nt.assert_equal(s, "12/hello") | ||
|
r5365 | s = f.format("$n $$HOME", **ns) | ||
|
r7875 | nt.assert_equal(s, "12 $HOME") | ||
|
r5375 | s = f.format("${foo}", foo="HOME") | ||
|
r7875 | nt.assert_equal(s, "$HOME") | ||
|
r7713 | |||
|
r7715 | def test_long_substr(): | ||
data = ['hi'] | ||||
|
r7875 | nt.assert_equal(text.long_substr(data), 'hi') | ||
|
r7715 | |||
def test_long_substr2(): | ||||
data = ['abc', 'abd', 'abf', 'ab'] | ||||
|
r7875 | nt.assert_equal(text.long_substr(data), 'ab') | ||
|
r7715 | |||
|
r7713 | def test_strip_email(): | ||
src = """\ | ||||
>> >>> def f(x): | ||||
>> ... return x+1 | ||||
|
r7714 | >> ... | ||
|
r7713 | >> >>> zz = f(2.5)""" | ||
cln = """\ | ||||
>>> def f(x): | ||||
... return x+1 | ||||
|
r7714 | ... | ||
|
r7713 | >>> zz = f(2.5)""" | ||
|
r7875 | nt.assert_equal(text.strip_email_quotes(src), cln) | ||
|
r7715 | |||
def test_strip_email2(): | ||||
src = '> > > list()' | ||||
cln = 'list()' | ||||
|
r7875 | nt.assert_equal(text.strip_email_quotes(src), cln) | ||