U
|
¸ý°dà; ã @ sþ d dl mZ ddlmZ ddlmZ ddlmZ ddlmZ ddl m
|
Z
|
dd l m Z d
|
d lm Z d
|
d lm Z d
|
d lmZ d
|
dlmZ d
|
dlmZ d
|
dlmZ d
|
dlmZ d
|
dlmZ d
|
dlmZ G dd dejZG dd dejZG dd dejZdZdS ) é )ÚDecimalé )Útestingé )Úfixtures)Úeq_)Ú requirements)ÚColumn)ÚTableé )ÚDouble)ÚFloat)ÚIdentity)ÚInteger)Úliteral)Úliteral_column)ÚNumeric)Úselect)ÚStringc @ sJ e Zd ZdZdZdZedd Zdd Zdd Z d
|
d Z
|
e j d d Z dS )Ú LastrowidTestÚeachT)Zimplements_get_lastrowidÚautoincrement_insertc C sT t d|tdtdddtdtddd t d |tdtddd
|
tdtddd d S ) NÚ
|
autoinc_pkÚidT©Ú primary_keyZtest_needs_autoincrementÚdataé2 F)Zimplicit_returningÚ manual_pk©r Z autoincrement©r
|
r r r ©ÚclsÚmetadata© r$ ú[d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\sqlalchemy/testing/suite/test_insert.pyÚ define_tables s& ÿ ù
|
ûzLastrowidTest.define_tablesc C s( | | ¡ ¡ ¡ }t||jjdf d S ©Nú some data©Úexecuter Úfirstr ÚdialectZdefault_sequence_base©ÚselfÚtableÚconnÚrowr$ r$ r% Ú_assert_round_trip1 s þþz LastrowidTest._assert_round_tripc C s. | | jj ¡ tdd¡ | | jj|¡ d S ©Nr( ©r ©r* Útablesr ÚinsertÚdictr2 ©r. Ú
|
connectionr$ r$ r% Ú test_autoincrement_on_insert; s
|
|
ÿz*LastrowidTest.test_autoincrement_on_insertc C sB | | jj ¡ tdd¡}| t| jjjj¡}t |j
|
|f d S r3 © r* r6 r r7 r8 Úscalarr Úcr r Úinserted_primary_key©r. r: ÚrÚpkr$ r$ r% Útest_last_inserted_idB s
|
ÿz#LastrowidTest.test_last_inserted_idc C sD | | jj ¡ tdd¡}|j}| t| jjjj ¡}t
|
|| d S r3 ) r* r6 r r7 r8 Ú lastrowidr= r r> r r )r. r: rA rD rB r$ r$ r% Ú test_native_lastrowid_autoincJ s
|
ÿz+LastrowidTest.test_native_lastrowid_autoincN)Ú__name__Ú
|
__module__Ú __qualname__Ú run_deletesÚ __backend__Ú __requires__Ú classmethodr& r2 r; rC r Zdbapi_lastrowidrE r$ r$ r$ r% r s
|
|
r c @ s e Zd ZdZdZedd Zejdd Z ej
|
dd Z ej d d
|
Z ejd d Zejd d Zejdd Zejdd Zejdd ZdS )ÚInsertBehaviorTestr Tc C s t d|tdtdddtdtd t d|tdtddd tdtd t d
|
|tdtdddtdtdtd td d tdttdtdtd d d S )Nr r Tr r r r Fr Úincludes_defaultsÚxé )ÚdefaultÚyÚ2)Útype_r )r
|
r r r r r r! r$ r$ r% r& X sD ÿ ú ü ÿ ýøz InsertBehaviorTest.define_tablesc C sR | | jj ¡ tdd¡}|js$t|jr.t|js8t|j rN|
|
¡ d ksNtd S r3 ) r* r6 r r7 r8 Ú _soft_closedÚAssertionErrorÚclosedÚ is_insertÚ returns_rowsÚfetchone©r. r: rA r$ r$ r% Útest_autoclose_on_insertw s
|
ÿ
|
|
|
z+InsertBehaviorTest.test_autoclose_on_insertc C sh | | jj ¡ ¡ tdd¡}|js(t|jr2t|j s<t|j
|
sFtt | ¡ d t | ¡ dg d S )Nr( r4 r )r* r6 r r7 Zreturn_defaultsr8 rU rV rW rX rY r rZ Úkeysr[ r$ r$ r% Ú+test_autoclose_on_insert_implicit_returning s û
|
|
|
|
z>InsertBehaviorTest.test_autoclose_on_insert_implicit_returningc C s` | | jj ¡ ¡}|js t|jr&t| | jj ¡ | jjj j
|
d k¡¡}t t | ¡ d d S )Nr ©r* r6 r r7 rU rV rW r Úwherer> r r ÚlenÚallr[ r$ r$ r% Útest_empty_insert£ s
|
|
ÿÿz$InsertBehaviorTest.test_empty_insertc C sh | | jj ¡ i i i g¡}|js$t|jr.t| | jj ¡ | jjj j
|
d k¡¡}t t | ¡ d d S )Nr r_ r[ r$ r$ r% Útest_empty_insert_multiple° s
|
|
ÿÿz-InsertBehaviorTest.test_empty_insert_multiplec C s® | j j}| j j}| | ¡ tdddtdddtdddg¡ | | ¡ dt|jj
|
|jj ddg¡¡¡¡}t |j d | t|jj |jj ¡¡}t | ¡ d
|
d g d S ) Nr Údata1©r r r Údata2r Údata3r4 ©N©rg ©rh )r6 r r r* r7 r8 Ú from_selectr r> r r` Úin_r r? Úorder_byÚfetchall©r. r: Z src_tableZ
|
dest_tableÚresultr$ r$ r% Ú test_insert_from_select_autoinc¾ s,
|
|
|
ýþ ÿþÿ ÿz2InsertBehaviorTest.test_insert_from_select_autoincc C s| | j j}| j j}| | ¡ dt|jj |jj
|
ddg¡¡¡¡}t |j d | t|jj |jj¡¡}t | ¡ g d S )Nr4 rg rh ri )r6 r r r* r7 rl r r> r r` rm r r? rn ro rp r$ r$ r% Ú'test_insert_from_select_autoinc_no_rowsÛ s ÿþÿ ÿz:InsertBehaviorTest.test_insert_from_select_autoinc_no_rowsc C sª | j j}| | ¡ tdddtdddtdddg¡ | | ¡ ¡ dt|jj d |jj
|
|jj
|
ddg¡¡¡¡ t | t|jj
|
|jj
|
¡¡ ¡ d d
|
d
|
d d g d S ) Nr re rf r rg r rh rP )re rj rk )r6 r r* r7 r8 Úinlinerl r r> r r r` rm r rn ro ©r. r: r/ r$ r$ r% Útest_insert_from_selectð s,
|
|
|
ýþ ÿüÿ ÿ üz*InsertBehaviorTest.test_insert_from_selectc C s¬ | j j}| | ¡ tdddtdddtdddg¡ | | ¡ ¡ dt|jj d |jj
|
|jj
|
ddg¡¡¡¡ t | t| |jj
|
|jj ¡¡ ¡ d d
|
d d d g d S )Nr re rf r rg r rh rP )r re rP é )r rg rP rw )é rg rP rw )r rh rP rw )é rh rP rw )r6 rN r* r7 r8 rt rl r r> r r r` rm r rn ro ru r$ r$ r% Ú%test_insert_from_select_with_defaults s6
|
|
|
ýþ ÿüÿ ÿûüz8InsertBehaviorTest.test_insert_from_select_with_defaultsN)rF rG rH rI rJ rL r& r r r\ Úinsert_returningr^ Z empty_insertsrc Zempty_inserts_executemanyrd Zinsert_from_selectrr rs rv rz r$ r$ r$ r% rM T s(
|
|
|
|
|
|
|
|
rM c @ sú e Zd ZdZdZdZdd Zedd Ze j
|
dd Z d
|
d Z d d Z dd Ze jdd Zeje ddfedddejjfe ddfedddejjfeddddddejjfeddddeddfdd e d ddg¡e d ddg¡d d Zd!S )"Ú ReturningTestr )r{ r Tc C s( | | ¡ ¡ ¡ }t||jjdf d S r' r) r- r$ r$ r% r2 8 s þþz ReturningTest._assert_round_tripc C s( t d|tdtdddtdtd d S )Nr r Tr r r r r! r$ r$ r% r& B s ÿ úzReturningTest.define_tablesc C sT | j j}| | ¡ |jj¡tdd¡}| ¡ d }| t
|
|jj¡}t || d S ©Nr( r4 r © r6 r r* r7 Ú returningr> r r8 r+ r= r r ©r. r: r/ rA rB Z
|
fetched_pkr$ r$ r% Ú%test_explicit_returning_pk_autocommitM s ÿ z3ReturningTest.test_explicit_returning_pk_autocommitc C sT | j j}| | ¡ |jj¡tdd¡}| ¡ d }| t
|
|jj¡}t || d S r} r~ r r$ r$ r% Ú(test_explicit_returning_pk_no_autocommitW s ÿ z6ReturningTest.test_explicit_returning_pk_no_autocommitc C s. | | jj ¡ tdd¡ | | jj|¡ d S r3 r5 r9 r$ r$ r% Ú/test_autoincrement_on_insert_implicit_returninga s
|
|
ÿz=ReturningTest.test_autoincrement_on_insert_implicit_returningc C sB | | jj ¡ tdd¡}| t| jjjj¡}t |j
|
|f d S r3 r< r@ r$ r$ r% Ú(test_last_inserted_id_implicit_returningh s
|
ÿz6ReturningTest.test_last_inserted_id_implicit_returningc C sp | | jj ¡ | jjjj¡ddiddiddiddiddig¡}| ¡ }| t| jjjj¡}t || ¡ d S )Nr Úd1Zd2Zd3Zd4Zd5)
|
r* r6 r r7 r r> r rb r r )r. r: rA ZrallZpksr$ r$ r% Ú test_insertmanyvalues_returningp s
|
ÿûü z-ReturningTest.test_insertmanyvalues_returninggxXM|Z!@é5 g_ÎQ!@Fry é é )Z precisionZscaleZ asdecimalz 8.5514716ztype_,value,do_rounding)ÚargnamesÚsort_by_parameter_orderÚ multiple_rowsc s t d|tdtt ddtd|}| |¡ | | ¡ j|jj |jj
|
t |d|rh fddt d D nd i¡} |rt d
|
d }
|
n
|
t d
|
d }
|
|rÞt d d | D fdd|
|
D t dd | t|jj
|
¡D t dh n:t t| fdd|
|
D t t| t|jj
|
¡ h dS )z¯test #9701.
|
|
this tests insertmanyvalues as well as decimal / floating point
|
RETURNING types
|
|
TODO: this might be better in suite/test_types?
|
|
Útr T)r Úvalue)r c s g | ] }d iqS ©r r$ )Ú.0Úir r$ r% Ú
|
<listcomp>Ä s z6ReturningTest.test_insert_w_floats.<locals>.<listcomp>é
|
r é r c S s h | ]\}}|t |d fqS ©rP ©Úround)r Úid_Úval_r$ r$ r% Ú <setcomp>Ø s z5ReturningTest.test_insert_w_floats.<locals>.<setcomp>c s h | ]}|t d fqS r r ©r r r r$ r% r Ù s c S s h | ]}t |d qS r r )r r r$ r$ r% r Ý s ÿrP c s h | ] }| fqS r$ r$ r r r$ r% r æ s N)r
|
r r r Úcreater* r7 r r> r r ÚboolÚranger Zscalarsr r Úset) r. r: r# r rT r Z do_roundingr r rq Zi_ranger$ r r% Útest_insert_w_floats sN 1ü
|
ýÿø
|
|
þþ
|
ûþþz"ReturningTest.test_insert_w_floatsN) rF rG rH Zrun_create_tablesrK rJ r2 rL r& r Zfetch_rows_post_commitr r r r Z insert_executemany_returningr r Ú combinationsr ÚrequiresZ-float_or_double_precision_behaves_genericallyr r Zliteral_float_coercionr Z variationr r$ r$ r$ r% r| 3 sP
|
|
|
|
|
|
|
ü
|
ü ü ýæ r| )r rM r| N)Údecimalr Ú r r Z
|
assertionsr Úconfigr Zschemar r
|
r r r r r r r r r Z
|
TablesTestr rM r| Ú__all__r$ r$ r$ r% Ú<module> s* > ` =
|
