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 | latex_documents = [ ('index', 'ipython.tex', 'IPython Documentation', |
|
165 | latex_documents = [ ('index', 'ipython.tex', 'IPython Documentation', | |
166 | ur"""The IPython Development Team""", |
|
166 | ur"""The IPython Development Team""", | |
167 | 'manual'), |
|
167 | 'manual', True), | |
168 | ] |
|
168 | ] | |
169 |
|
169 | |||
170 | # The name of an image file (relative to this directory) to place at the top of |
|
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