Edit File: profiling.cpython-37.pyc
B ��4]A! � @ s� d Z ddlZddlZddlZddlZddlZddlmZ ddlm Z ddlm Z ddlm Z dd lm Z dd lm Z y ddlZW n ek r� dZY nX dZdZG d d � d e�Zddd�Zejddd��ZdS )z�Profiling support for unit and performance tests. These are special purpose profiling methods which operate in a more fine-grained way than nose's profiling plugin. � N� )�config)� gc_collect� )�jython)�pypy)�update_wrapper)�win32c @ sT e Zd ZdZdd� Zedd� �Zdd� Zdd � Zd d � Z d d � Z dd� Z dd� Z dS )�ProfileStatsFilez�"Store per-platform/fn profiling results in a file. We're still targeting Py2.5, 2.4 on 0.7 with no dependencies, so no json lib :( need to roll something silly c C sz t jd k ot jj| _| jp*t jd k o*t jj| _tj�|�| _ tj� | j �d | _ t � dd� �| _| �� | jrv| �� d S )N���c S s t �t�S )N)� collections� defaultdict�dict� r r �O/opt/alt/python37/lib64/python3.7/site-packages/sqlalchemy/testing/profiling.py�<lambda>: � z+ProfileStatsFile.__init__.<locals>.<lambda>)r �optionsZforce_write_profiles� force_writeZwrite_profiles�write�os�path�abspath�fname�split� short_fnamer r �data�_read�_write)�self�filenamer r r �__init__0 s zProfileStatsFile.__init__c C s� t jjd t jj }d�dd� tjdd� D ��}|g}|�|� trP|�d� t r^|�d� t rl|�d � |�t jj j r~d nd � t j �� }|�|r�d p�d � d�|�S )N�_�.c S s g | ] }t |��qS r )�str)�.0�vr r r � <listcomp>H s z1ProfileStatsFile.platform_key.<locals>.<listcomp>r r r r �winZ nativeunicodeZ dbapiunicodeZ cextensionsZ nocextensions)r Zdb�nameZdriver�join�sys� version_info�appendr r r ZdialectZconvert_unicode� requirementsZ_has_cextensions)r Z dbapi_key� py_versionZplatform_tokensZ _has_cextr r r � platform_keyB s z ProfileStatsFile.platform_keyc C s t }|| jko | j| j| kS )N)� _current_testr r0 )r �test_keyr r r � has_stats[ s zProfileStatsFile.has_statsc C s� t }| j| }|| j }d|kr.g |d< }n|d }d|krLd |d< }n|d }t|�|k}|s�|�|� | jr|| �� d }n|d || f}|d d7 < |S )N�counts� current_countr �linenor )r1 r r0 �lenr- r r ) r � callcountr2 �per_fn� per_platformr4 r5 Z has_count�resultr r r r; a s$ zProfileStatsFile.resultc C s\ t }| j| }|| j }|d }|d }|t|�k rB|||d < n||d< | jrX| �� d S )Nr4 r5 r r )r1 r r0 r7 r r )r r8 r2 r9 r: r4 r5 r r r �replace| s zProfileStatsFile.replacec C s d| j S )Na # %s # This file is written out on a per-environment basis. # For each test in aaa_profiling, the corresponding function and # environment is located within this file. If it doesn't exist, # the test is skipped. # If a callcount does exist, it is compared to what we received. # assertions are raised if the counts do not match. # # To add a new callcount test, apply the function_call_count # decorator and re-run the tests using the --write-profiles # option - this file will be rewritten including the new count. # )r )r r r r �_header� s zProfileStatsFile._headerc C s� yt | j�}W n tk r" d S X x|t|�D ]p\}}|�� }|r.|�d�rNq.|�� \}}}| j| }|| }dd� |�d�D �} | |d< |d |d< d|d < q.W |�� d S ) N�#c S s g | ] }t |��qS r )�int)r% �countr r r r' � s z*ProfileStatsFile._read.<locals>.<listcomp>�,r4 r r6 r r5 ) �openr �IOError� enumerate�strip� startswithr r �close) r � profile_fr6 �liner2 r0 r4 r9 r: �cr r r r � s zProfileStatsFile._readc C s� t d| j � t| jd�}|�| �� � xpt| j�D ]b}| j| }|�d| � xDt|�D ]8}|| }d�dd� |d D ��}|�d|||f � qZW q4W |�� d S ) NzWriting profile file %s�wz # TEST: %s rA c s s | ]}t |�V qd S )N)r$ )r% r@ r r r � <genexpr>� s z*ProfileStatsFile._write.<locals>.<genexpr>r4 z %s %s %s ) �printr rB r r= �sortedr r* rG )r rH r2 r9 r0 r: rJ r r r r � s zProfileStatsFile._writeN) �__name__� __module__� __qualname__�__doc__r! �propertyr0 r3 r; r< r= r r r r r r r ( s r 皙�����?c s � fdd�}|S )a� Assert a target for a test case's function call count. The main purpose of this assertion is to detect changes in callcounts for various functions - the actual number is not as important. Callcounts are stored in a file keyed to Python version and OS platform information. This file is generated automatically for new tests, and versioned so that unexpected changes in callcounts will be detected. c s � �fdd�}t |� �S )Nc s t �d�� � | |�S Q R X d S )N)�variance)�count_functions)�args�kw)�fnrU r r �wrap� s z3function_call_count.<locals>.decorate.<locals>.wrap)r )rY rZ )rU )rY r �decorate� s z%function_call_count.<locals>.decorater )rU r[ r )rU r �function_call_count� s r\ c c s t d krt�d��t�� s0tjs0t�dtj � t� t � � }|� � d V |� � t j |tjd�}|j}t�|�}|d kr�d }n|\}}td||f � |�d� |�� |�rt|| �}t|| �|k}|s�tj�rtjr�t�|� ntd|| d |tjf ��d S )NzcProfile is not installedz�No profiling stats available on this platform for this function. Run tests with --write-profiles to add statistics to %s for this platform.)�streamz Pstats calls: %d Expected %sZ cumulativez�Adjusted function call count %s not within %s%% of expected %s, platform %s. Rerun with --write-profiles to regenerate this callcount.�d )�cProfiler Z_skip_test_exception�_profile_statsr3 r Z skip_testr r ZProfile�enable�disable�pstatsZStatsr+ �stdoutZ total_callsr; rM Z sort_statsZ print_statsr? �absr r< �AssertionErrorr0 ) rU ZprZstatsr8 �expectedZexpected_countZline_noZdeviance�failedr r r rV � s@ rV )rT )rT )rR r � contextlibr rc r+ � r �utilr r r r r r_ � ImportErrorr1 r` �objectr r\ �contextmanagerrV r r r r �<module> s,