Show More
@@ -0,0 +1,56 b'' | |||
|
1 | #!/usr/bin/env python | |
|
2 | # encoding: utf-8 | |
|
3 | """ | |
|
4 | Simple tests for :mod:`IPython.extensions.pretty`. | |
|
5 | """ | |
|
6 | ||
|
7 | #----------------------------------------------------------------------------- | |
|
8 | # Copyright (C) 2008-2009 The IPython Development Team | |
|
9 | # | |
|
10 | # Distributed under the terms of the BSD License. The full license is in | |
|
11 | # the file COPYING, distributed as part of this software. | |
|
12 | #----------------------------------------------------------------------------- | |
|
13 | ||
|
14 | #----------------------------------------------------------------------------- | |
|
15 | # Imports | |
|
16 | #----------------------------------------------------------------------------- | |
|
17 | ||
|
18 | import sys | |
|
19 | from unittest import TestCase | |
|
20 | ||
|
21 | from IPython.core.component import Component, masquerade_as | |
|
22 | from IPython.core.iplib import InteractiveShell | |
|
23 | from IPython.extensions import pretty as pretty_ext | |
|
24 | from IPython.external import pretty | |
|
25 | ||
|
26 | from IPython.utils.traitlets import Bool | |
|
27 | ||
|
28 | #----------------------------------------------------------------------------- | |
|
29 | # Tests | |
|
30 | #----------------------------------------------------------------------------- | |
|
31 | ||
|
32 | ||
|
33 | class InteractiveShellStub(Component): | |
|
34 | pprint = Bool(True) | |
|
35 | ||
|
36 | class A(object): | |
|
37 | pass | |
|
38 | ||
|
39 | def a_pprinter(o, p, c): | |
|
40 | return p.text("<A>") | |
|
41 | ||
|
42 | class TestPrettyResultDisplay(TestCase): | |
|
43 | ||
|
44 | def setUp(self): | |
|
45 | self.ip = InteractiveShellStub(None) | |
|
46 | # This allows our stub to be retrieved instead of the real InteractiveShell | |
|
47 | masquerade_as(self.ip, InteractiveShell) | |
|
48 | self.prd = pretty_ext.PrettyResultDisplay(self.ip, name='pretty_result_display') | |
|
49 | ||
|
50 | def test_for_type(self): | |
|
51 | self.prd.for_type(A, a_pprinter) | |
|
52 | a = A() | |
|
53 | result = pretty.pretty(a) | |
|
54 | self.assertEquals(result, "<A>") | |
|
55 | ||
|
56 |
@@ -164,7 +164,7 b" latex_font_size = '11pt'" | |||
|
164 | 164 | |
|
165 | 165 | latex_documents = [ ('index', 'ipython.tex', 'IPython Documentation', |
|
166 | 166 | ur"""The IPython Development Team""", |
|
167 | 'manual'), | |
|
167 | 'manual', True), | |
|
168 | 168 | ] |
|
169 | 169 | |
|
170 | 170 | # The name of an image file (relative to this directory) to place at the top of |
General Comments 0
You need to be logged in to leave comments.
Login now