File: //usr/local/lib/python3.7/test/__pycache__/test_capi.cpython-37.pyc
B
��g+Z � @ s� d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl Z d dl Z d dl
mZ d dlmZ d dlmZmZ y d dlZW n ek
r� dZY nX e�d�Zeed�Zdd � ZG d
d � d �ZG d d
� d
e j �Z G dd� de j �Z G dd� de j �Z G dd� de j �Z G dd� de j �Z!G dd� de j �Z"G dd� de"�Z#e �$e�%� d�G dd � d e"��Z&e �$ed �G d d � d e"��Z'e(d k�r�e �)� dS )!� )� OrderedDictN)�support)�MISSING_C_DOCSTRINGS)�assert_python_failure�assert_python_ok� _testcapiZgettotalrefcountc C s | S )zsome doc� )�selfr r �*/usr/local/lib/python3.7/test/test_capi.py� testfunction s r c @ s e Zd Ze�e�Ze�e�ZdS )�InstanceMethodN)�__name__�
__module__� __qualname__r Zinstancemethod�idr r r r r
r # s
r c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Ze�e d �d
d � �Z
e�e d �d d
� �Z e� e
d�dd� �Zdd� Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!S )"�CAPITestc C s~ t � }| �t|�|�� � | �|�� |k� | �|jjtj� | �t jjtj� dt j_| �tjd� | �tt |jdd� d S )N�test� attribute)
r � assertEqualr �
assertTruer �__doc__r � assertRaises�AttributeError�setattr)r �instr r r
�test_instancemethod) s z CAPITest.test_instancemethodc C s` t �� �" tjtjddgtjtjd�}W d Q R X |�� \}}| �|d� | � |�
� � d�� d S )Nz-cz4import _testcapi;_testcapi.crash_no_current_thread())�stdout�stderr� s8 Fatal Python error: PyThreadState_Get: no current thread) r �SuppressCrashReport�
subprocess�Popen�sys�
executable�PIPEZ communicater r �rstrip�
startswith)r �p�out�errr r r
� test_no_FatalError_infinite_loop4 s
z)CAPITest.test_no_FatalError_infinite_loopc C s | � ttj� d S )N)r �
ValueErrorr Z!make_memoryview_from_NULL_pointer)r r r r
�!test_memoryview_from_NULL_pointerB s z*CAPITest.test_memoryview_from_NULL_pointerc
C s� t d�}td�}y|�W n� t k
r� } z�|j}t�� }t�|j|d �}t�� }tj|� }t�� } | �|d |� | � ||j||f� | � ||� | � | |� | � ||j|d f� | � ||� W d d }~X Y n X | �
d� d S )N�5ZTEST� F) r+ � TypeError�
__traceback__r" �exc_infor Z set_exc_info� __class__r ZassertSequenceEqualr )
r Zraised_exception�new_exc�e�tbZorig_sys_exc_infoZ
orig_exc_infoZnew_sys_exc_infoZ new_exc_infoZreset_sys_exc_infor r r
�
test_exc_infoE s$
zCAPITest.test_exc_infoz(_posixsubprocess required for this test.c C s� G dd� dt �}| �ttjd|� dddddd d
d d d
ddddd� G dd� dt �}| �ttjd|� dddddd d
d d d
ddddd� d S )Nc @ s e Zd Zdd� ZdS )z1CAPITest.test_seq_bytes_to_charp_array.<locals>.Zc S s dS )Nr. r )r r r r
�__len__` s z9CAPITest.test_seq_bytes_to_charp_array.<locals>.Z.__len__N)r
r r r7 r r r r
�Z_ s r8 r. � )r. � � � � � � �
� � �
� � � � c @ s e Zd Zdd� Zdd� ZdS )z1CAPITest.test_seq_bytes_to_charp_array.<locals>.Zc S s t jS )N)r" �maxsize)r r r r
r7 f s z9CAPITest.test_seq_bytes_to_charp_array.<locals>.Z.__len__c S s dS )N� xr )r �ir r r
� __getitem__h s z=CAPITest.test_seq_bytes_to_charp_array.<locals>.Z.__getitem__N)r
r r r7 rK r r r r
r8 e s )�objectr r/ �_posixsubprocess� fork_exec� MemoryError)r r8 r r r
� test_seq_bytes_to_charp_array\ s
(
z&CAPITest.test_seq_bytes_to_charp_arrayc C sH G dd� dt �}| �ttj|� dgdddddd d
d d d
ddddd� d S )Nc @ s e Zd Zdd� ZdS )z-CAPITest.test_subprocess_fork_exec.<locals>.Zc S s dS )Nr. r )r r r r
r7 p s z5CAPITest.test_subprocess_fork_exec.<locals>.Z.__len__N)r
r r r7 r r r r
r8 o s r8 � 1r9 )r. r: r; r< r= r> r? r@ rA rB rC rD rE rF rG )rL r r/ rM rN )r r8 r r r
�test_subprocess_fork_execm s
z"CAPITest.test_subprocess_fork_execz6Signature information for builtins requires docstringsc C s | � tjjd � | � tjjd � | � tjjd � | � tjjd � | � tjjd� | � tjjd � | � tjjd� | � tjjd � | � tjjd� | � tjjd � | � tj jd� | � tj jd� | � tj
jd � | � tj
jd� | � tj jd� | � tj jd� d S )Nz This docstring has no signature.z[docstring_with_invalid_signature($module, /, boo)
This docstring has an invalid signature.zedocstring_with_invalid_signature2($module, /, boo)
--
This docstring also has an invalid signature.z%This docstring has a valid signature.z($module, /, sig)z>
This docstring has a valid signature and some extra newlines.z($module, /, parameter)) r r Z no_docstringr �__text_signature__Zdocstring_emptyZdocstring_no_signatureZ docstring_with_invalid_signatureZ!docstring_with_invalid_signature2Zdocstring_with_signatureZ#docstring_with_signature_but_no_docZ+docstring_with_signature_and_extra_newlines)r r r r
� test_docstring_signature_parsingw s.
z)CAPITest.test_docstring_signature_parsingc C s� t j}|� }|� }| �|| d||f� | �|d d|df� | �d| dd|f� |}|| }| �|d||f� |}|d }| �|d|df� d}|| }| �|dd|f� d S )N�matmul�* �imatmul)r Z
matmulTyper )r �MZm1Zm2�or r r
�&test_c_type_with_matrix_multiplication� s z/CAPITest.test_c_type_with_matrix_multiplicationc C sh t r4t�d�}td|�\}}}| �|�dd�d� n0| �t��}t� � W d Q R X | �t
|j �d� d S )Nz�
import _testcapi
from test import support
with support.SuppressCrashReport():
_testcapi.return_null_without_error()
z-c�
r s� Fatal Python error: a function returned NULL without setting an error\nSystemError: <built-in function return_null_without_error> returned NULL without setting an error\n\nCurrent thread.*:\n File .*", line 6 in <module>zBreturn_null_without_error.* returned NULL without setting an error) �Py_DEBUG�textwrap�dedentr � assertRegex�replacer � SystemErrorr Zreturn_null_without_error�str� exception)r �code�rcr( r) �cmr r r
� test_return_null_without_error� s z'CAPITest.test_return_null_without_errorc C sh t r4t�d�}td|�\}}}| �|�dd�d� n0| �t��}t� � W d Q R X | �t
|j �d� d S )Nz�
import _testcapi
from test import support
with support.SuppressCrashReport():
_testcapi.return_result_with_error()
z-cr[ r s+ Fatal Python error: a function returned a result with an error set\nValueError\n\nThe above exception was the direct cause of the following exception:\n\nSystemError: <built-in function return_result_with_error> returned a result with an error set\n\nCurrent thread.*:\n File .*, line 6 in <module>z>return_result_with_error.* returned a result with an error set) r\ r] r^ r r_ r` r ra r Zreturn_result_with_errorrb rc )r rd re r( r) rf r r r
� test_return_result_with_error� s z&CAPITest.test_return_result_with_errorc C s t �� d S )N)r �test_buildvalue_N)r r r r
ri � s zCAPITest.test_buildvalue_Nc C s< d}t d|�\}}}| �d|� | �d|� | �d|� d S )Na* if 1:
import _testcapi
class C(): pass
# The first loop tests both functions and that remove_mem_hooks()
# can be called twice in a row. The second loop checks a call to
# set_nomemory() after a call to remove_mem_hooks(). The third
# loop checks the start and stop arguments of set_nomemory().
for outer_cnt in range(1, 4):
start = 10 * outer_cnt
for j in range(100):
if j == 0:
if outer_cnt != 3:
_testcapi.set_nomemory(start)
else:
_testcapi.set_nomemory(start, start + 1)
try:
C()
except MemoryError as e:
if outer_cnt != 3:
_testcapi.remove_mem_hooks()
print('MemoryError', outer_cnt, j)
_testcapi.remove_mem_hooks()
break
z-cs MemoryError 1 10s MemoryError 2 20s MemoryError 3 30)r �assertIn)r rd re r( r) r r r
�test_set_nomemory� s
zCAPITest.test_set_nomemoryc C s� G dd� dt �}G dd� dt �}dddd�}x|i t� |� |� |t|�||�||�gD ]R}| �t�|�t|�� �� | �t�|�t|�� �� | �t� |�t|�
� �� qTW d S ) Nc s4 e Zd Z� fdd�Z� fdd�Z� fdd�Z� ZS )z9CAPITest.test_mapping_keys_values_items.<locals>.Mapping1c s t t� �� �S )N)�list�super�keys)r )r2 r r
rn s z>CAPITest.test_mapping_keys_values_items.<locals>.Mapping1.keysc s t t� �� �S )N)rl rm �values)r )r2 r r
ro s z@CAPITest.test_mapping_keys_values_items.<locals>.Mapping1.valuesc s t t� �� �S )N)rl rm �items)r )r2 r r
rp s z?CAPITest.test_mapping_keys_values_items.<locals>.Mapping1.items)r
r r rn ro rp �
__classcell__r r )r2 r
�Mapping1 s rr c s4 e Zd Z� fdd�Z� fdd�Z� fdd�Z� ZS )z9CAPITest.test_mapping_keys_values_items.<locals>.Mapping2c s t t� �� �S )N)�tuplerm rn )r )r2 r r
rn s z>CAPITest.test_mapping_keys_values_items.<locals>.Mapping2.keysc s t t� �� �S )N)rs rm ro )r )r2 r r
ro s z@CAPITest.test_mapping_keys_values_items.<locals>.Mapping2.valuesc s t t� �� �S )N)rs rm rp )r )r2 r r
rp s z?CAPITest.test_mapping_keys_values_items.<locals>.Mapping2.items)r
r r rn ro rp rq r r )r2 r
�Mapping2 s rt r. r: r9 )�fooZbarZspam) �dictr ZassertListEqualr �get_mapping_keysrl rn �get_mapping_valuesro �get_mapping_itemsrp )r rr rt Zdict_obj�mappingr r r
� test_mapping_keys_values_items s z'CAPITest.test_mapping_keys_values_itemsc C sx | � ttjd � | � ttjd � | � ttjd � G dd� d�}|� }| � ttj|� | � ttj|� | � ttj|� d S )Nc @ s$ e Zd Zdd� Zdd� Zdd� ZdS )zCCAPITest.test_mapping_keys_values_items_bad_arg.<locals>.BadMappingc S s d S )Nr )r r r r
rn 3 s zHCAPITest.test_mapping_keys_values_items_bad_arg.<locals>.BadMapping.keysc S s d S )Nr )r r r r
ro 5 s zJCAPITest.test_mapping_keys_values_items_bad_arg.<locals>.BadMapping.valuesc S s d S )Nr )r r r r
rp 7 s zICAPITest.test_mapping_keys_values_items_bad_arg.<locals>.BadMapping.itemsN)r
r r rn ro rp r r r r
�
BadMapping2 s r| )r r r rw rx ry r/ )r r| Z bad_mappingr r r
�&test_mapping_keys_values_items_bad_arg- s z/CAPITest.test_mapping_keys_values_items_bad_argc C s� ddl m} | �|dd�d� | �|dd�d� | �|dd�d � | �|dd
�d � | �|d d�d
� | �|d d�d� | �|d d�d� | �|d d
�d� | �t|dd� | �t|d d� | �t|dd� d S )Nr )�pynumber_tobase�{ r: Z 0b1111011r> Z0o173r@ Z123rF Z0x7bi����z
-0b1111011z-0o173z-123z-0x7bg �^@)r r~ r r r/ ra )r r~ r r r
�test_pynumber_tobase> s z CAPITest.test_pynumber_tobaseN)r
r r r r* r, r6 �unittest�
skipUnlessrM rP rR ZskipIfr rT rZ rg rh ri rk r{ r} r� r r r r
r ' s
* r c @ s6 e Zd Zdd� Zd dd�Zdd� Zdd � Zd
d � ZdS )
�TestPendingCallsc sF � fdd�}x4t |�D ](}t�t�� d � xt�|�r.P q.W qW d S )Nc s � � d � d S )N)�appendr )�lr r
�callbackP s z6TestPendingCalls.pendingcalls_submit.<locals>.callbackg{�G�z�?)�range�time�sleep�randomr Z_pending_threadfunc)r r� �nr� rJ r )r� r
�pendingcalls_submitO s
z$TestPendingCalls.pendingcalls_submitNc C s� d}xzt |�|kr~dr.tjr.tdt |�f � xtd�D ] }|| }q8W |rX|j�� sXq|d7 }| �|dk d|t |�f � qW dr�tjr�tdt |�f � d S )Nr Fz(%i)i� r. i' z(timeout waiting for %i callbacks, got %i)�lenr �verbose�printr� �eventZis_setr )r r� r� �context�countrJ �ar r r
�pendingcalls_wait^ s
z"TestPendingCalls.pendingcalls_waitc s� d}g }G dd� dt �}|� � g � _d� _|� j � _d� _t�� � _t�� � _ � �fdd�t
� j�D �}t � |�� ��
� j|� � W d Q R X d S )N� c @ s e Zd ZdS )z8TestPendingCalls.test_pendingcalls_threaded.<locals>.fooN)r
r r r r r r
ru v s ru r: r c s g | ]}t j�j� fd ��qS ))�target�args)� threading�Thread�pendingcalls_thread)�.0rJ )r� r r r
�
<listcomp> s z?TestPendingCalls.test_pendingcalls_threaded.<locals>.<listcomp>)rL r� r� �nThreads� nFinishedr� ZLock�lockZEventr� r� r Z
start_threadsr� )r r� Zthreadsru r )r� r r
�test_pendingcalls_threadedq s
z+TestPendingCalls.test_pendingcalls_threadedc C sj z| � |j|j� W d |j�. | jd7 _|j}drFtjrFtd|� W d Q R X ||jkrd|j �
� X d S )Nr. Fzfinished threads: ) r� r� r� r� r� r r� r� r� r� �set)r r� r� r r r
r� � s
z$TestPendingCalls.pendingcalls_threadc C s$ g }d}| � ||� | �||� d S )N�@ )r� r� )r r� r� r r r
� test_pendingcalls_non_threaded� s z/TestPendingCalls.test_pendingcalls_non_threaded)N)r
r r r� r� r� r� r� r r r r
r� M s
r� c @ s e Zd Zdd� ZdS )�SubinterpreterTestc C s| dd l }t�� \}}d�|�}t|d��J}t�|�}| �|d� | �t �
|�t t j
�� | �t �
|�t |�� W d Q R X d S )Nr z�if 1:
import sys, builtins, pickle
with open({:d}, "wb") as f:
pickle.dump(id(sys.modules), f)
pickle.dump(id(builtins), f)
�rb)�builtins�os�pipe�format�openr Zrun_in_subinterpr ZassertNotEqual�pickle�loadr r" �modules)r r� �r�wrd �f�retr r r
�test_subinterps� s
z"SubinterpreterTest.test_subinterpsN)r
r r r� r r r r
r� � s r� c @ s e Zd Zejdd� �ZdS )�TestThreadStatec s2 � fdd�}|� t j|d�}|�� |�� d S )Nc sH g � � fdd�} t �| � | }}t�d� ��� �t�� �dd� d S )Nc s � � t�� � d S )N)r� r� � get_identr )�identsr r
r� � s zCTestThreadState.test_thread_state.<locals>.target.<locals>.callbackr. r9 z/Couldn't find main thread correctly in the list)r Z_test_thread_stater� r� r r� r� r� )r� r� �b)r )r� r
r� � s
z1TestThreadState.test_thread_state.<locals>.target)r� )r� r� �start�join)r r� �tr )r r
�test_thread_state� s
z!TestThreadState.test_thread_stateN)r
r r r Z reap_threadsr� r r r r
r� � s r� c @ s& e Zd Ze� �dd� ee�D �� dS )�
Test_testcapic c s0 | ](}|� d �r|�d�s|tt|�fV qdS )Ztest_�_codeN)r& �endswith�getattrr )r� �namer r r
� <genexpr>� s zTest_testcapi.<genexpr>N)r
r r �locals�update�dirr r r r r
r� � s r� c @ sd e Zd ZdZdZdd� Zdd� Zdd� Zd d
� Zd d � Z d
d� Z
dd� Z dd� Z dd� Z
dd� ZdS )�PyMemDebugTests�debugz(?:0x)?[0-9a-fA-F]+c C s6 t �� � td|| jd�}W d Q R X |j}|�dd�S )Nz-c)� PYTHONMALLOC�asciir` )r r r r� r) �decode)r rd r( r r r r
�check� s
zPyMemDebugTests.checkc C s< | � d�}d}|j| jd�}tj|tjd�}| �||� d S )Nz3import _testcapi; _testcapi.pymem_buffer_overflow()a Debug memory block at address p={ptr}: API 'm'\n 16 bytes originally requested\n The [0-9] pad bytes at p-[0-9] are FORBIDDENBYTE, as expected.\n The [0-9] pad bytes at tail={ptr} are not all FORBIDDENBYTE \(0x[0-9a-f]{{2}}\):\n at tail\+0: 0x78 \*\*\* OUCH\n at tail\+1: 0xfd\n at tail\+2: 0xfd\n .*\n The block was made by call #[0-9]+ to debug malloc/realloc.\n Data at p: cd cd cd .*\n\nEnable tracemalloc to get the memory block allocation traceback\n\nFatal Python error: bad trailing pad byte)�ptr)�flags)r� r� � PTR_REGEX�re�compile�DOTALLr_ )r r( �regexr r r
�test_buffer_overflow� s
z$PyMemDebugTests.test_buffer_overflowc C s, | � d�}d}|j| jd�}| �||� d S )Nz.import _testcapi; _testcapi.pymem_api_misuse()a� Debug memory block at address p={ptr}: API 'm'\n 16 bytes originally requested\n The [0-9] pad bytes at p-[0-9] are FORBIDDENBYTE, as expected.\n The [0-9] pad bytes at tail={ptr} are FORBIDDENBYTE, as expected.\n The block was made by call #[0-9]+ to debug malloc/realloc.\n Data at p: cd cd cd .*\n\nEnable tracemalloc to get the memory block allocation traceback\n\nFatal Python error: bad ID: Allocated using API 'm', verified using API 'r'\n)r� )r� r� r� r_ )r r( r� r r r
�test_api_misuse� s
z PyMemDebugTests.test_api_misusec C s | � |�}d}| �||� d S )NzJFatal Python error: Python memory allocator called without holding the GIL)r� rj )r rd r( Zexpectedr r r
�check_malloc_without_gil� s
z(PyMemDebugTests.check_malloc_without_gilc C s d}| � |� d S )Nz6import _testcapi; _testcapi.pymem_malloc_without_gil())r� )r rd r r r
� test_pymem_malloc_without_gil s z-PyMemDebugTests.test_pymem_malloc_without_gilc C s d}| � |� d S )Nz9import _testcapi; _testcapi.pyobject_malloc_without_gil())r� )r rd r r r
� test_pyobject_malloc_without_gil s z0PyMemDebugTests.test_pyobject_malloc_without_gilc C s* t �d�}|j|d�}td|| jd� d S )Nas
import gc, os, sys, _testcapi
# Disable the GC to avoid crash on GC collection
gc.disable()
obj = _testcapi.{func}()
error = (_testcapi.pyobject_is_freed(obj) == False)
# Exit immediately to avoid a crash while deallocating
# the invalid object
os._exit(int(error))
)�funcz-c)r� )r] r^ r� r r� )r r� rd r r r
�check_pyobject_is_freed
s z'PyMemDebugTests.check_pyobject_is_freedc C s | � d� d S )NZpyobject_uninitialized)r� )r r r r
�$test_pyobject_is_freed_uninitialized s z4PyMemDebugTests.test_pyobject_is_freed_uninitializedc C s | � d� d S )NZpyobject_forbidden_bytes)r� )r r r r
�&test_pyobject_is_freed_forbidden_bytes s z6PyMemDebugTests.test_pyobject_is_freed_forbidden_bytesc C s | � d� d S )NZpyobject_freed)r� )r r r r
�test_pyobject_is_freed_free! s z+PyMemDebugTests.test_pyobject_is_freed_freeN)r
r r r� r� r� r� r� r� r� r� r� r� r� r� r r r r
r� � s r� c @ s e Zd ZdZdS )�PyMemMallocDebugTestsZ malloc_debugN)r
r r r� r r r r
r� % s r� z
need pymallocc @ s e Zd ZdZdS )�PyMemPymallocDebugTestsZpymalloc_debugN)r
r r r� r r r r
r� ) s r� z
need Py_DEBUGc @ s e Zd ZdZdS )�PyMemDefaultTests� N)r
r r r� r r r r
r� . s r� �__main__)*� collectionsr r� r� r� r� r r" Z sysconfigr] r� r� r� r r Z test.supportr Ztest.support.script_helperr r rM � ImportError�
import_moduler �hasattrr\ r r ZTestCaser r� r� r� r� r� r� r� Z
with_pymallocr� r� r
�mainr r r r
�<module> sJ
(OY