File: //usr/local/lib/python3.7/test/__pycache__/test_code.cpython-37.opt-2.pyc
B
��g�) � @ s� d dl Z d dlZd dlZd dlZd dlZy d dlZW n ek
rL dZY nX d dlmZm Z m
Z
m Z dd� Z dd� Z
dd� ZG d d
� d
ej�Zd d � ZG d
d� dej�ZG dd� dej�Ze dd��rpedk �rpejZe�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 s8 x2| D ]*}t |�}|�d�r*d|j V q|V qW d S )Nz <code objectz<code object %s>)�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��� d S )N) �nameZargcountZkwonlyargcount�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 @ 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 )N)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 s6 �|j �}�| �}|j}t| |� |t� |||�� d S )N)�__code__� __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 rF r � � � zForeign getitem: 1)�typesr& r' �listr r r+ �
cell_contentsZassertIs� assertFalser: r1 �inspectZ CO_NOFREE�hexr" )r rB rF ZfunctionZ class_ref�objr )r' r& r, r9 r �test_closure_injection� s z CodeTest.test_closure_injectionN)rC rD rE r r% rQ r r r r r � s r c C s | t �d| d dd� �kS )N�_rG ���)�sys�intern)�sr r r �
isinterned� s rW 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)rW rY )r rV r r r �assertIsInterned� s z CodeConstsTest.assertIsInternedc C s t |�r| �d|f � d S )NzString %r is interned)rW rY )r rV 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_ r` )ra r )rb 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_ r` )ra r )rb 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 )Nra c S s | S )Nr )�ar r r r? � s z6CodeConstsTest.test_interned_string_default.<locals>.f)ra )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_ r` z
str value!)rb r\ r r^ )r r r[ r r r � test_interned_string_with_null� s z-CodeConstsTest.test_interned_string_with_nullN) rC rD rE r\ r] r^ r rc rd rf rh ri r r r r rX � s rX 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)r@ )r r r �callback� s z,CodeWeakRefTest.test_basic.<locals>.callback) r` r= rk �weakref�refr: Z
assertTrue�boolrM )r � namespacer? rl Zcoderefr )r r �
test_basic� s zCodeWeakRefTest.test_basicN)rC rD rE rq r r r r rj � s rj T)�cpythonc C s | a d S )N)�
LAST_FREED)Zptrr r r �myfree s rt 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 )rw � 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 )Nry r )
rw rz 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 )Nry )rw r| r: r} r~ r r" rs )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, )
rw r~ r r| r: r} r" rs r� rZ )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" rs )r r r r �run[ s z:CoExtra.test_free_different_thread.<locals>.ThreadTest.run)rC rD rE r� r� �
__classcell__r r )r( r �
ThreadTestV s r� i� ) rw � threadingZThreadr| r: r} r~ r �start�joinr" rs )r r? r� �ttr r r �test_free_different_threadR s
z"CoExtra.test_free_different_threadN) rC rD rE rw r� r� r� r� r� r r r r ru # s ru c C sH ddl m} t|| � tttg}tdd�r<td k r<|�t � t
|� d S )Nr )� test_codeT)rr ) r� r� r r rX rj r r~ �appendru r )�verboser� Ztestsr r r � test_mainf s
r� �__main__)N),rN rT r� Zunittestrm r~ � ImportErrorZ test.supportr r r r r
r r ZTestCaser rW rX rj Z pythonapi�pyZ CFUNCTYPEr ZfreefuncZ _PyEval_RequestCodeExtraIndexZRequestCodeExtraIndexZargtypesZ c_ssize_tZrestypeZ_PyCode_SetExtrar| Z py_objectZc_intZ_PyCode_GetExtrar� ZPOINTERrs rt Z FREE_FUNCr} ru r� rC r r r r �<module>i sJ
30C