File: //opt/alt/python37/lib64/python3.7/site-packages/sqlalchemy/testing/__pycache__/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�timeZsleep)�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__�randomZshuffle�iter)�self�lr r r
r, P s
zRandomSet.__iter__c C s4 t �dt| �d �}tt�| ��| }| �|� |S )Nr � )r- Zrandint�lenr+ r r, �remove)r/ �index�itemr r r
�popU s
z
RandomSet.popc C s t t�| |��S )N)r* r �union)r/ �otherr r r
r7 [ s zRandomSet.unionc C s t t�| |��S )N)r* r �
difference)r/ r8 r r r
r9 ^ s zRandomSet.differencec C s t t�| |��S )N)r* r � intersection)r/ r8 r r r
r: a s zRandomSet.intersectionc C s t | �S )N)r* )r/ r r r
�copyd s zRandomSet.copyN) �__name__�
__module__� __qualname__r, r6 r7 r9 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_elementsZdeps�parent�child�iZnode�nr r r
�conforms_partial_orderingh s rE 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 )Nr1 )r2 r+ r r9 r: )�elements�elemZsubsetZ sub_ordering)�_all_orderings�edgesr r
rH { s
z-all_partial_orderings.<locals>._all_orderings)r r r r. )r@ rF rA rB r )rH rI r
�all_partial_orderingsv s
rJ 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
rT 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)ZctxrR �arg�kw�obj�resultrX Zraise_r r r
�run_as_contextmanager� s
r] 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 rc c C s ds t | ��d S )NF)�AssertionError)�msgr r r
�fail� s rf 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.r1 )�config)�enginesr )�schema�metadataN)
� rg rh �
sqlalchemyri �MetaData�db�getattrrj �drop_all_tables) rR r rZ rg rh ri rj r/ Z prev_metar r r
�provide_metadata� s
rq c s0 ddl m� ddlm� t� ��fdd��}|S )ziForce the given table names to be dropped after test complete,
isolating for foreign key cycles
r1 )rg r )�inspectc s( z
| ||�S t � j�� j��d� X d S )N)�
include_names)rp rn )rR r rZ )rg rr �namesr r
�go� s
z force_drop_names.<locals>.go)rk rg rl rr r )rt ru r )rg rr rt r
�force_drop_names� s rv 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__)r/ �keyr r r
rz � s zadict.__getattribute__c s t � fdd�|D ��S )Nc s g | ] }� | �qS r r )r_ r{ )r/ r r
�
<listcomp>� s z"adict.__call__.<locals>.<listcomp>)r^ )r/ �keysr )r/ r
�__call__� s zadict.__call__N)r< r= r> �__doc__rz r~ Zget_allr r r r
rw � s rw 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�Integerrm �ForeignKeyConstraint)� DropTable�DropConstraint)ri �x�y)rS )rl r� r� r� rm r� Zsqlalchemy.schemar� r� r Zconnect�reversedZ get_sorted_table_and_fkc_namesZexecuteZdialectZsupports_alter�cr� r� )ZengineZ inspectorri rs r� r� r� rm r� r� r� ZconnZtnameZfkcsZfkc�tbr r r
rp � s8
rp c s t � fdd��}|S )Nc s z
| ||�S � � � X d S )N)Z_clear)rR rY rZ )� event_clsr r
�decorate' s
z!teardown_events.<locals>.decorate)r )r� r� r )r� r
�teardown_events& s r� )NN)!r& r r- rW r
rL �utilr r r r r r Z
gc_collectr r r r r) r r* rE rJ rT r] rc rf rq rv ry rw rp r� r r r r
�<module> s@
.