Your IP : 216.73.217.129
U
i�f�6 � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZz d dlZW n ek
rT dZY nX d dlm Z m
Z
m Z m Z dd� Z
dd� Zdd� ZG d d
� d
ej�Zd d � ZG d
d� dej�ZG dd� dej�Ze dd��rxedk �rxejZe�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 )� N)� run_doctest� run_unittest� cpython_only�check_impl_detailc c s4 | D ]*}t |�}|�d�r(d|j V q|V qd S )Nz <code objectz<code object %s>)�repr�
startswith�co_name)�tZelt�r� r �3/opt/alt/python38/lib64/python3.8/test/test_code.py�consts� s
r
c C s< dD ] }t d|t| d| �f � qt dtt| j��� d S )N)
�nameZargcountZposonlyargcountZkwonlyargcount�namesZvarnamesZcellvarsZfreevarsZnlocals�flagsz%s: %sZco_zconsts:)�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 @ s4 e Zd Zedd� �Zedd� �Zdd� Zdd� Zd S )
�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� dd� �dd� �� ��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 )� FunctionTypec 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 S s | j | jd d�S )Nr( )� co_freevars)�replacer. )�cr r r �new_code� s z1CodeTest.test_closure_injection.<locals>.new_codec s6 �|j �}�| �}|j}t| |� |t� |||�� d S r' )�__code__� __defaults__�setattr�globals)�clsr �f�codeZclosure�defaults�r& r- r1 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 r@ r � � � zForeign getitem: 1)�typesr& �listr r r, �
cell_contentsZassertIs� assertFalser2 �co_flags�inspectZ CO_NOFREE�hexr" )r r; r@ ZfunctionZ class_ref�objr r: r �test_closure_injection� s
�z CodeTest.test_closure_injectionc C s\ dd� }|j }t|�}||j|j|j|j|j|j|j|j |j
|j |j |j
|j|j|j|j�S )Nc S s d S r' r r r r r �func� r+ z'CodeTest.test_constructor.<locals>.func)r2 �type� co_argcount�co_posonlyargcount�co_kwonlyargcount�
co_nlocals� co_stacksizerH �co_coder �co_names� co_varnamesr# r r$ � co_lnotabr. � co_cellvars)r rM r �CodeTyper r r �test_constructor� s( �zCodeTest.test_constructorc C s� dd� }|j }dd� }|j }ddddd d
|jtjB fd d |jfd
|jfdd|jfddddd|jffD ]D\}}| j||d��( |j f ||i�}| �
t ||�|� W 5 Q R X qdd S )Nc S s d} | S )NrA r ��xr r r rM � s z#CodeTest.test_replace.<locals>.funcc S s d} | S )NrB r )�yr r r �func2� s z$CodeTest.test_replace.<locals>.func2)rO r )rP r )rQ r )rR r )rS r rH )r$ �d rT r )rU )ZmynamerV )r. )Zfreevar)rX )Zcellvar)r# Z newfilename)r ZnewnamerW )r �value) r2 rH rI Z CO_COROUTINErT r rV rW ZsubTestr/ r" r )r rM r8 r^ �code2r r` r1 r r r � test_replace� s0 � zCodeTest.test_replaceN)r= r> r? r r% rL rZ rb r r r r r � s
$r c C s | t �d| d dd� �kS )N�_rA ���)�sys�intern)�sr r r �
isinterned
s rh c @ s` e Zd Zdd� Zdd� Zdd� Zedd� �Zed d
� �Zed d � �Z ed
d� �Z
edd� �Z dS )�CodeConstsTestc C s4 |D ]}||kr| S q| � ||� | �d� d S )NzShould never be reached)ZassertIn�fail)r r
r` �vr r r �
find_const s
zCodeConstsTest.find_constc C s t |�s| �d|f � d S )NzString %r is not interned�rh rj �r rg r r r �assertIsInterned s z CodeConstsTest.assertIsInternedc C s t |�r| �d|f � d S )NzString %r is internedrm rn r r r �assertIsNotInterned s z"CodeConstsTest.assertIsNotInternedc C s( t ddd�}| �|jd�}| �|� d S )Nzres = "str_value"�?�exec� str_value��compilerl r ro �r r rk r r r �test_interned_string! s z#CodeConstsTest.test_interned_stringc C s, t ddd�}| �|jd�}| �|d � d S )Nzres = ("str_value",)rq rr �rs r rt rv 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"}rq rr rx r )ru rl r � frozensetro r rv r r r �!test_interned_string_in_frozenset- s z0CodeConstsTest.test_interned_string_in_frozensetc C s ddd�}| � |� � d S )Nrs c S s | S r' r )�ar r r r7 5 s z6CodeConstsTest.test_interned_string_default.<locals>.f)rs )ro �r r7 r r r � test_interned_string_default3 s
z+CodeConstsTest.test_interned_string_defaultc C s( t ddd�}| �|jd�}| �|� d S )Nzres = "str\0value!"rq rr z
str value!)ru rl r rp rv r r r � test_interned_string_with_null9 s z-CodeConstsTest.test_interned_string_with_nullN) r= r> r? rl ro rp r rw ry r{ r~ r r r r r ri s
ri c @ s e Zd Zdd� ZdS )�CodeWeakRefTestc sn i }t dt� |� |d }~d� _� fdd�}t�|j|�}� �t|� �� ~� �t|� �� � �� j� d S )Nz
def f(): passr7 Fc s
d� _ d S )NT)�called)r8 �r r r �callbackK s z,CodeWeakRefTest.test_basic.<locals>.callback) rr r5 r� �weakref�refr2 Z
assertTrue�boolrG )r � namespacer7 r� Zcoderefr r� r �
test_basicB s zCodeWeakRefTest.test_basicN)r= r> r? r� r r r r r� @ s r� T�Zcpythonc C s | a d S r' )�
LAST_FREED)Zptrr r r �myfreek s r� c @ sH e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Ze d
d� �Z
dS )�CoExtrac C s t d�S )Nz lambda:42)�evalr� r r r �get_funcs s zCoExtra.get_funcc C s< | � � }| �ttdtt�d�� | �ttdtt�d�� d S )N�* r_ )r� � assertRaises� SystemError�SetExtra�
FREE_INDEX�ctypes�c_voidp�GetExtrar} r r r �test_get_non_codey s � �zCoExtra.test_get_non_codec C sJ | � � }| �tt|jtd t�d�� | �t |jtd t�d��d� d S )Nr_ r )
r� r� r� r� r2 r� r� r� r" r� r} r r r �test_bad_index� s ���zCoExtra.test_bad_indexc C s. | � � }t|jtt�d�� ~| �td� d S )Nr_ )r� r� r2 r� r� r� r" r� r} r r r �test_free_called� s zCoExtra.test_free_calledc C sn | � � }t�� }t|jtt�d�� t|jtt�d�� | �td� t�� }t|jt|� | �|j d� ~d S )N�� i, )
r� r� r� r� r2 r� r" r� r� r` )r r7 Zextrar r r � test_get_set� 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 r' )r �__init__r7 �test)r r7 r� r( r r r� � s
z?CoExtra.test_free_different_thread.<locals>.ThreadTest.__init__c S s | ` | j�td� d S )N�� )r7 r� r" r� r� r r r �run� s z:CoExtra.test_free_different_thread.<locals>.ThreadTest.run)r= r> r? r� r� �
__classcell__r r r( r �
ThreadTest� s r� r� ) r� � threadingZThreadr� r2 r� r� r� �start�joinr" r� )r r7 r� Zttr r r �test_free_different_thread� s
z"CoExtra.test_free_different_threadc
C s� dd� }|j }t|j �}t|j�}|�dtjd � |�dd� ||j|j|j |j
|j d |j t
|�|j|j|j|j|j|j|j|j|j�}t|�||jd|j|j�}G dd � d �}|� }t�|�} ||� ~d S )
Nc S s | S r' r r[ r r r r7 � s z-CoExtra.test_clean_stack_on_return.<locals>.f���Z LOAD_FASTr rA Znfc @ s e Zd ZdS )z/CoExtra.test_clean_stack_on_return.<locals>.VarNr< r r r r �Var� s r� ) r2 rN rE rT �insert�opcodeZopmaprO rP rQ rR rS rH �bytesr rU rV r# r r$ rW r. rX � __globals__r3 r, r� r� )
r r7 r8 Zct�bytecoder0 Z new_functionr� Z
the_object�varr r r �test_clean_stack_on_return� s:
�
z"CoExtra.test_clean_stack_on_returnN) r= r> r? r� r� r� r� r� r� r r� r r r r r� r s r� c C sH ddl m} t|| � tttg}tdd�r<td k r<|�t � t
|� d S )Nr )� test_codeTr� ) r� r� r r ri r� r r� �appendr� r )�verboser� Ztestsr r r � test_main� s
r� �__main__)N)-rI re r� Zunittestr� r� r� � ImportErrorZ test.supportr r r r r
r r ZTestCaser rh ri r� Z pythonapi�pyZ CFUNCTYPEr� ZfreefuncZ _PyEval_RequestCodeExtraIndexZRequestCodeExtraIndexZargtypesZ c_ssize_tZrestypeZ_PyCode_SetExtrar� Z py_objectZc_intZ_PyCode_GetExtrar� ZPOINTERr� r� Z FREE_FUNCr� r� r� r= r r r r �<module>� sN
i0
�h