관리-도구
편집 파일: test_select.cpython-38.pyc
U -?�f$� � @ s d dl mZ d dlZddlmZ ddlmZ ddlmZ ddlmZ ddl m Z ddl m Z dd l m Z dd l mZ dd lmZ dd lmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlm Z d dlm Z d dlm Z d dlm Z d d lm Z d d lm!Z! d d lm"Z" d d lm#Z# d d lm$Z$ d d!lm%Z% d d"lm&Z& d d#lm'Z' d d$lm(Z( d d%l)m*Z* d d&l)m+Z+ G d'd(� d(ej,�Z-G d)d*� d*ej,�Z.G d+d,� d,ej/�Z0G d-d.� d.ej,�Z1G d/d0� d0ej,�Z2G d1d2� d2ej,�Z3G d3d4� d4ej,�Z4G d5d6� d6eeej,�Z5G d7d8� d8ej,�Z6G d9d:� d:ej,�Z7G d;d<� d<ej,�Z8G d=d>� d>ej,�Z9G d?d@� d@ej,�Z:G dAdB� dBej,�Z;G dCdD� dDeej,�Z<G dEdF� dFej,�Z=G dGdH� dHej,�Z>G dIdJ� dJej,�Z?dS )K� N� )�AssertsCompiledSQL)�AssertsExecutionResults)�config)�fixtures)� assert_raises)�eq_)�in_)� CursorSQL)�Column)�Table� )� bindparam)�case)�column)�Computed)�exists)�false)� ForeignKey)�func)�Identity)�Integer)�literal)�literal_column)�null)�select)�String)�table)�testing)�text)�true)�tuple_)� TupleType)�union)�values)� DatabaseError)�ProgrammingErrorc @ s@ e Zd ZdZedd� �Zedd� �Zdd� Zej j dd � �Z d S ) � CollateTestTc C s& t d|tdtdd�tdtd��� d S )N� some_table�idT�� primary_key�data�d �r r r r ��cls�metadata� r2 �T/opt/hc_python/lib64/python3.8/site-packages/sqlalchemy/testing/suite/test_select.py� define_tables4 s �zCollateTest.define_tablesc C s( |� | jj�� ddd�ddd�g� d S )N� � collate data1�r) r, r � collate data2��execute�tablesr( �insert�r0 � connectionr2 r2 r3 � insert_data= s ��zCollateTest.insert_datac C s. t j�� �}t|�|��� |� W 5 Q R X d S �N�r �db�connectr r: �fetchall��selfr �result�connr2 r2 r3 �_assert_resultG s zCollateTest._assert_resultc C sB t j�t j�}| �t| jj��| jjj j � |�� � �ddg� d S )N)r5 r6 )r r8 ) r �requiresZget_order_by_collationr rI r r; r( �order_by�cr, Zcollate�asc)rF Z collationr2 r2 r3 �test_collate_order_byK s ��z!CollateTest.test_collate_order_byN) �__name__� __module__� __qualname__� __backend__� classmethodr4 r? rI r rJ Zorder_by_collationrN r2 r2 r2 r3 r' 1 s r' c @ sl e Zd ZdZdZedd� �Zedd� �Zdd� Zd d � Z d d � Z d d� Z dd� Z dd� Z ejjdd� �ZdS )�OrderByLabelTestz�Test the dialect sends appropriate ORDER BY expressions when labels are used. This essentially exercises the "supports_simple_order_by_label" setting. Tc C sB t d|tdtdd�tdt�tdt�tdtd��td td��� d S ) Nr( r) Tr* �x�y�q�2 �pr. r/ r2 r2 r3 r4 b s �z OrderByLabelTest.define_tablesc C sB |� | jj�� dddddd�dddddd�ddd d d d�g� d S ) Nr5 r �q1Zp3)r) rU rV rW rY r �q2Zp2� Zq3�p1r9 r= r2 r2 r3 r? n s ��z OrderByLabelTest.insert_datac C s. t j�� �}t|�|��� |� W 5 Q R X d S r@ rA rE r2 r2 r3 rI y s z OrderByLabelTest._assert_resultc C s6 | j j}|jj�d�}| �t|��|�dddg� d S )N�lx�r5 �r �r )r; r( rL rU �labelrI r rK �rF r r^ r2 r2 r3 � test_plain} s zOrderByLabelTest.test_plainc C s> | j j}|jj|jj �d�}| �t|��|�dddg� d S )Nr^ ra �� �� ) r; r( rL rU rV rb rI r rK rc r2 r2 r3 �test_composed_int� s z"OrderByLabelTest.test_composed_intc C sb | j j}|jj|jj �d�}t�|jj�|jj �d�}| � t ||�� ||� � �dddg� d S )Nr^ �ly)r Zq1p3)rf Zq2p2)rh Zq3p1)r; r( rL rU rV rb r �lowerrW rY rI r rK �desc)rF r r^ rj r2 r2 r3 �test_composed_multiple� s �z'OrderByLabelTest.test_composed_multiplec C s: | j j}|jj�d�}| �t|��|�� �dddg� d S )Nr^ ra r` r_ ) r; r( rL rU rb rI r rK rl rc r2 r2 r3 �test_plain_desc� s z OrderByLabelTest.test_plain_descc C sB | j j}|jj|jj �d�}| �t|��|� � �dddg� d S )Nr^ rg re ra ) r; r( rL rU rV rb rI r rK rl rc r2 r2 r3 �test_composed_int_desc� s z'OrderByLabelTest.test_composed_int_descc C sT | j j}|jj|jj �d�}tt�|jj �|�� |�� |�}| � |dddg� d S )Nr^ )r5 r )r5 rf )r5 rh ) r; r( rL rU rV rb r r �countr) Zgroup_byrK rI )rF r �expr�stmtr2 r2 r3 �test_group_by_composed� s �z'OrderByLabelTest.test_group_by_composedN)rO rP rQ �__doc__rR rS r4 r? rI rd ri rm rn ro r rJ Zgroup_by_complex_expressionrs r2 r2 r2 r3 rT W s rT c @ s e Zd ZdZdZdd� ZdS )�ValuesExpressionTest)Ztable_value_constructorTc C sH t tdt�tdt�dd��dddg�}t|�t|���� dddg� d S )Nr) �nameZ my_values)rv )r5 Zname1)r Zname2)r Zname3) r$ r r r r, r r: r �all)rF r> Z value_exprr2 r2 r3 � test_tuples� s ���z ValuesExpressionTest.test_tuplesN)rO rP rQ � __requires__rR rx r2 r2 r2 r3 ru � s ru c @ s8 e Zd ZdZedd� �Zedd� �ZdIdd �ZdJd d �Zd d � Z dd� Z e j j dd� �Ze j jdd� �Ze jdddgdddgdddgdd�e j jdd� ��Ze j j d d � �Ze j jd d � �Ze j jd d!� �Ze j jd"d#� �Ze j j d$d%� �Ze j jd&d'� �Ze j jd(d)� �Ze j jd*d+� �Z e j j d,d-� �Z e j j d.d/� �Z e j j d0d1� �Z e j j d2d3� �Z!e j j e j j"d4d5� ��Z#e j j d6d7� �Z$e j j d8d9� �Z%e j j&d:d;� �Z'e j j&e j j(d<d=� ��Z)e j j&e j j(d>d?� ��Z*e j j+d@dA� �Z,e j j+e j j(dBdC� ��Z-e j j&e j j+dDdE� ��Z.e j j&e j j+e j j(dFdG� ���Z/dHS )K�FetchLimitOffsetTestTc C s* t d|tdtdd�tdt�tdt�� d S �Nr( r) Tr* rU rV �r r r r/ r2 r2 r3 r4 � s �z"FetchLimitOffsetTest.define_tablesc C sJ |� | jj�� dddd�dddd�dddd�dddd�dddd�g� d S )Nr5 r �r) rU rV r r\ rf � r9 r= r2 r2 r3 r? � s ��z FetchLimitOffsetTest.insert_datar2 Fc C sT |r:|� ||��� }tt|�t|�� tt|�t|�� nt|� ||��� |� d S r@ )r: rD r �len�set)rF r> r rG �params�set_Z query_resr2 r2 r3 rI � s z#FetchLimitOffsetTest._assert_resultc C s0 t j�� � }t|�||��� |� W 5 Q R X d S r@ )r rB rC r Zexec_driver_sqlrD �rF r rG r� rH r2 r2 r3 �_assert_result_str� s z'FetchLimitOffsetTest._assert_result_strc C sP | j j}t|��|jj�}| �||�d�ddg� | �||�d�dddg� d S �Nr �r5 r5 r �r r r r �r r r\ )r; r( r rK rL r) rI �limit)rF r> r rr r2 r2 r3 �test_simple_limit� s ��z&FetchLimitOffsetTest.test_simple_limitc C sL | j j}t|jj��d��� }tt|�t|���� �� }| � ||dg� d S )Nr5 r_ ) r; r( r rL r) r� Zscalar_subqueryr# �subqueryrI )rF r> r rr �ur2 r2 r3 � test_limit_render_multiple_times� s ��z5FetchLimitOffsetTest.test_limit_render_multiple_timesc C sZ | j j}| �|t|��|jj��d�ddg� | �|t|��|jj��d�dddg� d S r� �r; r( rI r rK rL r) �fetch�rF r> r r2 r2 r3 �test_simple_fetch� s ��z&FetchLimitOffsetTest.test_simple_fetchc C sZ | j j}| �|t|��|jj��d�dddg� | �|t|��|jj��d�ddg� d S )Nr r� �r\ r\ rf �rf r\ r~ r �r; r( rI r rK rL r) �offsetr� r2 r2 r3 �test_simple_offset s ��z'FetchLimitOffsetTest.test_simple_offset)r r �r r5 )r r )r r5 �cases��argnamesc C sl | j j}|ji d�}dddddg}|D ]@\}}|||| � }| �|t|��|jj��|�� |�|� q&d S )N)Zcompiled_cacher� r� r� r� r� ) r; r( Zexecution_optionsrI r rK rL r) r� r� )rF r> r� r Z assert_datar� r� �expectedr2 r2 r3 �test_simple_limit_offset s �z-FetchLimitOffsetTest.test_simple_limit_offsetc C sf | j j}| �|t|��|jj��d��d�ddg� | �|t|��|jj��d��d�dddg� d S )Nr r5 r� r� r r� r� � r; r( rI r rK rL r) r� r� r� r2 r2 r3 �test_simple_fetch_offset* s � �z-FetchLimitOffsetTest.test_simple_fetch_offsetc C s2 | j j}| j|t|��d�dddddgdd� d S ) N� r� r� r� r� r� T�r� )r; r( rI r r� r� r2 r2 r3 �test_fetch_offset_no_order9 s �z/FetchLimitOffsetTest.test_fetch_offset_no_orderc C sb | j j}| �|t|��|jj��d�dddddg� | �|t|��|jj��d�ddddg� d S )Nr r� r� r� r� r� r5 r� r� r2 r2 r3 �test_simple_offset_zeroC s � �z,FetchLimitOffsetTest.test_simple_offset_zeroc C sX | j j}t|��|jj��d��d�}|jt j j ddid�}t |�}| � |ddg� dS � z7test that 'literal binds' mode works - no bound params.r r5 � literal_bindsT)�dialect�compile_kwargsr� r� N)r; r( r rK rL r) r� r� �compiler rB r� �strr� �rF r rr �sqlr2 r2 r3 �test_limit_offset_nobindsR s �z.FetchLimitOffsetTest.test_limit_offset_nobindsc C sX | j j}t|��|jj��d��d�}|jt j j ddid�}t |�}| � |ddg� dS r� )r; r( r rK rL r) r� r� r� r rB r� r� r� r� r2 r2 r3 �test_fetch_offset_nobinds_ s �z.FetchLimitOffsetTest.test_fetch_offset_nobindsc C sr | j j}| j|t|��|jj��td��ddgddid� | j|t|��|jj��td��dddgddid� d S )N�lr� r� r �r� r� r ) r; r( rI r rK rL r) r� r r� r2 r2 r3 �test_bound_limitl s ��z%FetchLimitOffsetTest.test_bound_limitc C sv | j j}| j|t|��|jj��td��dddgddid� | j|t|��|jj��td��ddddgddid� d S ) N�or� r� r� r r� r� r5 ) r; r( rI r rK rL r) r� r r� r2 r2 r3 �test_bound_offset} s � �z&FetchLimitOffsetTest.test_bound_offsetc C s� | j j}| j|t|��|jj��td��� td��ddgddd�d� | j|t|��|jj��td��� td��dd d gd dd�d� d S ) Nr� r� r� r� r r5 )r� r� r� r� r� r ) r; r( rI r rK rL r) r� r r� r� r2 r2 r3 �test_bound_limit_offset� s2 ���� ����z,FetchLimitOffsetTest.test_bound_limit_offsetc C s� | j j}| j|t|��|jj��td��� td��ddgddd�d� | j|t|��|jj��td��� td��dd d gd dd�d� d S ) N�fr� r� r� r r5 )r� r� r� r� r� r ) r; r( rI r rK rL r) r� r r� r� r2 r2 r3 �test_bound_fetch_offset� s2 ���� ����z,FetchLimitOffsetTest.test_bound_fetch_offsetc C s> | j j}| �|t|��|jj��td�td� �ddg� d S )N�1�2r� r� ) r; r( rI r rK rL r) r� r r� r2 r2 r3 �test_expr_offset� s ���z%FetchLimitOffsetTest.test_expr_offsetc C s@ | j j}| �|t|��|jj��td�td� �dddg� d S )Nr� r� r� r� r� ) r; r( rI r rK rL r) r� r r� r2 r2 r3 �test_expr_limit� s ���z$FetchLimitOffsetTest.test_expr_limitc C sP | j j}| �|t|��|jj��td�td� �� td�td� �ddg� d S �Nr� r� r� � r; r( rI r rK rL r) r� r r� r� r2 r2 r3 �test_expr_limit_offset� s ����z+FetchLimitOffsetTest.test_expr_limit_offsetc C sP | j j}| �|t|��|jj��td�td� �� td�td� �ddg� d S r� ) r; r( rI r rK rL r) r� r r� r� r2 r2 r3 �test_expr_fetch_offset� s ����z+FetchLimitOffsetTest.test_expr_fetch_offsetc C s~ | j j}| �|t|��|jj��d��t d�t d� �ddg� | �|t|��|jj��d��t d�t d� �dddg� d S )Nr r� r� r� r r� ) r; r( rI r rK rL r) r� r� r r� r2 r2 r3 � test_simple_limit_expr_offset� s. ���� ����z2FetchLimitOffsetTest.test_simple_limit_expr_offsetc C s| | j j}| �|t|��|jj��td�td� �� d�ddg� | �|t|��|jj��td�td� �� d�ddg� d S )Nr� r r� r� r5 r� r� r� r2 r2 r3 � test_expr_limit_simple_offset s. ���� ����z2FetchLimitOffsetTest.test_expr_limit_simple_offsetc C sr | j j}| j|t|��|jj�� �jddd�ddgdd� | j|t|��|jj�� �jddd�dddgdd� d S ) Nr5 T�� with_tiesr� r� r� r r� � r; r( rI r rK rL rU rl r� r� r2 r2 r3 �test_simple_fetch_ties s � �z+FetchLimitOffsetTest.test_simple_fetch_tiesc C sZ | j j}|�t|��|jj�jddd��d��� � }t |d d� t t |�dddh� d S )Nr Tr� r r� r� r� � r; r( r: r rK rL rU r� r� rD r r� �rF r> r �far2 r2 r3 �test_fetch_offset_ties& s � ���z+FetchLimitOffsetTest.test_fetch_offset_tiesc C sl | j j}| �|t|��|jj�jddd��d�ddg� | �|t|��|jj�jddd��d�dd g� d S ) Nr Tr� r5 r� r� r r� r� ) r; r( rI r rK rL rU r� r� r� r2 r2 r3 �#test_fetch_offset_ties_exact_number3 s2 � ��� � ���z8FetchLimitOffsetTest.test_fetch_offset_ties_exact_numberc C s4 | j j}| �|t|��|jj�jddd�dg� d S )N� T��percentr� r� r� r2 r2 r3 �test_simple_fetch_percentI s �z.FetchLimitOffsetTest.test_simple_fetch_percentc C s< | j j}| �|t|��|jj�jddd��d�ddg� d S )N�( Tr� r5 r� r� r� r� r2 r2 r3 �test_fetch_offset_percentR s � ���z.FetchLimitOffsetTest.test_fetch_offset_percentc C s@ | j j}| j|t|��|jj�� �jdddd�ddgdd� d S )Nr� T�r� r� r� r� r� r� r� r2 r2 r3 � test_simple_fetch_percent_ties_ s � ��z3FetchLimitOffsetTest.test_simple_fetch_percent_tiesc C s\ | j j}|�t|��|jj�jdddd��d��� � }t |d d� t t |�dddh� d S ) Nr� Tr� r r r� r� r� r� r� r2 r2 r3 � test_fetch_offset_percent_tiesl s � ���z3FetchLimitOffsetTest.test_fetch_offset_percent_tiesN)r2 F)r2 )0rO rP rQ rR rS r4 r? rI r� r� r� r rJ Z fetch_firstr� r� r� � combinationsr� r� Zfetch_no_order_byr� r� r� r� Zbound_limit_offsetr� r� r� r� Zsql_expression_limit_offsetr� r� r� Zfetch_expressionr� r� r� Z fetch_tiesr� Zfetch_offset_with_optionsr� r� Z fetch_percentr� r� r� r� r2 r2 r2 r3 rz � s� � � rz c @ sH e Zd ZdZdZdZedd� �Zedd� �Zdd � Z d d � Z d d � Z dS )�SameNamedSchemaTableTestztests for #7471T)Zschemasc C sB t d|tdtdd�tjd� t d|tdtdd�tdtdd�� d S ) Nr( r) Tr* )�schema� some_table_idF�Znullable)r r r r � test_schemar/ r2 r2 r3 r4 � s � ��z&SameNamedSchemaTableTest.define_tablesc C sD | � ddtj �\}}|�|�� ddi� |�|�� ddd�� d S )Nr( � %s.some_tabler) r5 )r) r� )r; r r� r: r< )r0 r> r( �some_table_schemar2 r2 r3 r? � s �z$SameNamedSchemaTableTest.insert_datac C sJ | � ddtj �\}}t|�t||��|||jj|jj k��� � d� d S )Nr( r� �r5 r5 r5 ) r; r r� r r: r � join_fromrL r� r) �first�rF r> r( r� r2 r2 r3 � test_simple_join_both_tables� s � ���z5SameNamedSchemaTableTest.test_simple_join_both_tablesc C sV | � ddtj �\}}t|�t|��|||jj|jj k�� |jj dk��� � d� d S )Nr( r� r5 �r5 r5 ) r; r r� r r: r r� rL r� r) �wherer� r� r2 r2 r3 �!test_simple_join_whereclause_only� s �� �� �z:SameNamedSchemaTableTest.test_simple_join_whereclause_onlyc C s� | � ddtj �\}}t|��|||jj|jjk��|jjdk�� � }t |� t||jj��|||jj|jjk��|jjdk��� � d� d S )Nr( r� r5 r� ) r; r r� r r� rL r� r) r� r� r r: r� )rF r> r( r� Zsubqr2 r2 r3 � test_subquery� s0 �� �� � �� �z&SameNamedSchemaTableTest.test_subqueryN) rO rP rQ rt rR ry rS r4 r? r� r� r� r2 r2 r2 r3 r� { s r� c @ sZ e Zd ZdZddd�Zedd� �Zedd� �Zd d � Zd d � Z d d� Z dd� Z dd� Z dS )�JoinTestTr2 c C s0 t j�� � }t|�||��� |� W 5 Q R X d S r@ rA r� r2 r2 r3 rI � s zJoinTest._assert_resultc C s@ t d|tdtdd�� t d|tdtdd�tdtd�dd �� d S ) N�ar) Tr* �b�a_idza.idFr� )r r r r r/ r2 r2 r3 r4 � s �zJoinTest.define_tablesc C sj |� | jj�� ddiddiddiddiddig� |� | jj�� ddd�ddd�ddd�ddd�g� d S )Nr) r5 r r r\ rf )r) r� )r: r; r� r< r� r= r2 r2 r3 r? � s � ��zJoinTest.insert_datac C sN | � dd�\}}t||��|�|���|jj|jj�}| �|ddddg� d S �Nr� r� r� )r5 r r5 )r r\ r )r rf r �r; r � select_from�joinrK rL r) rI �rF r� r� rr r2 r2 r3 �test_inner_join_fk s &zJoinTest.test_inner_join_fkc C sn | � dd�\}}t||��|�|t� ���|jj|jj�}| �|dd� t � ddddd gd d d d g�D �� d S )Nr� r� c S s g | ]\\}\}}|||f�qS r2 r2 )�.0r� r� rL r2 r2 r3 � <listcomp> s �z1JoinTest.test_inner_join_true.<locals>.<listcomp>r_ r` ra �r\ re r� r� )r\ r )rf r ) r; r r� r� r rK rL r) rI � itertools�productr� r2 r2 r3 �test_inner_join_true s" � �� ���z JoinTest.test_inner_join_truec C sJ | � dd�\}}t||��|�|t� ���|jj|jj�}| �|g � d S )Nr� r� ) r; r r� r� r rK rL r) rI r� r2 r2 r3 �test_inner_join_false s � ��z JoinTest.test_inner_join_falsec C sT | � dd�\}}t||��|�|t� ���|jj|jj�}| �|dddddg� d S )Nr� r� )r5 NN)r NN)r NN)r\ NN)rf NN) r; r r� Z outerjoinr rK rL r) rI r� r2 r2 r3 �test_outer_join_false& s" � ����z JoinTest.test_outer_join_falsec C sN | � dd�\}}t||��|�|���|jj|jj�}| �|ddddg� d S r� r� r� r2 r2 r3 �test_outer_join_fk: s &zJoinTest.test_outer_join_fkN)r2 ) rO rP rQ rR rI rS r4 r? r� r� r� r� r� r2 r2 r2 r3 r� � s r� c @ s� e Zd ZdZedd� �Zedd� �Zddd�Zd d � Zd d � Z e j j e j j d d� ��Ze j jdd� �Zdd� Ze j j dd� �Zdd� ZdS )�CompoundSelectTestTc C s* t d|tdtdd�tdt�tdt�� d S r{ r| r/ r2 r2 r3 r4 E s �z CompoundSelectTest.define_tablesc C s@ |� | jj�� dddd�dddd�dddd�dddd�g� d S )Nr5 r r} r r\ rf r9 r= r2 r2 r3 r? O s ��z CompoundSelectTest.insert_datar2 c C s0 t j�� � }t|�||��� |� W 5 Q R X d S r@ rA r� r2 r2 r3 rI [ s z!CompoundSelectTest._assert_resultc C s\ | j j}t|��|jjdk�}t|��|jjdk�}t||�}| �|�|j j�ddg� d S �Nr r r� r� ) r; r( r r� rL r) r# rI rK �selected_columns�rF r �s1�s2�u1r2 r2 r3 �test_plain_union_ s �z#CompoundSelectTest.test_plain_unionc C sd | j j}t|��|jjdk�}t|��|jjdk�}t||��� �� }| �|� |j j�ddg� d S r� ) r; r( r r� rL r) r# �aliasrI rK r� r� r2 r2 r3 � test_select_from_plain_unioni s �z/CompoundSelectTest.test_select_from_plain_unionc C s� | j j}t|��|jjdk��d��|jj�}t|��|jjdk��d��|jj�}t||��d�}| � |�|j j�ddg� d S �Nr r5 r r� r� ) r; r( r r� rL r) r� rK r# rI r� r� r2 r2 r3 �&test_limit_offset_selectable_in_unionss s && �z9CompoundSelectTest.test_limit_offset_selectable_in_unionsc C sv | j j}t|��|jjdk��|jj�}t|��|jjdk��|jj�}t||��d�}| � |�|j j�ddg� d S r� ) r; r( r r� rL r) rK r# r� rI r� r� r2 r2 r3 �"test_order_by_selectable_in_unions s �z5CompoundSelectTest.test_order_by_selectable_in_unionsc C sj | j j}t|��|jjdk��� }t|��|jjdk��� }t||��d�}| � |� |j j�ddg� d S r� ) r; r( r r� rL r) �distinctr# r� rI rK r� r� r2 r2 r3 �"test_distinct_selectable_in_unions� s �z5CompoundSelectTest.test_distinct_selectable_in_unionsc C s� | j j}t|��|jjdk��d��|jj�}t|��|jjdk��d��|jj�}t||�� � }| � |�� �d��|jj�ddg� d S r� ) r; r( r r� rL r) r� rK r# r� rI r� r2 r2 r3 �&test_limit_offset_in_unions_from_alias� s && �z9CompoundSelectTest.test_limit_offset_in_unions_from_aliasc C s� | j j}t|��|jjdk��d��|jj��� �� }t|��|jjdk��d��|jj��� �� }t ||��d�}| � |�|j j�ddg� d S r� ) r; r( r r� rL r) r� rK r� r# rI r� r� r2 r2 r3 �.test_limit_offset_aliased_selectable_in_unions� s, ��� � ��� � �zACompoundSelectTest.test_limit_offset_aliased_selectable_in_unionsN)r2 )rO rP rQ rR rS r4 r? rI r� r� r rJ Zorder_by_col_from_unionZ/parens_in_union_contained_select_w_limit_offsetr Z0parens_in_union_contained_select_wo_limit_offsetr r r r r2 r2 r2 r3 r� B s" r� c @ sl e Zd ZdZdZedd� �Zedd� �Zdd� Zd d � Z d d � Z d d� Z e j jdd� �Ze j jdd� �ZdS )�PostCompileParamsTestT)Zstandard_cursor_sqlc C s6 t d|tdtdd�tdt�tdt�tdtd��� d S � Nr( r) Tr* rU rV �zrX r. r/ r2 r2 r3 r4 � s �z#PostCompileParamsTest.define_tablesc C sH |� | jj�� ddddd�ddddd�ddddd�ddd d d�g� d S � Nr5 r �z1)r) rU rV r r �z2r\ �z3rf �z4r9 r= r2 r2 r3 r? � s ��z!PostCompileParamsTest.insert_datac C s< | j j}t|jj��|jjtddd�k�}| �|di � d S )NrW T��literal_executezKSELECT some_table.id FROM some_table WHERE some_table.x = __[POSTCOMPILE_q]� r; r( r rL r) r� rU r �assert_compile�rF r rr r2 r2 r3 � test_compile� s ��z"PostCompileParamsTest.test_compilec C sB | j j}t|jj��|jjtdddd�k�}| j|di dd� d S )NrW r� Tr z<SELECT some_table.id FROM some_table WHERE some_table.x = 10)r� r r r2 r2 r3 �test_compile_literal_binds� s ��z0PostCompileParamsTest.test_compile_literal_bindsc C s� | j j}t|jj��|jjtddd�k�}| �� �.}t j � � �}|� |t dd�� W 5 Q R X W 5 Q R X |�tdt j jjr|dni �� d S )NrW Tr r� �rW z>SELECT some_table.id FROM some_table WHERE some_table.x = 10r2 )r; r( r rL r) r� rU r �sql_execution_asserterr rB rC r: �dict�assert_r r� � positional�rF r rr ZasserterrH r2 r2 r3 � test_execute� s � &��z"PostCompileParamsTest.test_executec C s� | j j}t|jj��|jj�tdddd���}| � � �4}t j � � � }|� |tdddgd�� W 5 Q R X W 5 Q R X |�tdt j jjr�d ni �� d S ) NrW T�� expandingr rf r~ rh r zFSELECT some_table.id FROM some_table WHERE some_table.x IN (5, 6, 7)r2 )r; r( r rL r) r� rU r r r r rB rC r: r r r r� r r r2 r2 r3 �+test_execute_expanding_plus_literal_execute s � ,��zAPostCompileParamsTest.test_execute_expanding_plus_literal_executec C s� | j j}t|jj��t|jj|jj�� t dddd���}| � � �2}t j �� � }|�|tddgd�� W 5 Q R X W 5 Q R X |�tdt j jjr�dnd t j jjr�d ni �� d S ) NrW Tr )rf r� )� � r zbSELECT some_table.id FROM some_table WHERE (some_table.x, some_table.y) IN (%s(5, 10), (12, 18))�VALUES � r2 )r; r( r rL r) r� r! rU rV r r r r rB rC r: r r r r� �tuple_in_valuesr r r2 r2 r3 �1test_execute_tuple_expanding_plus_literal_execute s" �� *���zGPostCompileParamsTest.test_execute_tuple_expanding_plus_literal_executec C s� | j j}t|jj��t|jj|jj�� t dddd���}| � � �2}t j �� � }|�|tddgd�� W 5 Q R X W 5 Q R X |�tdt j jjr�dnd t j jjr�d ni �� d S ) NrW Tr )rf r )r r r zfSELECT some_table.id FROM some_table WHERE (some_table.x, some_table.z) IN (%s(5, 'z1'), (12, 'z3'))r! r" r2 )r; r( r rL r) r� r! rU r r r r r rB rC r: r r r r� r# r r r2 r2 r3 �?test_execute_tuple_expanding_plus_literal_heterogeneous_execute1 s" �� *���zUPostCompileParamsTest.test_execute_tuple_expanding_plus_literal_heterogeneous_executeN)rO rP rQ rR ry rS r4 r? r r r r r rJ �tuple_inr$ r% r2 r2 r2 r3 r � s r c @ sr e Zd ZdZedd� �Zedd� �ZdDdd�Zd d � Zd d � Z e j j d d� �Z e j j dd� �Ze j j dd� �Ze j j dd� �Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Ze j jd!d"� �Ze j jd#d$� �Ze j jd%d&� �Ze j jd'd(� �Ze j jd)d*� �Z e j jd+d,� �Z e j jd-d.� �Z d/d0� Z d1d2� Z d3d4� Z!d5d6� Z"d7d8� Z#d9d:� Z$d;d<� Z%d=d>� Z&d?d@� Z'dAdB� Z(dCS )E�ExpandingBoundInTestTc C s6 t d|tdtdd�tdt�tdt�tdtd��� d S r r. r/ r2 r2 r3 r4 M s �z"ExpandingBoundInTest.define_tablesc C sH |� | jj�� ddddd�ddddd�ddddd�ddd d d�g� d S r r9 r= r2 r2 r3 r? X s ��z ExpandingBoundInTest.insert_datar2 c C s0 t j�� � }t|�||��� |� W 5 Q R X d S r@ rA r� r2 r2 r3 rI d s z#ExpandingBoundInTest._assert_resultc C s` | j j}t|jj��|jj�td����|jj �td���� |jj�}| j |g g g d�d� d S )NrW rY )rW rY r� ) r; r( r rL r) r� rU r r rV rK rI r r2 r2 r3 �"test_multiple_empty_sets_bindparamh s ����z7ExpandingBoundInTest.test_multiple_empty_sets_bindparamc C sN | j j}t|jj��|jj�g ���|jj�g ��� |jj�}| � |g � d S r@ ) r; r( r rL r) r� rU r rV rK rI r r2 r2 r3 � test_multiple_empty_sets_directt s � ���z4ExpandingBoundInTest.test_multiple_empty_sets_directc C sT | j j}t|jj��t|jj|jj�� t d���� |jj�}| j |g dg id� d S �NrW r� � r; r( r rL r) r� r! rU r r r rK rI r r2 r2 r3 �)test_empty_heterogeneous_tuples_bindparam� s ���z>ExpandingBoundInTest.test_empty_heterogeneous_tuples_bindparamc sD � j j�� �fdd�}|g g � |dddgdddg� |g g � d S ) Nc s@ t �jj��t�jj�jj��| ����jj�}� � ||� d S r@ ) r rL r) r� r! rU r r rK rI ��valr� rr �rF r r2 r3 �go� s ���zGExpandingBoundInTest.test_empty_heterogeneous_tuples_direct.<locals>.go�r r �r r �r\ r r` ra r� �r; r( �rF r0 r2 r/ r3 �&test_empty_heterogeneous_tuples_direct� s z;ExpandingBoundInTest.test_empty_heterogeneous_tuples_directc C sT | j j}t|jj��t|jj|jj�� t d���� |jj�}| j |g dg id� d S r* � r; r( r rL r) r� r! rU rV r r rK rI r r2 r2 r3 �'test_empty_homogeneous_tuples_bindparam� s ���z<ExpandingBoundInTest.test_empty_homogeneous_tuples_bindparamc sD � j j�� �fdd�}|g g � |dddgdddg� |g g � d S ) Nc s@ t �jj��t�jj�jj��| ����jj�}� � ||� d S r@ ) r rL r) r� r! rU rV r rK rI r- r/ r2 r3 r0 � s ���zEExpandingBoundInTest.test_empty_homogeneous_tuples_direct.<locals>.go)r5 r �r r �r r\ r_ r` ra r4 r5 r2 r/ r3 �$test_empty_homogeneous_tuples_direct� s z9ExpandingBoundInTest.test_empty_homogeneous_tuples_directc C sV | j j}t|jj��|jj�td���� |jj�}| j |dddgddddgid� d S ) NrW r` ra r� r r r\ r� � r; r( r rL r) r� rU r r rK rI r r2 r2 r3 � test_bound_in_scalar_bindparam� s ���z3ExpandingBoundInTest.test_bound_in_scalar_bindparamc C sJ | j j}t|jj��|jj�dddg���|jj�}| � |dddg� d S )Nr r r\ r` ra r� � r; r( r rL r) r� rU r rK rI r r2 r2 r3 �test_bound_in_scalar_direct� s ���z0ExpandingBoundInTest.test_bound_in_scalar_directc C sV | j j}t|jj��|jj�ddg���|jj�g ��� |jj�}| � |ddg� d S )Nr r r` ra � r; r( r rL r) r� rU r �not_inrK rI r r2 r2 r3 �!test_nonempty_in_plus_empty_notin� s � ���z6ExpandingBoundInTest.test_nonempty_in_plus_empty_notinc C sR | j j}t|jj��|jj�g ���|jj�ddg��� |jj�}| � |g � d S )Nr r r@ r r2 r2 r3 �!test_empty_in_plus_notempty_notin� s ����z6ExpandingBoundInTest.test_empty_in_plus_notempty_notinc C s< t d��tdtdd��}| j|dddgddd d gid � d S ) z�test related to #7292. as a type is given to the bound param, there is no ambiguity to the type of element. �3select id FROM some_table WHERE z IN :q ORDER BY idrW T��type_r r` ra r� r r r r� N)r � bindparamsr r rI �rF rr r2 r2 r3 �test_typed_str_in� s � � �z&ExpandingBoundInTest.test_typed_str_inc C s: t d��tddd��}| j|dddgddd d gid � d S ) z�test related to #7292. for untyped expression, we look at the types of elements. Test for Sequence to detect tuple in. but not strings or bytes! as always.... rD rW T�r r` ra r� r r r r� N�r rG r rI rH r2 r2 r3 �test_untyped_str_in� s � � �z(ExpandingBoundInTest.test_untyped_str_inc C s` | j j}t|jj��t|jj|jj�� t d���� |jj�}| j |dddgddddgid� d S ) NrW r` ra r� r9 r: �r\ rf r� r7 r r2 r2 r3 �!test_bound_in_two_tuple_bindparam� s ��� �z6ExpandingBoundInTest.test_bound_in_two_tuple_bindparamc C sT | j j}t|jj��t|jj|jj�� dddg��� |jj�}| � |dddg� d S )Nr9 r: rM r` ra r� ) r; r( r rL r) r� r! rU rV r rK rI r r2 r2 r3 � test_bound_in_two_tuple_direct s ���z3ExpandingBoundInTest.test_bound_in_two_tuple_directc C s` | j j}t|jj��t|jj|jj�� t d���� |jj�}| j |dddgddddgid� d S ) NrW r` ra r� r1 r2 r3 r� r+ r r2 r2 r3 �/test_bound_in_heterogeneous_two_tuple_bindparam s ��� �zDExpandingBoundInTest.test_bound_in_heterogeneous_two_tuple_bindparamc C sT | j j}t|jj��t|jj|jj�� dddg��� |jj�}| � |dddg� d S )Nr1 r2 r3 r` ra r� ) r; r( r rL r) r� r! rU r r rK rI r r2 r2 r3 �,test_bound_in_heterogeneous_two_tuple_direct! s ���� �zAExpandingBoundInTest.test_bound_in_heterogeneous_two_tuple_directc C s: t d��tddd��}| j|dddgddd d gid � d S ) N�8select id FROM some_table WHERE (x, z) IN :q ORDER BY idrW TrJ r` ra r� r1 r2 r3 r� rK rH r2 r2 r3 �4test_bound_in_heterogeneous_two_tuple_text_bindparam2 s � � �zIExpandingBoundInTest.test_bound_in_heterogeneous_two_tuple_text_bindparamc C sj G dd� dt j�}td��tdtt� t� �dd��}| j|ddd gd|d d �|d d �|dd�gid� d S )Nc @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) zhExpandingBoundInTest.test_bound_in_heterogeneous_two_tuple_typed_bindparam_non_tuple.<locals>.LikeATuplec W s || _ d S r@ ��_data�rF r, r2 r2 r3 �__init__B s zqExpandingBoundInTest.test_bound_in_heterogeneous_two_tuple_typed_bindparam_non_tuple.<locals>.LikeATuple.__init__c S s t | j�S r@ ��iterrU �rF r2 r2 r3 �__iter__E s zqExpandingBoundInTest.test_bound_in_heterogeneous_two_tuple_typed_bindparam_non_tuple.<locals>.LikeATuple.__iter__c S s | j | S r@ rT �rF �idxr2 r2 r3 � __getitem__H s ztExpandingBoundInTest.test_bound_in_heterogeneous_two_tuple_typed_bindparam_non_tuple.<locals>.LikeATuple.__getitem__c S s t | j�S r@ �r rU rZ r2 r2 r3 �__len__K s zpExpandingBoundInTest.test_bound_in_heterogeneous_two_tuple_typed_bindparam_non_tuple.<locals>.LikeATuple.__len__N�rO rP rQ rW r[ r^ r` r2 r2 r2 r3 � LikeATupleA s rb rR rW TrE r` ra r� r r r r r\ r r� ) �collections_abc�Sequencer rG r r" r r rI �rF rb rr r2 r2 r3 �?test_bound_in_heterogeneous_two_tuple_typed_bindparam_non_tuple? s( � �����zTExpandingBoundInTest.test_bound_in_heterogeneous_two_tuple_typed_bindparam_non_tuplec C s^ G dd� dt j�}td��tddd��}| j|ddd gd|d d �|d d �|dd�gid� d S )Nc @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) zgExpandingBoundInTest.test_bound_in_heterogeneous_two_tuple_text_bindparam_non_tuple.<locals>.LikeATuplec W s || _ d S r@ rT rV r2 r2 r3 rW g s zpExpandingBoundInTest.test_bound_in_heterogeneous_two_tuple_text_bindparam_non_tuple.<locals>.LikeATuple.__init__c S s t | j�S r@ rX rZ r2 r2 r3 r[ j s zpExpandingBoundInTest.test_bound_in_heterogeneous_two_tuple_text_bindparam_non_tuple.<locals>.LikeATuple.__iter__c S s | j | S r@ rT r\ r2 r2 r3 r^ m s zsExpandingBoundInTest.test_bound_in_heterogeneous_two_tuple_text_bindparam_non_tuple.<locals>.LikeATuple.__getitem__c S s t | j�S r@ r_ rZ r2 r2 r3 r` p s zoExpandingBoundInTest.test_bound_in_heterogeneous_two_tuple_text_bindparam_non_tuple.<locals>.LikeATuple.__len__Nra r2 r2 r2 r3 rb f s rb rR rW TrJ r` ra r� r r r r r\ r r� )rc rd r rG r rI re r2 r2 r3 �>test_bound_in_heterogeneous_two_tuple_text_bindparam_non_tuplea s � ����zSExpandingBoundInTest.test_bound_in_heterogeneous_two_tuple_text_bindparam_non_tuplec C sJ | j j}t|jj��|jj�td���� |jj�}| j |g dg id� d S r* r< r r2 r2 r3 �(test_empty_set_against_integer_bindparam� s ���z=ExpandingBoundInTest.test_empty_set_against_integer_bindparamc C s> | j j}t|jj��|jj�g ���|jj�}| � |g � d S r@ r> r r2 r2 r3 �%test_empty_set_against_integer_direct� s &z:ExpandingBoundInTest.test_empty_set_against_integer_directc C sR | j j}t|jj��|jj�td���� |jj�}| j |ddddgdg id� d S �NrW r_ r` ra r� r� ) r; r( r rL r) r� rU rA r rK rI r r2 r2 r3 �1test_empty_set_against_integer_negation_bindparam� s ���zFExpandingBoundInTest.test_empty_set_against_integer_negation_bindparamc C sF | j j}t|jj��|jj�g ���|jj�}| � |ddddg� d S �Nr_ r` ra r� ) r; r( r rL r) r� rU rA rK rI r r2 r2 r3 �.test_empty_set_against_integer_negation_direct� s $�zCExpandingBoundInTest.test_empty_set_against_integer_negation_directc C sJ | j j}t|jj��|jj�td���� |jj�}| j |g dg id� d S r* ) r; r( r rL r) r� r r r rK rI r r2 r2 r3 �'test_empty_set_against_string_bindparam� s ���z<ExpandingBoundInTest.test_empty_set_against_string_bindparamc C s> | j j}t|jj��|jj�g ���|jj�}| � |g � d S r@ ) r; r( r rL r) r� r r rK rI r r2 r2 r3 �$test_empty_set_against_string_direct� s &z9ExpandingBoundInTest.test_empty_set_against_string_directc C sR | j j}t|jj��|jj�td���� |jj�}| j |ddddgdg id� d S rj ) r; r( r rL r) r� r rA r rK rI r r2 r2 r3 �0test_empty_set_against_string_negation_bindparam� s ���zEExpandingBoundInTest.test_empty_set_against_string_negation_bindparamc C sF | j j}t|jj��|jj�g ���|jj�}| � |ddddg� d S rl ) r; r( r rL r) r� r rA rK rI r r2 r2 r3 �-test_empty_set_against_string_negation_direct� s $�zBExpandingBoundInTest.test_empty_set_against_string_negation_directc C sD t tt� �tddd��t� ft� d��}t|�|��� d d� d S )N�foor2 )�value�Zelse_r �Fr ) r r r r r r r r: �fetchone�rF r> rr r2 r2 r3 �)test_null_in_empty_set_is_false_bindparam� s ��� z>ExpandingBoundInTest.test_null_in_empty_set_is_false_bindparamc C s< t tt� �g �t� ft� d��}t|�|��� d d� d S )Nrt r ru )r r r r r r r: rv rw r2 r2 r3 �&test_null_in_empty_set_is_false_direct� s ��� z;ExpandingBoundInTest.test_null_in_empty_set_is_false_directN)r2 ))rO rP rQ rR rS r4 r? rI r( r) r rJ Ztuple_in_w_emptyr, r6 r8 r; r= r? rB rC rI rL r& rN rO rP rQ rS rf rg rh ri rk rm rn ro rp rq rx ry r2 r2 r2 r3 r' J s\ ! r' c @ s> e Zd ZdZdZdZedd� �Zedd� �Zdd � Z d d � Z d d � Z dd� Z dd� Z dd� Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zejjd&d'� �Zejj d(d)� �Z ejje� d*d+d,d-d.hfd/d+d,d-d.d0hfd1d+d,d-d.d0hfd2d3hfd4e e d+d5��fd6e e d+d5��fd7e � f�d8d9� ��Z!dS ):�LikeFunctionsTestT�onceNc C s& t d|tdtdd�tdtd��� d S )Nr( r) Tr* r, rX r. r/ r2 r2 r3 r4 � s �z LikeFunctionsTest.define_tablesc C sp |� | jj�� ddd�ddd�ddd�dd d�d d d�d d d�ddd�ddd�ddd�ddd�dd d�g � d S )Nr5 Zabcdefgr7 r zab/cdefgr zab%cdefgr\ Zab_cdefgrf zabcde/fgr~ zabcde%fgrh zab#cdefg� Zab9cdefg� zabcde#fgr� Zabcd9fg� r9 r= r2 r2 r3 r? � s ��z LikeFunctionsTest.insert_datac C sN | j j}tj�� �(}dd� |�t|jj�� |��D �}W 5 Q R X t ||� d S )Nc S s h | ] \}|�qS r2 r2 )r� rs r2 r2 r3 � <setcomp>� s �z*LikeFunctionsTest._test.<locals>.<setcomp>) r; r( r rB rC r: r rL r) r� r )rF rq r� r( rH �rowsr2 r2 r3 �_test� s �zLikeFunctionsTest._testc C s6 | j jjj}| �|�d�dddddddd d d h � d S ) N�ab%cr5 r r r\ rf r~ rh r| r} r� �r; r( rL r, r� � startswith�rF �colr2 r2 r3 �test_startswith_unescaped s z+LikeFunctionsTest.test_startswith_unescapedc C s( | j jjj}| �|jddd�dh� d S )Nr� T�� autoescaper r� r� r2 r2 r3 �test_startswith_autoescape s z,LikeFunctionsTest.test_startswith_autoescapec C s: | j jjj}| �|�td��dddddddd d d h � d S ) Nz'ab%c'r5 r r r\ rf r~ rh r| r} r� )r; r( rL r, r� r� r r� r2 r2 r3 �test_startswith_sqlexpr s �z)LikeFunctionsTest.test_startswith_sqlexprc C s( | j jjj}| �|jddd�dh� d S )Nzab##c�#��escaperh r� r� r2 r2 r3 �test_startswith_escape s z(LikeFunctionsTest.test_startswith_escapec C sD | j jjj}| �|jdddd�dh� | �|jdddd�dh� d S )Nr� Tr� �r� r� r zab#crh r� r� r2 r2 r3 �!test_startswith_autoescape_escape s z3LikeFunctionsTest.test_startswith_autoescape_escapec C s4 | j jjj}| �|�d�dddddddd d h � d S ) N�e%fgr5 r r r\ rf r~ rh r| r} �r; r( rL r, r� �endswithr� r2 r2 r3 �test_endswith_unescaped s z)LikeFunctionsTest.test_endswith_unescapedc C s8 | j jjj}| �|�td��dddddddd d h � d S ) Nz'e%fg'r5 r r r\ rf r~ rh r| r} )r; r( rL r, r� r� r r� r2 r2 r3 �test_endswith_sqlexpr s �z'LikeFunctionsTest.test_endswith_sqlexprc C s( | j jjj}| �|jddd�dh� d S )Nr� Tr� r~ r� r� r2 r2 r3 �test_endswith_autoescape& s z*LikeFunctionsTest.test_endswith_autoescapec C s( | j jjj}| �|jddd�dh� d S )Nze##fgr� r� r} r� r� r2 r2 r3 �test_endswith_escape* s z&LikeFunctionsTest.test_endswith_escapec C sD | j jjj}| �|jdddd�dh� | �|jdddd�dh� d S )Nr� Tr� r� r~ ze#fgr} r� r� r2 r2 r3 � test_endswith_autoescape_escape. s z1LikeFunctionsTest.test_endswith_autoescape_escapec C s4 | j jjj}| �|�d�dddddddd d h � d S ) N�b%cder5 r r r\ rf r~ rh r| r} �r; r( rL r, r� �containsr� r2 r2 r3 �test_contains_unescaped3 s z)LikeFunctionsTest.test_contains_unescapedc C s( | j jjj}| �|jddd�dh� d S )Nr� Tr� r r� r� r2 r2 r3 �test_contains_autoescape7 s z*LikeFunctionsTest.test_contains_autoescapec C s( | j jjj}| �|jddd�dh� d S )Nzb##cder� r� rh r� r� r2 r2 r3 �test_contains_escape; s z&LikeFunctionsTest.test_contains_escapec C sD | j jjj}| �|jdddd�dh� | �|jdddd�dh� d S )Nzb%cdTr� r� r zb#cdrh r� r� r2 r2 r3 � test_contains_autoescape_escape? s z1LikeFunctionsTest.test_contains_autoescape_escapec C s0 | j jjj}| �|�d� ddddddh� d S )N�a.cder r r\ rh r| r� �r; r( rL r, r� � regexp_matchr� r2 r2 r3 �test_not_regexp_matchD s z'LikeFunctionsTest.test_not_regexp_matchc C s2 | j jjj}| �|�dd��d�ddddh� d S )Nr� ZFOOr5 rf r~ r} )r; r( rL r, r� �regexp_replacer� r� r2 r2 r3 �test_regexp_replaceI s �z%LikeFunctionsTest.test_regexp_replacer� r5 rf r~ r} �abcr� z^abcZ9cder| z^ar~ z(b|c)z^(b|c)c C s" | j jjj}| �|�|�|� d S r@ r� )rF r r� r� r2 r2 r3 �test_regexp_matchP s z#LikeFunctionsTest.test_regexp_match)"rO rP rQ rR � run_inserts� run_deletesrS r4 r? r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r rJ r� r� r� r� r� r� �ranger� r2 r2 r2 r3 rz � sJ � rz c @ s< e Zd ZdZdZedd� �Zedd� �Zdd� Zd d � Z d S ) �ComputedColumnTestT)Zcomputed_columnsc C s> t d|tdtdd�tdt�tdttd��tdttd ��� d S ) N�squarer) Tr* �side�areaz side * side� perimeterz4 * side)r r r r r/ r2 r2 r3 r4 c s �z ComputedColumnTest.define_tablesc C s( |� | jj�� ddd�ddd�g� d S )Nr5 r� )r) r� �* )r: r; r� r< r= r2 r2 r3 r? n s �z ComputedColumnTest.insert_datac C sV t j�� �B}|�ttd���| jj�� | jjj j ��� � }t |ddg� W 5 Q R X d S )N�*)r5 r� r- r� )r� r� �� � )r rB rC r: r r r� r; r� rK rL r) rD r �rF rH �resr2 r2 r3 �test_select_allu s � ��z"ComputedColumnTest.test_select_allc C sd t j�� �P}|�t| jjjj| jjjj �� | jj�� | jjjj ��� � }t|ddg� W 5 Q R X d S )N)r- r� )r� r� )r rB rC r: r r; r� rL r� r� r� rK r) rD r r� r2 r2 r3 �test_select_columns~ s �� ��z&ComputedColumnTest.test_select_columnsN) rO rP rQ rR ry rS r4 r? r� r� r2 r2 r2 r3 r� _ s r� c @ sT e Zd ZdZdZdZdZedd� �Zedd� �Z dd � Z d d � Z e j jd d � �ZdS )�IdentityColumnTestT)Zidentity_columnsr{ c C sd t d|tdttddddd�dd�tdtd��� t d |tdttd d d d d �dd�tdtd��� d S )N�tbl_ar) Tr� )�always�startZ nominvalueZ nomaxvaluer* rl r- �tbl_b���r i���)� incrementr� ZminvalueZmaxvalue�r r r r r r/ r2 r2 r3 r4 � s6 �� � � �z IdentityColumnTest.define_tablesc C s` |� | jj�� ddiddig� |� | jj�� ddiddig� |� | jj�� ddd�g� d S )Nrl r� r� r� rL �r) rl )r: r; r� r< r� r= r2 r2 r3 r? � s � � �z IdentityColumnTest.insert_datac C s~ |� ttd���| jj��| jjjj��� � }t |ddg� |� ttd���| jj ��| jj jj��� � }t |dddg� d S )Nr� )r� r� )�+ r� )r� r� )r r� )r� rL ) r: r r r� r; r� rK rL r) rD r r� �rF r> r� r2 r2 r3 r� � s � �� � ��z"IdentityColumnTest.test_select_allc C s: |� t| jjjj��| jjjj���� }t|ddg� d S )N)r� )r� ) r: r r; r� rL r) rK rD r r� r2 r2 r3 r� � s �z&IdentityColumnTest.test_select_columnsc s � �fdd�}t ttf|� d S )Nc s � � �jj�� ddd�g� d S )N�� r� r� )r: r; r� r< r2 �r> rF r2 r3 �fn� s �z7IdentityColumnTest.test_insert_always_error.<locals>.fn)r r% r& )rF r> r� r2 r� r3 �test_insert_always_error� s z+IdentityColumnTest.test_insert_always_errorN)rO rP rQ rR ry r� r� rS r4 r? r� r� r rJ Zidentity_columns_standardr� r2 r2 r2 r3 r� � s r� c @ s( e Zd ZdZdZedd� �Zdd� ZdS )�IdentityAutoincrementTestT)Z autoincrement_without_sequencec C s, t d|tdtt� ddd�tdtd��� d S )N�tblr) T)r+ Z autoincrementrl r- r� r/ r2 r2 r3 r4 � s � �z'IdentityAutoincrementTest.define_tablesc C s< |� | jj�� ddi�}|� | jj�� ��� }t|d� d S )Nrl �row)r5 r� )r: r; r� r< r r� r r� r2 r2 r3 � test_autoincrement_with_identity� s z:IdentityAutoincrementTest.test_autoincrement_with_identityN)rO rP rQ rR ry rS r4 r� r2 r2 r2 r3 r� � s r� c @ s8 e Zd ZdZedd� �Zedd� �Zdd� Zdd � Zd S ) � ExistsTestTc C s& t d|tdtdd�tdtd��� d S )N�stuffr) Tr* r, rX r. r/ r2 r2 r3 r4 � s �zExistsTest.define_tablesc C s8 |� | jj�� ddd�ddd�ddd�ddd�g� d S )Nr5 � some datar7 r r r\ zsome other data)r: r; r� r< r= r2 r2 r3 r? � s ��zExistsTest.insert_datac C s@ | j j}t|�ttd���t� �|jj dk���� � dg� d S )Nr5 r� r_ � r; r� r r: r r r� r rL r, rD �rF r> r� r2 r2 r3 �test_select_exists s ���z ExistsTest.test_select_existsc C s> | j j}t|�ttd���t� �|jj dk���� � g � d S )Nr5 zno datar� r� r2 r2 r3 �test_select_exists_false s ���z#ExistsTest.test_select_exists_falseN) rO rP rQ rR rS r4 r? r� r� r2 r2 r2 r3 r� � s r� c @ s& e Zd ZdZe�ejj�dd� �ZdS )�DistinctOnTestTc C sB t d��td���td��}t�d�� | �|d� W 5 Q R X d S )Nr� rW rr z:DISTINCT ON is currently supported only by the PostgreSQL zSELECT DISTINCT * FROM foo)r r r r� r r Zexpect_deprecatedr )rF Zstmr2 r2 r3 �test_distinct_on% s �z DistinctOnTest.test_distinct_onN) rO rP rQ rR r Zfails_ifrJ Zsupports_distinct_onr� r2 r2 r2 r3 r� " s r� c @ s@ e Zd ZdZdZedd� �Zejddddd d d d �d d� �Z dS )�IsOrIsNotDistinctFromTestT)Zsupports_is_distinct_fromc C s2 t d|tdtdd�tdtdd�tdtdd�� d S )N�is_distinct_testr) Tr* �col_ar� �col_br| r/ r2 r2 r3 r4 2 s �z'IsOrIsNotDistinctFromTest.define_tables)Zboth_int_differentr r5 r5 )Z both_int_samer5 r5 r )Zone_null_firstNr5 r5 )Zone_null_secondr Nr5 )Z both_nullNNr Ziaaaz3col_a_value, col_b_value, expected_row_count_for_is)Zid_r� c C s� | j j}|�|�� d||d�g� |�|�� �|jj�|jj ���� � }t t |�|� |dkrbdnd}|�|�� �|jj� |jj ���� � }t t |�|� d S )Nr5 )r) r� r� r )r; r� r: r< r r� rL r� Zis_distinct_fromr� rD r r Zis_not_distinct_from)rF Z col_a_valueZ col_b_valueZexpected_row_count_for_isr> r� rG Z expected_row_count_for_is_notr2 r2 r3 � test_is_or_is_not_distinct_from< s* ������z9IsOrIsNotDistinctFromTest.test_is_or_is_not_distinct_fromN) rO rP rQ rR ry rS r4 r r� r� r2 r2 r2 r3 r� . s � r� c @ s< e Zd ZdZdZedd� �Zedd� �Zdd� Zd d � Z d S ) �WindowFunctionTest)Zwindow_functionsTc C s* t d|tdtdd�tdt�tdt�� d S )Nr( r) Tr* �col1�col2r| r/ r2 r2 r3 r4 h s �z WindowFunctionTest.define_tablesc C s( |� | jj�� dd� tdd�D �� d S )Nc S s g | ]}|||d d��qS )rf )r) r� r� r2 �r� �ir2 r2 r3 r� v s z2WindowFunctionTest.insert_data.<locals>.<listcomp>r5 rX )r: r; r( r< r� r= r2 r2 r3 r? r s �z WindowFunctionTest.insert_datac C s` | j j}|�tt�|jj�j|jj � � gd��� |jj dk ��� � }t |dd� td�D �� d S )N)rK r� c S s g | ]}d �qS ))�_ r2 r� r2 r2 r3 r� � s z2WindowFunctionTest.test_window.<locals>.<listcomp>� )r; r( r: r r �maxrL r� �overr� rl r� rw r r� �rF r> r( r� r2 r2 r3 � test_windowy s �� ��z WindowFunctionTest.test_windowc C sT | j j}|�tt�|jj�j|jj gdd���� � }t |dd� t ddd�D �� d S )N)r� r )rK r� c S s g | ] }|f�qS r2 r2 r� r2 r2 r3 r� � s z?WindowFunctionTest.test_window_rows_between.<locals>.<listcomp>rf � ) r; r( r: r r r� rL r� r� r� rw r r� r� r2 r2 r3 �test_window_rows_between� s ��� z+WindowFunctionTest.test_window_rows_betweenN) rO rP rQ ry rR rS r4 r? r� r� r2 r2 r2 r3 r� c s r� c @ s� e Zd ZdZd ZZdd� ed�D �Zedd� �Z edd � �Z e j d d � d d� ed�D �e j jfd d � eed��e j jfdd � eedd��e j jfdd � dge j jfdd � eedd��e j jfdd � eedd��e j jfdd�dd� �ZdS )� BitwiseTestTr{ c C s g | ]}||d d��qS )r5 )r� r� r2 r� r2 r2 r3 r� � s zBitwiseTest.<listcomp>r� c C s t d|tdt�tdt�� d S )N�bitwiser� r� r| r/ r2 r2 r3 r4 � s zBitwiseTest.define_tablesc C s |� | jj�� | j� d S r@ )r: r; r� r< � inserted_datar= r2 r2 r3 r? � s zBitwiseTest.insert_datac C s | � d�S )Nrf )Z bitwise_xor�r� r2 r2 r3 �<lambda>� � zBitwiseTest.<lambda>c C s g | ]}|d kr|�qS re r2 r� r2 r2 r3 r� � s c C s | � d�S �Nr5 )Z bitwise_orr� r2 r2 r3 r� � r� c C s | � d�S )Nr\ )Z bitwise_andr� r2 r2 r3 r� � r� r\ r| c C s | d � � S �Nr )Z bitwise_notr� r2 r2 r3 r� � r� r c C s | � d�S r� )Zbitwise_lshiftr� r2 r2 r3 r� � r� r5 c C s | � d�S r� )Zbitwise_rshiftr� r2 r2 r3 r� � r� zcase, expectedr� c sd � j j}|jj}tj||d�}t|��|dk��|�}|� |�� � � � }t |� fdd�|D �� d S )Nr� r c s g | ]}� j | �qS r2 )r� r� rZ r2 r3 r� � s z,BitwiseTest.test_bitwise.<locals>.<listcomp>) r; r� rL r� r Zresolve_lambdar r� rK r: Zmappingsrw r ) rF r r� r> r� r� �oprr r� r2 rZ r3 � test_bitwise� s "zBitwiseTest.test_bitwiseN)rO rP rQ rR r� r� r� r� rS r4 r? r r� rJ Zsupports_bitwise_xor�listZsupports_bitwise_orZsupports_bitwise_andZsupports_bitwise_notZsupports_bitwise_shiftr� r2 r2 r2 r3 r� � sF � � �� � ��!r� )@�collections.abcr� rc r� r" r r r r Z assertionsr r r Z assertsqlr r� r r r r r r r r r r r r r r r r r r r r r r! r" r# r$ �excr% r& Z TablesTestr' rT ZTestBaseru rz r� r� r� r r' rz r� r� r� r� r� r� r� r� r2 r2 r2 r3 �<module> s� &M Jcdw � +P/ 53