관리-도구
편집 파일: util.cpython-37.pyc
B ��4]: � @ s0 d dl Z d dlZd dlZd dlZd dlZd dlZddlmZ ddlmZ ddlm Z ddlm Z ddlm Z e r�dd � Z e Z Zn$e r�d d � Ze Z ZnejZ d d � Zdd� Zdd� ZG dd� de�Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zed d!� �Zd"d#� Z G d$d%� d%e �Z d*d&d'�Z d(d)� Z dS )+� N� )� decorator)� defaultdict)�jython)�py2k)�pypyc G s& t �� t�d� t �� t �� dS )z aggressive gc.collect for tests.g�������?r )�gc�collect�time�sleep)�args� r �J/opt/alt/python37/lib64/python3.7/site-packages/sqlalchemy/testing/util.py�jython_gc_collect s r c G s t �� t �� d S )N)r r )r r r r �pypy_gc_collect$ s r c C s d S )Nr r r r r �lazy_gc- s r c # s� t � } tr6ydd l}| �|� W n tk r4 Y nX dd l}| �|� x8| D ]0� x*td|j�D ]�� j� �fdd�fV q`W qNW d S )Nr ���c s � � | ��S )N)�dumps)�d)�pickle_�protocolr r �<lambda>B � zpicklers.<locals>.<lambda>) �setr �cPickle�add� ImportError�pickle�rangeZHIGHEST_PROTOCOL�loads)�picklersr r r )r r r r 1 s r c C s< t | t�rt| |�S | t�dd| � �tj�td|� S )N�1�0� )� isinstance�float�round�decimalZDecimalZ to_integralZ ROUND_FLOOR�pow)�valueZprecr r r � round_decimalE s r* c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd S )� RandomSetc C s t t�| ��}t�|� t|�S )N)�listr �__iter__�random�shuffle�iter)�self�lr r r r- P s zRandomSet.__iter__c C s4 t �dt| �d �}tt�| ��| }| �|� |S )Nr � )r. �randint�lenr, r r- �remove)r1 �index�itemr r r �popU s z RandomSet.popc C s t t�| |��S )N)r+ r �union)r1 �otherr r r r: [ s zRandomSet.unionc C s t t�| |��S )N)r+ r � difference)r1 r; r r r r< ^ s zRandomSet.differencec C s t t�| |��S )N)r+ r � intersection)r1 r; r r r r= a s zRandomSet.intersectionc C s t | �S )N)r+ )r1 r r r �copyd s zRandomSet.copyN) �__name__� __module__� __qualname__r- r9 r: r< r= r> r r r r r+ O s r+ c C sl t t�}x | D ]\}}|| �|� qW x>t|�D ].\}}x$||d� D ]}||| krHdS qHW q2W dS dS )zATrue if the given sorting conforms to the given partial ordering.NFT)r r r � enumerate)�tuplesZsorted_elements�deps�parent�child�i�node�nr r r �conforms_partial_orderingh s rJ c sB t t��x | D ]\}}�| �|� qW � �fdd�� t� |��S )Nc 3 sf t | �dkrt| �V nJxH| D ]@}t| ��|g�}|��| �s x � |�D ]}|g| V qJW q W d S )Nr3 )r5 r, r r< r= )�elements�elem�subsetZ sub_ordering)�_all_orderings�edgesr r rN { s z-all_partial_orderings.<locals>._all_orderings)r r r r0 )rC rK rE rF r )rN rO r �all_partial_orderingsv s rP c C s> y || _ W n. tk r8 t�| j| j|| j| j�} Y nX | S )ac Return a function with a given __name__. Will assign to __name__ and return the original function if possible on the Python implementation, otherwise a new function will be constructed. This function should be phased out as much as possible in favor of @decorator. Tests that "generate" many named tests should be modernized. )r? � TypeError�types� FunctionType�__code__� __globals__� __defaults__� __closure__)�fn�namer r r �function_named� s rZ c O sZ | � � }y"||f|�|�}| �ddd� |S t�� }| j|� }|sL� n|S Y nX dS )a Run the given function under the given contextmanager, simulating the behavior of 'with' to support older Python versions. This is not necessary anymore as we have placed 2.6 as minimum Python version, however some tests are still using this structure. N)� __enter__�__exit__�sys�exc_info)�ctxrX �arg�kw�obj�resultr^ Zraise_r r r �run_as_contextmanager� s rd c C s dd� | D �S )z�Converts the results of sql execution into a plain set of column tuples. Useful for asserting the results of an unordered query. c S s h | ] }t |��qS r )�tuple)�.0�rowr r r � <setcomp>� s zrowset.<locals>.<setcomp>r )�resultsr r r �rowset� s rj c C s ds t | ��d S )NF)�AssertionError)�msgr r r �fail� s rm c O sp ddl m} ddl m} ddlm} |�|j�}|d }t|dd�}||_z | ||�S |� ||j� ||_X dS )z>Provide bound MetaData for a single test, dropping afterwards.r3 )�config)�enginesr )�schema�metadataN) � rn ro � sqlalchemyrp �MetaData�db�getattrrq �drop_all_tables) rX r ra rn ro rp rq r1 Z prev_metar r r �provide_metadata� s rx c s0 ddl m� ddlm� t� ��fdd��}|S )ziForce the given table names to be dropped after test complete, isolating for foreign key cycles r3 )rn r )�inspectc s( z | ||�S t � j�� j��d� X d S )N)� include_names)rw ru )rX r ra )rn ry �namesr r �go� s z force_drop_names.<locals>.go)rr rn rs ry r )r{ r| r )rn ry r{ r �force_drop_names� s r} c @ s$ e Zd ZdZdd� Zdd� ZeZdS )�adictz,Dict keys available as attributes. Shadows.c C s* y| | S t k r$ t�| |�S X d S )N)�KeyError�dict�__getattribute__)r1 �keyr r r r� � s zadict.__getattribute__c s t � fdd�|D ��S )Nc s g | ] }� | �qS r r )rf r� )r1 r r � <listcomp>� s z"adict.__call__.<locals>.<listcomp>)re )r1 �keysr )r1 r �__call__� s zadict.__call__N)r? r@ rA �__doc__r� r� �get_allr r r r r~ � s r~ c C s ddl m}m}m}m}m} ddlm} m} |d k r<t |�}| � � ��} x�t |j |d��D ]�\} } | r�|d k rv| |krvqX| � | || |� |d��� qX| rX| jjs�qXxf| D ]^\} }|d k r�| |kr�q�|| |� |d|�|d|�|d�}| � | ||jjg|jjg|d��� q�W qXW W d Q R X d S )Nr )�Column�Table�Integerrt �ForeignKeyConstraint)� DropTable�DropConstraint)rp �x�y)rY )rs r� r� r� rt r� Zsqlalchemy.schemar� r� r �connect�reversedZ get_sorted_table_and_fkc_names�executeZdialectZsupports_alter�cr� r� )ZengineZ inspectorrp rz r� r� r� rt r� r� r� ZconnZtnameZfkcsZfkc�tbr r r rw � s8 rw c s t � fdd��}|S )Nc s z | ||�S � � � X d S )N)�_clear)rX r` ra )� event_clsr r �decorate' s z!teardown_events.<locals>.decorate)r )r� r� r )r� r �teardown_events& s r� )NN)!r' r r. r] r rR �utilr r r r r r Z gc_collectr r r r r* r r+ rJ rP rZ rd rj rm rx r} r� r~ rw r� r r r r �<module> s@ .