test_formatters.py
30 lines
| 593 B
| text/x-python
|
PythonLexer
Robert Kern
|
r3209 | """Tests for the Formatters. | ||
| """ | ||||
| import nose.tools as nt | ||||
| from IPython.core.formatters import FormatterABC, DefaultFormatter | ||||
| class A(object): | ||||
| def __repr__(self): | ||||
| return 'A()' | ||||
| class B(A): | ||||
| def __repr__(self): | ||||
| return 'B()' | ||||
| def foo_printer(obj, pp, cycle): | ||||
| pp.text('foo') | ||||
| def test_pretty(): | ||||
| f = DefaultFormatter() | ||||
| f.for_type(A, foo_printer) | ||||
| nt.assert_equals(f(A()), 'foo') | ||||
| nt.assert_equals(f(B()), 'foo') | ||||
| f.pprint = False | ||||
| nt.assert_equals(f(A()), 'A()') | ||||
| nt.assert_equals(f(B()), 'B()') | ||||
| def test_deferred(): | ||||
| f = DefaultFormatter() | ||||