File: //usr/local/lib/python3.7/test/__pycache__/test_code.cpython-37.pyc
B
��g�) � @ s� d Z ddlZddlZddlZddlZddlZy ddlZW n ek
rP dZY nX ddlm Z m
Z
m Z m Z dd� Z
dd� Zdd � ZG d
d � d ej�Zd d
� ZG dd� dej�ZG dd� dej�Ze dd��rtedk �rtejZe�dej�ZejZefe_ ej e_ ej Z ej!ej ejfe _ ej"e _ ej#Z$ej!ej e�%ej�fe$_ ej"e$_ da&dd� Z'ee'�Z(ee(�Z)G dd� dej�Z*ddd�Z+e,dk�r�e+� dS ) a7 This module includes tests of the code object representation.
>>> def f(x):
... def g(y):
... return x + y
... return g
...
>>> dump(f.__code__)
name: f
argcount: 1
kwonlyargcount: 0
names: ()
varnames: ('x', 'g')
cellvars: ('x',)
freevars: ()
nlocals: 2
flags: 3
consts: ('None', '<code object g>', "'f.<locals>.g'")
>>> dump(f(4).__code__)
name: g
argcount: 1
kwonlyargcount: 0
names: ()
varnames: ('y',)
cellvars: ()
freevars: ('x',)
nlocals: 1
flags: 19
consts: ('None',)
>>> def h(x, y):
... a = x + y
... b = x - y
... c = a * b
... return c
...
>>> dump(h.__code__)
name: h
argcount: 2
kwonlyargcount: 0
names: ()
varnames: ('x', 'y', 'a', 'b', 'c')
cellvars: ()
freevars: ()
nlocals: 5
flags: 67
consts: ('None',)
>>> def attrs(obj):
... print(obj.attr1)
... print(obj.attr2)
... print(obj.attr3)
>>> dump(attrs.__code__)
name: attrs
argcount: 1
kwonlyargcount: 0
names: ('print', 'attr1', 'attr2', 'attr3')
varnames: ('obj',)
cellvars: ()
freevars: ()
nlocals: 1
flags: 67
consts: ('None',)
>>> def optimize_away():
... 'doc string'
... 'not a docstring'
... 53
... 0x53
>>> dump(optimize_away.__code__)
name: optimize_away
argcount: 0
kwonlyargcount: 0
names: ()
varnames: ()
cellvars: ()
freevars: ()
nlocals: 0
flags: 67
consts: ("'doc string'", 'None')
>>> def keywordonly_args(a,b,*,k1):
... return a,b,k1
...
>>> dump(keywordonly_args.__code__)
name: keywordonly_args
argcount: 2
kwonlyargcount: 1
names: ()
varnames: ('a', 'b', 'k1')
cellvars: ()
freevars: ()
nlocals: 3
flags: 67
consts: ('None',)
� N)� run_doctest� run_unittest� cpython_only�check_impl_detailc c s8 x2| D ]*}t |�}|�d�r*d|j V q|V qW dS )z.Yield a doctest-safe sequence of object reprs.z <code objectz<code object %s>N)�repr�
startswith�co_name)�tZelt�r� r �*/usr/local/lib/python3.7/test/test_code.py�constsv s
r
c C s@ x&dD ] }t d|t| d| �f � qW t dtt| j��� dS )z1Print out a text representation of a code object.) �nameZargcountZkwonlyargcount�namesZvarnamesZcellvarsZfreevarsZnlocals�flagsz%s: %sZco_zconsts:N)�print�getattr�tupler
� co_consts)�co�attrr r r �dump s
r c C s dt � �|�� �S )NzForeign getitem: )�super� __getitem__)�self�ir r r �external_getitem� s r c @ s$ e Zd Zedd� �Zedd� �ZdS )�CodeTestc C sD dd l }|�ddd�}| �|jd� | �|jd� | �|jd� d S )Nr �filename�funcname� )� _testcapiZ
code_newempty� assertEqual� co_filenamer �co_firstlineno)r r! r r r r �
test_newempty� s
zCodeTest.test_newemptyc s� ddl m�m� dd� �� fdd�����fdd�}G d d
� d
t�}||d t� |j}|jd j}| �||� | � |j
j t j
@ t|j
j �� |d d
dg�}| �|d d� d S )Nr )� FunctionType�CodeTypec s � fdd�j S )Nc s � S )Nr r )� __class__r r �<lambda>� � zICodeTest.test_closure_injection.<locals>.create_closure.<locals>.<lambda>)� __closure__)r( r )r( r �create_closure� s z7CodeTest.test_closure_injection.<locals>.create_closurec sF � | j | j| j| j| j| j| j| j| j| j | j
| j | j | j
d | j�S )z9A new code object with a __class__ cell added to freevars)r( )� co_argcount�co_kwonlyargcount�
co_nlocals� co_stacksize�co_flags�co_coder �co_names� co_varnamesr# r r$ � co_lnotab� co_freevars� co_cellvars)�c)r' r r �new_code� s
z1CodeTest.test_closure_injection.<locals>.new_codec s@ �|j �}|jrt��| �}|j}t| |� |t� |||�� d S )N)�__code__r+ �AssertionError� __defaults__�setattr�globals)�clsr �f�codeZclosure�defaults)r& r, r9 r r �add_foreign_method� s
z;CodeTest.test_closure_injection.<locals>.add_foreign_methodc @ s e Zd ZdS )z-CodeTest.test_closure_injection.<locals>.ListN)�__name__�
__module__� __qualname__r r r r �List� s rG r � � � zForeign getitem: 1)�typesr& r' �listr r r+ �
cell_contentsZassertIs� assertFalser: r1 �inspectZ CO_NOFREE�hexr" )r rC rG ZfunctionZ class_ref�objr )r' r& r, r9 r �test_closure_injection� s z CodeTest.test_closure_injectionN)rD rE rF r r% rR r r r r r � s r c C s | t �d| d dd� �kS )N�_rH ���)�sys�intern)�sr r r �
isinterned� s rX c @ s` e Zd Zdd� Zdd� Zdd� Zedd� �Zed d
� �Zed d � �Z ed
d� �Z
edd� �Z dS )�CodeConstsTestc C s4 x|D ]}||kr|S qW | � ||� | �d� d S )NzShould never be reached)ZassertIn�fail)r r
�value�vr r r �
find_const� s
zCodeConstsTest.find_constc C s t |�s| �d|f � d S )NzString %r is not interned)rX rZ )r rW r r r �assertIsInterned� s z CodeConstsTest.assertIsInternedc C s t |�r| �d|f � d S )NzString %r is interned)rX rZ )r rW r r r �assertIsNotInterned� s z"CodeConstsTest.assertIsNotInternedc C s( t ddd�}| �|jd�}| �|� d S )Nzres = "str_value"�?�exec� str_value)�compiler] r r^ )r r r\ r r r �test_interned_string� s z#CodeConstsTest.test_interned_stringc C s, t ddd�}| �|jd�}| �|d � d S )Nzres = ("str_value",)r` ra )rb r )rc r] r r^ )r r r\ r r r � test_interned_string_in_tuple� s z,CodeConstsTest.test_interned_string_in_tuplec C s4 t ddd�}| �|jtd��}| �t|�d � d S )Nzres = a in {"str_value"}r` ra )rb r )rc r] r � frozensetr^ r )r r r\ r r r �!test_interned_string_in_frozenset� s z0CodeConstsTest.test_interned_string_in_frozensetc C s ddd�}| � |� � d S )Nrb c S s | S )Nr )�ar r r r@ � s z6CodeConstsTest.test_interned_string_default.<locals>.f)rb )r^ )r r@ r r r � test_interned_string_default� s
z+CodeConstsTest.test_interned_string_defaultc C s( t ddd�}| �|jd�}| �|� d S )Nzres = "str\0value!"r` ra z
str value!)rc r] r r_ )r r r\ r r r � test_interned_string_with_null� s z-CodeConstsTest.test_interned_string_with_nullN) rD rE rF r] r^ r_ r rd re rg ri rj r r r r rY � s rY c @ s e Zd Zdd� ZdS )�CodeWeakRefTestc sn i }t dt� |� |d }~d� _� fdd�}t�|j|�}� �t|� �� ~� �t|� �� � �� j� d S )Nz
def f(): passr@ Fc s
d� _ d S )NT)�called)rA )r r r �callback� s z,CodeWeakRefTest.test_basic.<locals>.callback) ra r> rl �weakref�refr: Z
assertTrue�boolrN )r � namespacer@ rm Zcoderefr )r r �
test_basic� s zCodeWeakRefTest.test_basicN)rD rE rF rr r r r r rk � s rk T)�cpythonc C s | a d S )N)�
LAST_FREED)Zptrr r r �myfree s ru c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�CoExtrac C s t d�S )Nz lambda:42)�eval)r r r r �get_func$ s zCoExtra.get_funcc C s< | � � }| �ttdtt�d�� | �ttdtt�d�� d S )N�* �d )rx � assertRaises� SystemError�SetExtra�
FREE_INDEX�ctypes�c_voidp�GetExtra)r r@ r r r �test_get_non_code* s
zCoExtra.test_get_non_codec C sJ | � � }| �tt|jtd t�d�� | �t |jtd t�d��d� d S )Nrz r )
rx r{ r| r} r: r~ r r� r" r� )r r@ r r r �test_bad_index2 s
zCoExtra.test_bad_indexc C s. | � � }t|jtt�d�� ~| �td� d S )Nrz )rx r} r: r~ r r� r" rt )r r@ r r r �test_free_called9 s zCoExtra.test_free_calledc C sn | � � }t�� }t|jtt�d�� t|jtt�d�� | �td� t�� }t|jt|� | �|j d� ~d S )N�� i, )
rx r r� r} r: r~ r" rt r� r[ )r r@ Zextrar r r � test_get_setB s zCoExtra.test_get_setc C sZ | � � }G dd� dtj�}t|jtt�d�� ||| �}~|�� |� � | �
t d� d S )Nc s$ e Zd Z� fdd�Zdd� Z� ZS )z6CoExtra.test_free_different_thread.<locals>.ThreadTestc s t � �� || _|| _d S )N)r �__init__r@ �test)r r@ r� )r( r r r� W s
z?CoExtra.test_free_different_thread.<locals>.ThreadTest.__init__c S s | ` | j�td� d S )Ni� )r@ r� r" rt )r r r r �run[ s z:CoExtra.test_free_different_thread.<locals>.ThreadTest.run)rD rE rF r� r� �
__classcell__r r )r( r �
ThreadTestV s r� i� ) rx � threadingZThreadr} r: r~ r r� �start�joinr" rt )r r@ r� �ttr r r �test_free_different_threadR s
z"CoExtra.test_free_different_threadN) rD rE rF rx r� r� r� r� r� r r r r rv # s rv c C sH ddl m} t|| � tttg}tdd�r<td k r<|�t � t
|� d S )Nr )� test_codeT)rs ) r� r� r r rY rk r r �appendrv r )�verboser� Ztestsr r r � test_mainf s
r� �__main__)N)-�__doc__rO rU r� Zunittestrn r � ImportErrorZ test.supportr r r r r
r r ZTestCaser rX rY rk Z pythonapi�pyZ CFUNCTYPEr� ZfreefuncZ _PyEval_RequestCodeExtraIndexZRequestCodeExtraIndexZargtypesZ c_ssize_tZrestypeZ_PyCode_SetExtrar} Z py_objectZc_intZ_PyCode_GetExtrar� ZPOINTERrt ru Z FREE_FUNCr~ rv r� rD r r r r �<module>g sL
30C