Format: application/x-python-code
1 o
2
3 ��h�9 �
4 @ s� d Z ddlmZmZmZmZmZ ddlZddlZddl Z ddl
5 Z
6 ddl Z ddl Z ddlZddl
7 mZ ddlmZ G dd� de�ZG dd � d e�ZG d
8 d � d e�ZG d d
9 � d
10 e�ZG dd� de�ZG dd� de�Zde_G dd� de�ZG dd� de�ZG dd� de�Z dd� Z dd� Z d d � Z ej �!d e"e#� e$� e#d g�e$d g�e#d d g�e$d d g�e#g d!��gg d"���d#d$� �Z%d%d&� Z&d'd(� Z'd)d*� Z(d+d,� Z)d-d.� Z*G d/d0� d0e+�Z,G d1d2� d2e�Z-d3d4� Z.G d5d6� d6e�Z/G d7d8� d8e/�Z0G d9d:� d:e j1�Z2G d;d<� d<e3�Z4e4d=�Z5d>d?� Z6d@dA� Z7dBdC� Z8dDdE� Z9dFdG� Z:dHdI� Z;dJdK� Z<dLdM� Z=dNdO� Z>dPdQ� Z?dRdS� Z@dTdU� ZAG dVdW� dWee�ZBG dXdY� dYe#�ZCdZd[� ZDdS )\z Tests for IPython.lib.pretty.� )�Counter� defaultdict�deque� OrderedDict�UserListN��pretty)�StringIOc @ � e Zd Zdd� Zdd� ZdS )�MyListc C s
11 || _ d S �N)�content)�selfr
12 � r �Sc:\Users\shubh\Videos\agent\nova\Lib\site-packages\IPython/lib/tests/test_pretty.py�__init__ �
13 zMyList.__init__c C s� |r |� d� d S |�ddd��) t| j�D ]\}}|r&|� d� |�� n|�d� |�|� qW d � d S 1 s<w Y d S )Nz MyList(...)� zMyList(�)�,� )�text�group� enumerater
14 Z breakabler )r �p�cycle�i�childr r r �
15 _repr_pretty_ s
16
17
18 �"�zMyList._repr_pretty_N)�__name__�
19 __module__� __qualname__r r r r r r r s r c @ � e Zd Zdd� ZdS )�MyDictc C � |� d� d S )N� MyDict(...)�r �r r r r r r r ) � zMyDict._repr_pretty_N�r r r! r r r r r r# ( � r# c @ r" )�MyObjc C s d S r r �r r r r �
20 somemethod- � �MyObj.somemethodN)r r r! r- r r r r r+ , r* r+ c @ r" )�Dummy1c C r$ )N� Dummy1(...)r& r' r r r r 2 r( zDummy1._repr_pretty_Nr) r r r r r0 1 r* r0 c @ s e Zd ZdZdS )�Dummy2Nr) r r r r r2 5 s r2 c @ � e Zd ZdS )�NoModuleN�r r r! r r r r r4 8 � r4 c @ r" )�Breakingc C sP |� ddd�� |�d� |�� |�d� W d � d S 1 s!w Y d S )N� zTG: �:z Breaking(r )r r Zbreak_r' r r r r > s
21
22 "�zBreaking._repr_pretty_Nr) r r r r r7 = r* r7 c @ r" )� BreakingReprc C � dS )Nz Breaking(
23 )r r, r r r �__repr__E r. zBreakingRepr.__repr__N�r r r! r< r r r r r: D r* r: c @ r" )�BadReprc C s dd S )N� r r r, r r r r< I s zBadRepr.__repr__Nr= r r r r r> H r* r> c C sF d} t � tt| ���}dd�dd� t| �D �� d }||ks!J �dS )z"Test correct indentation in groups�( zMyList(
24 z,
25 c s s � | ]}d | V qdS )z %dNr ��.0r r r r � <genexpr>Q � � z#test_indentation.<locals>.<genexpr>r N)r r �range�join)�count� gotoutput�expectedoutputr r r �test_indentationM s rJ c C � t � t� �} d}| |ksJ �dS )z�
26 Test correct dispatching: The _repr_pretty_ method for MyDict
27 must be found before the registered printer for dict.
28 r% N)r r# �rH rI r r r �
29 test_dispatchV � rM c C rK )z^
30 Test that the _repr_pretty_ method is tested for callability and skipped if
31 not.
32 r1 N)r r2 rL r r r �test_callability_checkinga rN rO zobj,expected_outputr? � )���������������)zset()z frozenset()z{1}zfrozenset({1})z{1, 2}zfrozenset({1, 2})z {-3, -2, -1}c C s t � | �}||ks J �dS )z>
33 Test that set and frozenset use Python 3 formatting.
34 Nr )�obj�expected_outputZ
35 got_outputr r r � test_setsl s
36 rV c C s< d} t jdk r dnd}t�| �}t�|j�}||ks J �dS )z:
37 Test that pprint works for heap allocated types.
38 Z xxlimited_35)r �
39 � zxxlimited.Nullzxxlimited_35.NullN)�sys� version_info�pytestZ importorskipr ZNull)� module_namerU Z xxlimited�outputr r r � test_pprint_heap_allocated_type� s �
40 r^ c C s t � t�} | dks J �dS )z@
41 Test that pprint works for classes with no __module__.
42 r4 N)r r4 �r] r r r �test_pprint_nomod� s
43 r` c C rK )z5
44 Test that p.break_ produces expected output
45 zTG: Breaking(
46 ):N)r r7 �r] �expectedr r r �test_pprint_break� s rc c C sH t � t� gg�} d}| |ksJ �t � t� gd g�} d}| |ks"J �dS )z,
47 Test that p.break_ is used in repr
48 z[[Breaking(
49 )]]rP z"[[Breaking(
50 ),
51 Breaking(
52 )]]N)r r: ra r r r �test_pprint_break_repr� s rd c C s< t �t�� t�t� � W d � dS 1 sw Y dS )zDon't catch bad repr errorsN)r[ �raises�ZeroDivisionErrorr r> r r r r �
53 test_bad_repr� s "�rg c @ r" )� BadExceptionc C r; )NrQ r r, r r r �__str__� r. zBadException.__str__N)r r r! ri r r r r rh � r* rh c @ s$ e Zd ZdZedd� �Zdd� ZdS )�
54 ReallyBadReprr? c C s t d��)Nz
55 I am horrible)�
56 ValueErrorr, r r r � __class__� s zReallyBadRepr.__class__c C s t � �r )rh r, r r r r< � s zReallyBadRepr.__repr__N)r r r! �propertyrl r< r r r r rj � s
57
58 rj c C s< t �t�� t�t� � W d � d S 1 sw Y d S r )r[ re rh r rj r r r r �test_really_bad_repr� s "�rn c @ r3 )�SANr5 r r r r ro � r6 ro c @ r3 )�SBNr5 r r r r rp � r6 rp c sD e Zd Z� fdd�Zdd� Zdd� Zdd� Zd d
59 � Zd d � Z� Z S )
60 � TestsPrettyc s@ t � tt��}| �|d� t� }t � tt|��}| �|d� d S )Nz<super: \S+.SA, None>z$<super: \S+.SA,\s+<\S+.SB at 0x\S+>>)r �superro � assertRegexrp )r r] �sb�rl r r �test_super_repr� s
61 zTestsPretty.test_super_reprc C �> t td��}t�|�}|�dd�dd � }| �|ddg� d S )N�' �
62 rP rR � 999,z ...])�listrE r �rsplit� assertEqual)r Zlisr �last2r r r �test_long_list� �
63 zTestsPretty.test_long_listc C rw )Nrx ry rP rR rz � ...})�setrE r r| r} )r �sr r~ r r r �
64 test_long_set� r� zTestsPretty.test_long_setc C rw )Nrx ry rP rR rz z ...))�tuplerE r r| r} )r �tupr r~ r r r �test_long_tuple� r� zTestsPretty.test_long_tuplec C sD dd� t d�D �}t�|�}|�dd�dd � }| �|ddg� d S ) Nc S s i | ]}||�qS r r )rB �nr r r �
65 <dictcomp>� s z.TestsPretty.test_long_dict.<locals>.<dictcomp>rx ry rP rR z
66 999: 999,r� )rE r r| r} )r �dr r~ r r r �test_long_dict� s
67 zTestsPretty.test_long_dictc C s t � tj�}| �d|� d S )Nr/ )r r+ r- ZassertIn)r r] r r r �test_unbound_method� s z TestsPretty.test_unbound_method)
68 r r r! rv r r� r� r� r� �
69 __classcell__r r ru r rq � s rq c @ r
70 )� MetaClassc C s t �| |tfd|i�S )N�name)�type�__new__�object)�clsr� r r r r� � s zMetaClass.__new__c C s
71 d| j S )Nz[CUSTOM REPR FOR CLASS %s])r� r, r r r r< � r zMetaClass.__repr__N)r r r! r� r< r r r r r� � s r� �
72 ClassWithMetac C s t � t�} | dks J �d S )Nz%[CUSTOM REPR FOR CLASS ClassWithMeta])r r� r_ r r r �test_metaclass_repr s
73 r� c sX d} | � G � fdd�dt �}|� }t�|�}|| ks J �t�|g�}|d| ks*J �d S )Nu
74 üniçodéc s e Zd Z� fdd�ZdS )z test_unicode_repr.<locals>.Cc s � S r r r, �Zustrr r r< r. z%test_unicode_repr.<locals>.C.__repr__Nr= r r� r r �C s r� z[%s])r� r )�ur� �cr r r� r �test_unicode_repr s
75 r� c sh � fdd�� d� _ t� } t�| �}� |jt< |�t� |�� | �� }|dt ks+J �� j du s2J �d S )Nc s | t u rd� _t�| ||�S )NT)r+ �calledr Z _type_pprint)rT r r ��type_pprint_wrapperr r r� s z-test_basic_class.<locals>.type_pprint_wrapperFz%s.MyObjT)
76 r� r r ZRepresentationPrinterZtype_pprintersr� r+ �flush�getvaluer )�stream�printerr] r r� r �test_basic_class s
77
78
79 r� c C �` t � } | �| � t � dft dd� tdd�D ��df| dfg}|D ]
80 \}}t�|�|ks-J �q d S )Nz UserList([])c s � � | ]}|V qd S r r rA r r r rC 4 � � z,test_collections_userlist.<locals>.<genexpr>�� �� a@ UserList([1000,
81 1001,
82 1002,
83 1003,
84 1004,
85 1005,
86 1006,
87 1007,
88 1008,
89 1009,
90 1010,
91 1011,
92 1012,
93 1013,
94 1014,
95 1015,
96 1016,
97 1017,
98 1018,
99 1019])zUserList([UserList(...)]))r �appendrE r ��a�casesrT rb r r r �test_collections_userlist, s
100 �� �r� c C sj t � } | | _t t�}||d< t t�dft tddi�df| df|dfg}|D ]
101 \}}t�|�|ks2J �q%d S )N�keyzdefaultdict(list, {})z2--------------------------------------------------z\defaultdict(list,
102 {'key': '--------------------------------------------------'})z!defaultdict(defaultdict(...), {})z,defaultdict(list, {'key': defaultdict(...)}))r �default_factoryr{ r )r� �br� rT rb r r r � test_collections_defaultdictQ s
103 �� �r� c C s^ t � } | | d< t � dft dd� tdd�D ��df| dfg}|D ]
104 \}}t�|�|ks,J �q d S ) Nr� z
105 OrderedDict()c s s � | ]}||fV qd S r r rA r r r rC m rD z/test_collections_ordereddict.<locals>.<genexpr>r� i� a OrderedDict([(1000, 1000),
106 (1001, 1001),
107 (1002, 1002),
108 (1003, 1003),
109 (1004, 1004),
110 (1005, 1005),
111 (1006, 1006),
112 (1007, 1007),
113 (1008, 1008),
114 (1009, 1009)])z(OrderedDict([('key', OrderedDict(...))]))r rE r r� r r r � test_collections_ordereddictf s � � �r� c C r� )Nz deque([])c s r� r r rA r r r rC � r� z)test_collections_deque.<locals>.<genexpr>r� r� a deque([1000,
115 1001,
116 1002,
117 1003,
118 1004,
119 1005,
120 1006,
121 1007,
122 1008,
123 1009,
124 1010,
125 1011,
126 1012,
127 1013,
128 1014,
129 1015,
130 1016,
131 1017,
132 1018,
133 1019])zdeque([deque(...)]))r r� rE r r� r r r �test_collections_deque s
134 �� �r� c C sf G dd� dt �} t � dft dd�df| dd�dft ddd �d
135 fg}|D ]
136 \}}t�|�|ks0J �q#d S ) Nc @ r3 )z+test_collections_counter.<locals>.MyCounterNr5 r r r r � MyCounter� r6 r� z Counter()r? )r� zCounter({'a': 1})zMyCounter({'a': 1})� )r� r� zCounter({'c': 22, 'a': 1}))r r )r� r� rT rb r r r �test_collections_counter� s � �r� c C s� t j} i }| |�}||d< ||d< | i �df| d | i �i�df| dd� tjD ��df|df|d fg}|D ]
137 \}}t�|�|ksAJ �q4d S )
138 NrP r zmappingproxy({})z&mappingproxy({None: mappingproxy({})})c S s i | ]}||� � �qS r )�upper)rB �kr r r r� � s z%test_mappingproxy.<locals>.<dictcomp>ap mappingproxy({'a': 'A',
139 'b': 'B',
140 'c': 'C',
141 'd': 'D',
142 'e': 'E',
143 'f': 'F',
144 'g': 'G',
145 'h': 'H',
146 'i': 'I',
147 'j': 'J',
148 'k': 'K',
149 'l': 'L',
150 'm': 'M',
151 'n': 'N',
152 'o': 'O',
153 'p': 'P',
154 'q': 'Q',
155 'r': 'R',
156 's': 'S',
157 't': 'T',
158 'u': 'U',
159 'v': 'V',
160 'w': 'W',
161 'x': 'X',
162 'y': 'Y',
163 'z': 'Z'})z1mappingproxy({2: {...}, 3: {2: {...}, 3: {...}}})z1{2: mappingproxy({2: {...}, 3: {...}}), 3: {...}})�types�MappingProxyType�string�ascii_lowercaser )ZMPZunderlying_dictZ mp_recursiver� rT rb r r r �test_mappingproxy� s$
164 ��� "�r� c C sx t j} | � }||_||_| � df| | � d�df| | tjd�gd d d�df|dfg}|D ]
165 \}}t�|�|ks9J �q,d S ) Nz namespace())�xznamespace(x=namespace()))r� r )Z a_long_nameZ a_short_namez�namespace(a_long_name=[namespace(s='abcdefghijklmnopqrstuvwxyz'),
166 namespace(s='abcdefghijklmnopqrstuvwxyz'),
167 namespace(s='abcdefghijklmnopqrstuvwxyz')],
168 a_short_name=None)z6namespace(first=namespace(...), second=namespace(...)))r� �SimpleNamespace�first�secondr� r� r )ZSNZ sn_recursiver� rT rb r r r �test_simplenamespace� s ��
169 �r� c C sH t � ttj��} | �dddtd� �}t � tj�}|d| ks"J �d S )Nry � �environ)r �dict�osr� �replace�len)Z dict_reprZ
170 dict_indentedZenv_reprr r r �test_pretty_environ� s r� c C s< ddl } t�| j�dksJ �ddd�}dt�|�v s J �dS )z Test pretty print of functionr Nz <function posixpath.join(a, *p)>c S s | rdS dS )N�* z Don't panicr )�questionr r r �meaning_of_life s z-test_function_pretty.<locals>.meaning_of_lifez meaning_of_life(question=None)r )� posixpathr rF )r� r� r r r �test_function_pretty� s
171 r� c @ s e Zd ZdZdd� Zdd� ZdS )�OrderedCounterz?Counter that remembers the order elements are first encounteredc C s d| j jt| �f S )Nz%s(%r))rl r r r, r r r r<
172 s zOrderedCounter.__repr__c C s | j t| �ffS r )rl r r, r r r �
173 __reduce__ s zOrderedCounter.__reduce__N)r r r! �__doc__r< r� r r r r r�
174 s r� c @ r" )�MySetc C r; )N�miner r, r r r r< r. zMySet.__repr__Nr= r r r r r� r* r� c C s2 t d�} dt�| �v s
175 J �t�t� �dksJ �dS )z@A custom repr should override a pretty printer for a parent typeZ abracadabrazOrderedCounter(OrderedDictr� N)r� r r� )�ocr r r �test_custom_repr s r� )Er� � collectionsr r r r r r� r[ r� r� rY ZunittestZ IPython.libr �ior r� r r� r# r+ r0 r2 r4 r r7 r: r> rJ rM rO �markZ parametrize�zipr� � frozensetrV r^ r` rc rd rg � Exceptionrh rj rn ro rp ZTestCaserq r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r �<module> s�
176
177
178 � ��
179
180 *%#
181 .