U
|
ý°dÒ ã @ sÆ d Z ddlmZ ddlZddlmZ ddlZddlmZmZmZm Z ddl
|
m Z ddl m Z ddlZddlZddlZddlmZ ddlm mZ dd lmZmZ ddlZ dd
|
lm Z m Z m Z m Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z& ddl'm(Z) dd l*m+Z+m,Z, dd l-m.Z. dd l/m0Z0 ddl1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8 zddl9Z9dZ:W n e;k
|
r\ dZ:Y nX dddddddddddddZ<d d d d Z=d d!d"d#d$Z>d!d d%d&d'Z?d(d) Z@d d d*d+ZAd d,d-d.d/ZBd,d d0d1d2ZCd3d4d5d6ZDd d7d8d9ZEejFd:d; ZGejFd<d= ZHejFd>d? ZIejFd@dA ZJejFdBdC ZKejFdDdE ZLejFdFdG ZMejFdHdI ZNejFdJdK ZOejFdLdM ZPejFdNdO ZQejFdPdQ ZRejFdRdS ZSejFdTdU ZTejFdVdW ZUejFdXdY ZVejFdZd[ ZWdEdGgZXdIdKgZYdOdQdMgZZdUdWdSgZ[eXeY eZ Z\eXeY e[ Z]e\dYg Z^e]d[g Z_ej`jaej` bd\e^¡d]d^ Zcej`jaej` bd\e^¡d_d` Zdej`jaej` bd\e^¡dadb Zeej`jaej` bd\e^¡ej` bdcdddg¡dedf Zfej`jaej` bd\e^¡ej` bdgdhdig¡djdk Zgej`jaej` bd\e^¡dldm Zhej`jaej` bd\e_¡dndo Ziej`jaej` bd\e_¡dpdq Zjej`jaej` bd\e]¡drds Zkej`jaej` bd\e_¡dtdu Zlej`jaej` bd\e]¡dvdw Zmej`jaej` bd\e]¡dxdy Znej`jaej` bd\e\¡dzd{ Zoej`jaej` bd\eX¡d|d} Zpej`jaej` bd\eX¡d~d Zqej`jaej` bd\eY¡ej` bdddg¡dd Zrd
d Zsdd ZtG dd dZuG dd deuZvG dd deuZwG dd dZxG dd dexZyej`jze: ddG dd deweyZ{G dd deveyZ|ej`jze: ddG dd dewexZ}G dd de}Z~ej`jaG dd de}Zej`jaG dd d e}ZG d¡d¢ d¢evexZed£jed£jejd£je
d£jed¤jed¥jedd¦d§ ejd¨jed©ji Zdªd« Zd°d¬dZG d®d¯ d¯ZdS )±a· SQL io tests
|
|
The SQL tests are broken down in different classes:
|
|
- `PandasSQLTest`: base class with common methods for all test classes
|
- Tests for the public API (only tests with sqlite3)
|
- `_TestSQLApi` base class
|
- `TestSQLApi`: test the public API with sqlalchemy engine
|
- `TestSQLiteFallbackApi`: test the public API with a sqlite DBAPI
|
connection
|
- Tests for the different SQL flavors (flavor specific type conversions)
|
- Tests for the sqlalchemy mode: `_TestSQLAlchemy` is the base class with
|
common methods. The different tested flavors (sqlite3, MySQL,
|
PostgreSQL) derive from the base class
|
- Tests for the fallback mode (`TestSQLiteFallback`)
|
|
é )Ú annotationsN)Úclosing)ÚdateÚdatetimeÚtimeÚ timedelta)ÚStringIO©ÚPath)Úlib)Úis_datetime64_dtypeÚis_datetime64tz_dtype)
|
Ú DataFrameÚIndexÚ
|
MultiIndexÚSeriesÚ TimestampÚconcatÚ
|
date_rangeÚisnaÚ to_datetimeÚ to_timedelta)ÚArrowStringArrayÚ StringArray)ÚVersion)Úsql)ÚSQLAlchemyEngineÚ SQLDatabaseÚSQLiteDatabaseÚ
|
get_engineÚpandasSQL_builderÚread_sql_queryÚread_sql_tableTFz1SELECT * FROM iris WHERE Name=? AND SepalLength=?z7SELECT * FROM iris WHERE `Name`=%s AND `SepalLength`=%sz7SELECT * FROM iris WHERE "Name"=%s AND "SepalLength"=%s)ÚsqliteÚmysqlÚ
|
postgresqlz]
|
SELECT * FROM iris WHERE Name=:name AND SepalLength=:length
|
zw
|
SELECT * FROM iris WHERE
|
`Name`=%(name)s AND `SepalLength`=%(length)s
|
zw
|
SELECT * FROM iris WHERE
|
"Name"=%(name)s AND "SepalLength"=%(length)s
|
z&SELECT * FROM iris WHERE Name LIKE '%'z(SELECT * FROM iris WHERE `Name` LIKE '%'z(SELECT * FROM iris WHERE "Name" LIKE '%')Úread_parametersÚread_named_parametersÚ read_no_parameters_with_percentÚstr)Údialectc
|
C sp ddl m}m}m}m}m}m} | dkr,|n|}| }|d||d||d||d||d||d |d
|
} | S ) Nr )ÚREALÚColumnÚFloatÚMetaDataÚStringÚTabler% ÚirisÚ SepalLengthÚ
|
SepalWidthZ PetalLengthZ
|
PetalWidthÚNameéÈ )Ú
|
sqlalchemyr+ r, r- r. r/ r0 )
|
r* r+ r, r- r. r/ r0 ÚdtypeÚmetadatar1 © r9 úOd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/io/test_sql.pyÚiris_table_metadatam s ù r; zsqlite3.Connectionr
|
)ÚconnÚ iris_filec C sT | ¡ }d}| |¡ |jd d(}t |¡}t| d}| ||¡ W 5 Q R X d S )Nz³CREATE TABLE iris (
|
"SepalLength" REAL,
|
"SepalWidth" REAL,
|
"PetalLength" REAL,
|
"PetalWidth" REAL,
|
"Name" TEXT
|
)©Únewlinez&INSERT INTO iris VALUES(?, ?, ?, ?, ?))ÚcursorÚexecuteÚopenÚcsvÚreaderÚnextÚ executemany)r< r= ÚcurÚstmtÚcsvfilerD r9 r9 r: Ú create_and_load_iris_sqlite3
s
|
|
rJ )r= r* c
|
sþ ddl m} ddlm} t|}|jd dÈ}t |¡}t| fdd|D }|| |¡} t
|
| |r¸| ¡ >} | ¡ * |j | dd |j| d | | ¡ W 5 Q R X W 5 Q R X n8| ¡ * |j | dd |j| d | | ¡ W 5 Q R X W 5 Q R X d S )
|
Nr ©Úinsert©ÚEnginer> c s g | ]}t t |qS r9 ©ÚdictÚzip©Ú.0Úrow©Úheaderr9 r: Ú
|
<listcomp> s z(create_and_load_iris.<locals>.<listcomp>T©Z
|
checkfirst©Úbind)r6 rL Úsqlalchemy.enginerN r; rB rC rD rE ÚvaluesÚ
|
isinstanceÚconnectÚbeginÚdropÚcreaterA )
|
r< r= r* rL rN r1 rI rD ÚparamsrH r9 rU r: Úcreate_and_load_iris s$
|
|
|
|
|
rc c
|
C s¤ d}t | tjr$| ¡ }| |¡ n|ddlm} ddlm} ||}t | |r| ¡ $} |
|
¡ | |¡ W 5 Q R X W 5 Q R X n |
|
¡ | |¡ W 5 Q R X d S )Nz+CREATE VIEW iris_view AS SELECT * FROM irisr ©ÚtextrM ) r] Úsqlite3Ú
|
Connectionr@ rA r6 re r[ rN r^ r_ )r< rH rG re rN r9 r9 r: Úcreate_and_load_iris_view® s
|
|
|
|
rh c C s ddl m}m}m}m}m}m}m}m} | dkr4|n|} | dkrD|n|}
|
| } |d| |d||d| |d||d||d ||d
|
||d |
|
|d ||d |
|
} | dkr¾| |d|dd¡ | S )Nr )ÚTEXTÚBooleanr, ÚDateTimer- ÚIntegerr. r0 r# ÚtypesÚTextColÚDateColÚ
|
IntDateColÚIntDateOnlyColÚFloatColÚIntColÚBoolColÚIntColWithNullÚBoolColWithNullr% Ú DateColWithTzT)Útimezone)
|
r6 ri rj r, rk r- rl r. r0 Z append_column) r* ri rj r, rk r- rl r. r0 Z date_typeZ bool_typer8 rm r9 r9 r: Útypes_table_metadataÁ s( ( õ ry z
|
list[dict])r< Ú
|
types_datac C s* | ¡ }d}| |¡ d}| ||¡ d S )Na CREATE TABLE types (
|
"TextCol" TEXT,
|
"DateCol" TEXT,
|
"IntDateCol" INTEGER,
|
"IntDateOnlyCol" INTEGER,
|
"FloatCol" REAL,
|
"IntCol" INTEGER,
|
"BoolCol" INTEGER,
|
"IntColWithNull" INTEGER,
|
"BoolColWithNull" INTEGER
|
)zY
|
INSERT INTO types
|
VALUES(?, ?, ?, ?, ?, ?, ?, ?, ?)
|
)r@ rA rF )r< rz rG rH r9 r9 r: Ú create_and_load_types_sqlite3â s
|
|
r{ )rz r* c
|
C s ddl m} ddlm} t|}|| |¡}t| |r| ¡ >} | ¡ * |j | dd |j
|
| d | |¡ W 5 Q R X W 5 Q R X n8| ¡ * |j | dd |j
|
| d | |¡ W 5 Q R X d S )Nr rK rM TrX rY ) r6 rL r[ rN ry r\ r] r^ r_ r` ra rA )r< rz r* rL rN rm rH r9 r9 r: Úcreate_and_load_typesø s
|
|
|
|
r| r ©Úframec C sP | j d j}| jd }t|tjs&tt |j dddddg¡ | j
|
dksLtd S )Nr çffffff@ç @çffffffö?çÉ?ú Iris-setosa))é é )é r
) ÚdtypesÚtypeÚilocÚ
|
issubclassÚnpÚfloatingÚAssertionErrorÚtmÚ equalContentsr\ Úshape)r~ ZpytyperT r9 r9 r: Úcheck_iris_frame s
|
|
r )Ú
|
table_namec
|
C sâ d| }t | tjr0| ¡ }| |¡ ¡ d S ddlm} ddlm } t | t
|
rz:|| }| ¡ } | |¡ ¡ W 5 Q R £ W ¢ S Q R X W 5 | ¡ X nDt | |rÐ| ¡ } | |¡ ¡ W 5 Q R £ S Q R X n| |¡ ¡ S d S )Nz SELECT count(*) AS count_1 FROM r )Ú create_enginerM )r] rf rg r@ rA Úfetchoner6 r r[ rN r) Údisposer^ Úexec_driver_sqlZ
|
scalar_one)r< r rH rG r rN Úenginer9 r9 r: Ú
|
count_rows s
|
|
|
(
|
|
"r c C s | dddd}t |S )NÚioÚdatarC ziris.csvr )ÚdatapathÚ iris_pathr9 r9 r: r + s r c C s4 ddddddddddd
|
dd
|
d d dddd d d d
|
gS )NÚfirstú2000-01-03 00:00:00i wð i·2g333333$@é Fz2000-01-01 00:00:00-08:00)
|
rn ro rp rq rr rs rt ru rv rw ú2000-01-04 00:00:00i 'âPi\¸2z2000-06-01 00:00:00-07:00r9 r9 r9 r9 r: rz 1 s. ö öórz c
|
C s2 dddddddddd }t | }|| ¡ |¡S )Nr) Úint64Úfloat) rn ro rp rq rr rs rt ru rv )r ÚkeysÚastype)rz r Údfr9 r9 r: Útypes_data_frameO s ÷ r¦ c C s& dddddg} dddd g}t || d
|
S ) NÚindexÚAÚBÚCÚD)r gNÿ\^ï?g9\x`| @g%Ú TO׿gá(nIò¿)r gî}ÒCÄð?çc×^ÜV ¥¿g~q"B¶Ä¿gæÐ4Ë?)ú2000-01-05 00:00:00gGà Í¿èß?ç2r¹eç?g>ݦ4á¿gÑöÙRõ?)ú2000-01-06 00:00:00gÔ?{mXìñ?ç$ôóùù?gnk3¥AÓm?g Ø
|
S«å?©Úcolumns©r ©r² r r9 r9 r: Ú test_frame1` s ê rµ c C s" dddg} ddddg}t || dS ) Nr§ r¨ r© )r iÿÿÿgZØÓÍÿ¿)r iãÿÿÿr¬ )r i N r® )r¯ iÍûÿr° r± r³ r´ r9 r9 r: Ú test_frame3 s
|
ür¶ c
|
c sÊ t d¡}t d¡}|jdd|jjji|jjd}| |¡}| d¡sRt
|
|| d | d¡s||D ]}| d ¡ q`t ||d |V | ¡ .}| ¡ | d
|
¡}| |¡ W 5 Q R X W 5 Q R X | ¡ d S ) Nr6 Úpymysqlz*mysql+pymysql://root@localhost:3306/pandasÚ client_flag)Ú connect_argsÚ poolclassr1 r$ rm rw ú DROP TABLE IF EXISTS test_frame;)ÚpytestÚ importorskipr Ú constantsÚCLIENTÚMULTI_STATEMENTSÚpoolÚNullPoolÚinspectÚ has_tablerc Úpopr| r^ r_ re rA r ) r rz r6 r· r ÚinspÚentryr< rH r9 r9 r: Úmysql_pymysql_engine s(
|
|
ý
|
|
|
|
|
|
rÈ c c s | ¡ }|V W 5 Q R X d S ©N©r^ )rÈ r< r9 r9 r: Úmysql_pymysql_conn¨ s
|
rË c
|
c sª t d¡}t d¡ |jd|jjd}| |¡}| d¡sFt|| d | d¡s\t||d |V | ¡ .}|
|
¡ | d¡}| |¡ W 5 Q R X W 5 Q R X | ¡ d S ) Nr6 Úpsycopg2z=postgresql+psycopg2://postgres:postgres@localhost:5432/pandas©rº r1 r% rm r» )r¼ r½ r rÁ r rà rÄ rc r| r^ r_ re rA r )r rz r6 r rÆ r< rH r9 r9 r: Úpostgresql_psycopg2_engine® s"
|
|
þ
|
|
|
|
|
|
rÎ c c s | ¡ }|V W 5 Q R X d S rÉ rÊ )rÎ r< r9 r9 r: Úpostgresql_psycopg2_connà s
|
rÏ c c s, t d¡ t ¡ } d| V W 5 Q R X d S )Nr6 ú
|
sqlite:///)r¼ r½ r Ú ensure_clean©Únamer9 r9 r: Ú
|
sqlite_strÉ s
|
|
rÔ c c s. t d¡}|j| |jjd}|V | ¡ d S )Nr6 rÍ )r¼ r½ r rÁ r r )rÔ r6 r r9 r9 r: Ú sqlite_engineÐ s
|
rÕ c c s | ¡ }|V W 5 Q R X d S rÉ rÊ )rÕ r< r9 r9 r: Ú sqlite_connØ s
|
rÖ c C s, t d¡}| | ¡}t||d | ¡ | S )Nr6 r# )r¼ r½ r rc r )rÔ r r6 r r9 r9 r: Úsqlite_iris_strÞ s
|
|
|
r× c C s t | |d | S )Nr# )rc )rÕ r r9 r9 r: Úsqlite_iris_engineç s rØ c c s | ¡ }|V W 5 Q R X d S rÉ rÊ )rØ r< r9 r9 r: Úsqlite_iris_conní s
|
rÙ c
|
c s6 t t d¡¡ } | }|V W 5 Q R X W 5 Q R X d S ©Nú:memory:)Ú
|
contextlibr rf r^ )Z closing_connr< r9 r9 r: Úsqlite_buildinó s rÝ c C s t | | | S rÉ )rJ )rÝ r r9 r9 r: Úsqlite_buildin_irisú s
|
rÞ r< c C s | | ¡} |jd| ddd d S )NÚtestÚappendF©Ú if_existsr§ )ÚgetfixturevalueÚto_sql)r< rµ Úrequestr9 r9 r: Útest_dataframe_to_sql" s
|
ræ c C s t d¡ ttjdgddtjtdddgddtjtdgddtjdgd dd
|
}| | ¡} tj t
|
d d |j d | ddd W 5 Q R X d S )NÚpyarrowr z int8[pyarrow]©r7 éç útimestamp[ns][pyarrow]zduration[ns][pyarrow]Úazstring[pyarrow])Úintr r Ústringzthe 'timedelta'©ÚmatchÚ
|
test_arrowÚreplaceFrá ) r¼ r½ r ÚpdÚarrayr r rã r Úassert_produces_warningÚ UserWarningrä ©r< rå r¥ r9 r9 r: Ú"test_dataframe_to_sql_arrow_dtypes* s
|
ÿúÿ
|
|
r÷ c C sL t d¡ tdtjtddd|gddi}| | ¡} |jd| dd d
|
d S ) Nrç r ré r rê rè rð rñ Frá )r¼ r½ r rò ró r rã rä )r< rå Z nulls_fixturer¥ r9 r9 r: Ú*test_dataframe_to_sql_arrow_dtypes_missing> s
|
ÿÿÿ
|
rø ÚmethodZmultic C sZ | | ¡} t| dd$}|j|d|d | d¡s6tW 5 Q R X t| dt|ksVtd S )NT©Zneed_transactionÚ
|
test_frame©rù ©rã r rä rÄ r r Úlen)r< rù rµ rå Ú pandasSQLr9 r9 r: Ú test_to_sqlN s
|
|
r zmode, num_row_coef)rñ r )rà é c C sn | | ¡} t| dd4}|j|ddd |j|d|d | d¡sFtW 5 Q R X t| d|t| ksjtd S )NTrú rû Úfail©râ rý )r< ÚmodeZ num_row_coefrµ rå rÿ r9 r9 r: Útest_to_sql_existY s
|
r c
|
C sr | | ¡} t| ddR}|j|ddd | d¡s6td}tjt|d |j|ddd W 5 Q R X W 5 Q R X d S )NTrú rû r r z!Table 'test_frame' already existsrî )rã r rä rÄ r r¼ ÚraisesÚ
|
ValueError)r< rµ rå rÿ Úmsgr9 r9 r: Útest_to_sql_exist_faile s
|
r c C s\ | | ¡} td| }t| t d| ¡}t| t d| ¡}|jdksJtd|jksXtd S )NúSELECT * FROM irisú SELECT * FROM iris where 0=1©r r
r3 )rã r! r rò Úread_sqlr r r² ©r< rå Ú
|
iris_framer9 r9 r: Útest_read_iris_queryr s
|
|
r c C st | | ¡} ttd| dd}t| ttjd| dd}t| ttjd| dd}|jdksbtd|jksptd S )Nr
|
é ©Ú chunksizer r r3 ) rã r r! r rò r r r r² r r9 r9 r: Ú test_read_iris_query_chunksize s
|
r c
|
C s | | ¡} ddlm}m}m}m} | }t| tr:|| n| }|d||d}t||| dddd} t | t| tr~|
|
¡ d S ) Nr )r. r0 r Úselectr1 )Z autoload_withr r ©rÓ Úlength©rb ) rã r6 r. r0 r r r] r) r! r r )
|
r< rå r. r0 r r r8 Z autoload_conr1 r r9 r9 r: Ú.test_read_iris_query_expression_with_parameter s
|
ÿ
|
r c C sV t d ¡ D ]\}}|| kr q2q td| d| | ¡} t|| dd}t| d S )Nr& z No part of z( found in SQL_STRINGS['read_parameters']©r r r )Ú SQL_STRINGSÚitemsÚKeyErrorrã r! r )r< rå ÚdbÚqueryr r9 r9 r: Ú*test_read_iris_query_string_with_parameter¢ s
|
r c C s4 | | ¡} td| }t| t d| ¡}t| d S ©Nr1 )rã r" r rò r r r9 r9 r: Útest_read_iris_table¯ s
|
|
|
r" c C sD | | ¡} ttd| dd}t| ttjd| dd}t| d S )Nr1 r r )rã r r" r rò r r r9 r9 r: Ú test_read_iris_table_chunksizeº s
|
|
r# c sx | | ¡} g fdd}t| dd$}|j|d|d | d¡sFtW 5 Q R X dgks^tt| dt|ksttd S )Nc s2 d¡ fdd|D }| | j ¡ |¡ d S )Nr c s g | ]}t t |qS r9 rO rR ©r£ r9 r: rW Í s z8test_to_sql_callable.<locals>.sample.<locals>.<listcomp>)rà rA ÚtablerL )Zpd_tabler< r£ Ú data_iterr ©Úcheckr$ r: ÚsampleË s
|
z$test_to_sql_callable.<locals>.sampleTrú rû rü r rý )r< rµ rå r) rÿ r9 r' r: Útest_to_sql_callableÄ s
|
r* c C s | | ¡} t d| ¡}t|jjjtjs,t t|j
|
jjtj sBt t|j jjtj sXt t|j jjtjsnt t|jjjtjst d S ©Nrm )rã r r" r rr r7 r r r r rs Úintegerrt ru rv rö r9 r9 r: Ú test_default_type_conversion× s
|
r- c
|
C sè | | ¡} ddlm} ddlm} tdddgddd gd
|
}|jd | d d d}||}t| |r| ¡ $}| ¡ |
|
|¡ W 5 Q R X W 5 Q R X n | ¡ |
|
|¡ W 5 Q R X t d| ¡}t ||¡ t d| ¡}t ||¡ d S )Nr rd rM r r é ç¹?r g333333Ó?©rë Úbrû F©r§ z}DROP PROCEDURE IF EXISTS get_testdb;
|
|
CREATE PROCEDURE get_testdb ()
|
|
BEGIN
|
SELECT * FROM test_frame;
|
ENDzCALL get_testdb();)rã r6 re r[ rN r rä r] r^ r_ rA r r! r Úassert_frame_equalr ) r< rå re rN r¥ ÚprocZ engine_connÚres1Úres2r9 r9 r: Útest_read_procedureê s"
|
|
|
|
|
r7 Úexpected_countr zSuccess!c s fdd}| | ¡} tddgddgddgd }|jd
|
| d |d }t ts\|d kshtn | kshtt d
|
| ¡}t ||¡ d S ) Nc s |j }| ¡ ~}t }t |¡}| |¡ | d¡ d dd |D ¡}| jrb| j d| j } n| j } d| d| d}
|
|j
|
|
|
|d W 5 Q R X S )
|
Nr z, c S s g | ]}d | d qS )ú"r9 )rS Úkr9 r9 r: rW s zVtest_copy_from_callable_insertion_method.<locals>.psql_insert_copy.<locals>.<listcomp>Ú.zCOPY z (z) FROM STDIN WITH CSV)r Úfile) Ú
|
connectionr@ r rC ÚwriterÚ writerowsÚseekÚjoinÚschemarÓ Z copy_expert) r% r< r£ r& Z
|
dbapi_connrG Zs_bufr> r² r Z sql_query©r8 r9 r: Úpsql_insert_copy s
|
|
|
|
zBtest_copy_from_callable_insertion_method.<locals>.psql_insert_copyr r r/ r rë Ún©Úcol1Úcol2Zcol3rû F)r§ rù )
|
rã r rä r] rì r r r" r r3 )r< r8 rå rD ÚexpectedÚ result_countÚresultr9 rC r: Ú(test_copy_from_callable_insertion_method s
|
ÿ
|
rL c
|
C sD t jtdd, tjtdd t d| ¡ W 5 Q R X W 5 Q R X d S )Nz+pandas.io.sql.execute requires a connectionrî úP`pandas.io.sql.execute` is deprecated and will be removed in the future version.úselect * from iris)r¼ r Ú TypeErrorr rô Ú FutureWarningr rA )rØ r9 r9 r: Útest_execute_typeerror8 s þrQ c C s* t jtdd t d| ¡ W 5 Q R X d S )NrM rî rN )r rô rP r rA )rÞ r9 r9 r: Útest_execute_deprecatedB s
|
þrR c @ s e Zd Zdd ZdS )Ú MixInBasec C s t | dr| j ¡ z | ¡ }W n tjjtjfk
|
r> Y nJX |> | |¡D ]}| ||¡ qP|
|
|¡D ]}| ||¡ qlW 5 Q R X d S )Nr< ) Úhasattrr< Úcloser^ r6 ÚexcÚOperationalErrorrf Ú_get_all_viewsÚ drop_viewÚ_get_all_tablesÚ
|
drop_table)Úselfr< ÚviewÚtblr9 r9 r: Úteardown_methodM s
|
|
zMixInBase.teardown_methodN)Ú__name__Ú
|
__module__Ú __qualname__r_ r9 r9 r9 r: rS L s rS c @ s4 e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd S ) Ú SQLiteMixInc C s
|
t d¡S rÚ ©rf r^ ©r\ r9 r9 r: r^ _ s zSQLiteMixIn.connectc C s" | dt |¡ ¡ | ¡ d S ©NzDROP TABLE IF EXISTS ©rA r Ú_get_valid_sqlite_nameÚcommit©r\ r r< r9 r9 r: r[ b s zSQLiteMixIn.drop_tablec C s | d¡}dd | ¡ D S )Nz1SELECT name FROM sqlite_master WHERE type='table'c S s g | ] }|d qS ©r r9 )rS r% r9 r9 r: rW h s z/SQLiteMixIn._get_all_tables.<locals>.<listcomp>©rA Úfetchall©r\ r< Úcr9 r9 r: rZ f s
|
zSQLiteMixIn._get_all_tablesc C s" | dt |¡ ¡ | ¡ d S ©NzDROP VIEW IF EXISTS rg )r\ Ú view_namer< r9 r9 r: rY j s zSQLiteMixIn.drop_viewc C s | d¡}dd | ¡ D S )Nz0SELECT name FROM sqlite_master WHERE type='view'c S s g | ] }|d qS rk r9 )rS r] r9 r9 r: rW p s z.SQLiteMixIn._get_all_views.<locals>.<listcomp>rl rn r9 r9 r: rX n s
|
zSQLiteMixIn._get_all_viewsN)r` ra rb r^ r[ rZ rY rX r9 r9 r9 r: rc ^ s
|
rc c @ s@ e Zd Zedd Zdd Zdd Zdd Zd d
|
Zd d Z d S )ÚSQLAlchemyMixInc C s | j ¡ d S rÉ )r r ©Úclsr9 r9 r: Úteardown_classt s z SQLAlchemyMixIn.teardown_classc C s
|
| j ¡ S rÉ )r r^ re r9 r9 r: r^ x s zSQLAlchemyMixIn.connectc C s< | ¡ r| ¡ ¡ | ¡ t |¡ |¡ W 5 Q R X d S rÉ )Úin_transactionÚget_transactionÚrollbackr_ r r r[ rj r9 r9 r: r[ { s
|
zSQLAlchemyMixIn.drop_tablec C s ddl m} || ¡ S ©Nr ©rà )r6 rà Zget_table_names©r\ r< rà r9 r9 r: rZ s z SQLAlchemyMixIn._get_all_tablesc C sL |j jj |¡}| ¡ r$| ¡ ¡ | ¡ | d| ¡ W 5 Q R X d S rp ) r r* Zidentifier_preparerZquote_identifierrv rw rx r_ r )r\ rq r< Z quoted_viewr9 r9 r: rY s
|
ÿ
|
zSQLAlchemyMixIn.drop_viewc C s ddl m} || ¡ S ry )r6 rà Zget_view_namesr{ r9 r9 r: rX s z SQLAlchemyMixIn._get_all_viewsN)
|
r` ra rb Ú classmethodru r^ r[ rZ rY rX r9 r9 r9 r: rr s s
|
rr c @ sj e Zd ZdZdd Zdd Zdd Zdd Zd
|
d Zd d Z dddZ
|
dd Z dd Z dd Z dd ZdS )Ú PandasSQLTestzT
|
Base class with common private methods for SQLAlchemy and fallback cases.
|
|
c C s> | d| j¡ t| jtjr*t| j| nt| j|| j d S r! )r[ r< r] rf rg rJ rc Úflavor)r\ r r9 r9 r: Úload_iris_data s z PandasSQLTest.load_iris_datac C s\ | j dkr |D ]}| d¡ qt| jtjrHdd |D }t| j| nt| j|| j d S )Nr% rw c S s g | ]}t | ¡ qS r9 )Útupler\ )rS rÇ r9 r9 r: rW § s z1PandasSQLTest.load_types_data.<locals>.<listcomp>)r~ rÅ r] r< rf rg r{ r| )r\ rz rÇ r9 r9 r: Úload_types_data¢ s
|
z PandasSQLTest.load_types_datac C s. t d | j }d}| jj||d}t| d S )Nr& r r ©r r~ rÿ Ú
|
read_queryr ©r\ r rb r r9 r9 r: Ú_read_sql_iris_parameter¬ s z&PandasSQLTest._read_sql_iris_parameterc C s4 t d | j }ddd}| jj||d}t| d S )Nr' r r r r r r r9 r9 r: Ú _read_sql_iris_named_parameter² s
|
z,PandasSQLTest._read_sql_iris_named_parameterc C s* t d | j }| jj|d d}t| d S )Nr( r r )r\ r r r9 r9 r: Ú(_read_sql_iris_no_parameter_with_percent¸ s z6PandasSQLTest._read_sql_iris_no_parameter_with_percentc C s2 | d| j¡ | j |jd d
d¡dks.td S )Nrµ r )r[ r< rÿ rä r r ©r\ rµ r9 r9 r: Ú _to_sql_empty½ s zPandasSQLTest._to_sql_emptyÚautoc K sr | d| j¡ | jj|dfd|i|dks0t| j d¡s@tt|}t| jd}||ks`t| d| j¡ dS )z `to_sql` with the `engine` paramrµ r é N)r[ r< rÿ rä r rÄ rþ r )r\ rµ r Z engine_kwargsÚ num_entriesÚnum_rowsr9 r9 r: Ú_to_sql_with_sql_engineÁ s" ÿÿÿýÿ z%PandasSQLTest._to_sql_with_sql_enginec C sV | d| j¡ | j |d¡dks$t| j d¡}|jddd d |j_t
|
||¡ d S )NÚtest_frame_roundtripr ú"SELECT * FROM test_frame_roundtripÚlevel_0T©Zinplace) r[ r< rÿ rä r r Ú set_indexr§ rÓ r r3 ©r\ rµ rK r9 r9 r: Ú
|
_roundtripÕ s zPandasSQLTest._roundtripc C s. | j d¡}| ¡ }t |dddddg¡ d S ©Nr
|
r r r r r )rÿ rA r r r )r\ Ú iris_resultsrT r9 r9 r: Ú _execute_sqlá s zPandasSQLTest._execute_sqlc C sP t jddgdddgdgd}| j |d¡dks2t| d¡}|dggksLtd S ) N©r çÍÌÌÌÌÌ @Zline1©r ç ø?Zline2r¨ r© rª ©r² r§ Ztest_to_sql_saves_indexr )r Ú from_recordsrÿ rä r Ú_get_index_columns)r\ r¥ Úix_colsr9 r9 r: Ú_to_sql_save_indexç s ÿ
|
z PandasSQLTest._to_sql_save_indexc C s$ | j ¡ @}d}t| j tr(| |¡ n ddlm} ||}| |¡ W 5 Q R X G dd dt}d}t| j trddlm} ||}z,| j ¡ }| |¡ |dW 5 Q R X W n |k
|
rÄ Y nX | j d¡}t
|
|dksât | j ¡ }| |¡ W 5 Q R X | j d¡}t
|
|d ks t d S )
|
Nz'CREATE TABLE test_trans (A INT, B TEXT)r rd c @ s e Zd ZdS )z7PandasSQLTest._transaction_test.<locals>.DummyExceptionN)r` ra rb r9 r9 r9 r: ÚDummyExceptionú s r¢ z/INSERT INTO test_trans (A,B) VALUES (1, 'blah')ÚerrorzSELECT * FROM test_transr ) rÿ Zrun_transactionr] r rA r6 re Ú Exceptionr r rþ r )r\ ZtransrH re r¢ Zins_sqlÚresr6 r9 r9 r: Ú_transaction_testï s0
|
z PandasSQLTest._transaction_testN)r )r` ra rb Ú__doc__r r r
r r r r r r r¡ r¦ r9 r9 r9 r: r} s
|
|
r} c
|
@ s¢ e Zd ZU dZdZded< ejdddd Zd d
|
Z d d Z
|
d d Z dd Z dd Z dd Zdd Zdd Zdd Zdd Zd d Zd d Zd!d" Zej d#d$d%d&g¡ej d'ejd(d)fejd*d+fejd(d)fejd*d+fg¡d,d- Z d.d/ Z d0d1 Z d2d3 Z ej d4d5d6d7d8d9d:g¡d;d< Z d=d> Z!d?d@ Z"ej dAdBe#e$e#e$dCg¡dDdE Z%dFdG Z&dHdI Z'dJdK Z(dLdM Z)dNdO Z*dPdQ Z+dRdS Z,dTdU Z-dVdW Z.dBS )XÚ _TestSQLApiañ
|
Base class to test the public API.
|
|
From this two classes are derived to run these tests for both the
|
sqlalchemy mode (`TestSQLApi`) and the fallback mode
|
(`TestSQLiteFallbackApi`). These tests are run with sqlite3. Specific
|
tests for the different sql flavours are included in `_TestSQLAlchemy`.
|
|
Notes:
|
flavor can always be passed even in SQLAlchemy mode,
|
should be correctly ignored.
|
|
we don't use drop_table because that isn't part of the public api
|
|
r# r) r T©Zautousec C s* | ¡ | _| |¡ | |¡ | ¡ d S rÉ )r^ r< r r Úload_test_data_and_sql©r\ r rz r9 r9 r: Ú setup_method, s
|
|
|
z_TestSQLApi.setup_methodc C s t | j d S rÉ )rh r< re r9 r9 r: rª 3 s z"_TestSQLApi.load_test_data_and_sqlc C s t d| j¡}t| d S )NzSELECT * FROM iris_view)r r! r< r ©r\ r r9 r9 r: Útest_read_sql_view6 s z _TestSQLApi.test_read_sql_viewc C s8 d}t j|| jdd}t || j¡}t t||¡ d S )Nz/SELECT * FROM iris_view WHERE SepalLength < 0.0r
r )r r! r< r r3 r )r\ r Z
|
with_batchZ without_batchr9 r9 r: Ú&test_read_sql_with_chunksize_no_result: s z2_TestSQLApi.test_read_sql_with_chunksize_no_resultc C s& t |d| j¡ t d| j¡s"td S )Nrµ )r rä r< rÄ r r r9 r9 r: r @ s z_TestSQLApi.test_to_sqlc C s\ t j|d| jdd t d| j¡s&td}tjt|d t j|d| jdd W 5 Q R X d S )NÚ test_frame2r r z"Table 'test_frame2' already existsrî )r rä r< rÄ r r¼ r r ©r\ rµ r r9 r9 r: Útest_to_sql_failD s
|
z _TestSQLApi.test_to_sql_failc C s^ t j|d| jdd t j|d| jdd t d| j¡s:tt|}t| jd}||ksZtd S )Nr¶ r r rñ )r rä r< rÄ r rþ r ©r\ rµ r r r9 r9 r: Útest_to_sql_replaceL s z _TestSQLApi.test_to_sql_replacec C sr t j|d| jdddks tt j|d| jdddks8tt d| j¡sJtdt| }t| jd}||ksntd S )NÚ test_frame4r r r rà r )r rä r< r rÄ rþ r r³ r9 r9 r: Útest_to_sql_appendW s ÿ z _TestSQLApi.test_to_sql_appendc C s2 t j|d| jdd t d| j¡}t ||¡ d S )NZ test_frame5Fr2 zSELECT * FROM test_frame5)r rä r< r r r3 )r\ r¶ rK r9 r9 r: Útest_to_sql_type_mappinge s z$_TestSQLApi.test_to_sql_type_mappingc C sL t tjddddd}tj|d| jdd t d | j¡}t | ¡ |¡ d S )
|
Nr
r¡ rè ZseriesrÒ Z test_seriesFr2 zSELECT * FROM test_series)
|
r r Úaranger rä r< r! r r3 Úto_frame)r\ ÚsÚs2r9 r9 r: Útest_to_sql_seriesk s z _TestSQLApi.test_to_sql_seriesc C s\ t j|d| jd t jd| jd}|j|_|jddd |j t¡ d |j_t
|
||¡ d S )Nr ©Úconr r Tr ) r rä r< r! r§ r r¤ rì rÓ r r3 r r9 r9 r: Útest_roundtripq s z_TestSQLApi.test_roundtripc C s6 t j|d| jddd t jd| jd}t ||¡ d S )Nr Fr )r¾ r§ r r r½ )r rä r< r! r r3 r r9 r9 r: Útest_roundtrip_chunksize| s ûz$_TestSQLApi.test_roundtrip_chunksizec C sD t | j¡}| d¡}W 5 Q R X | ¡ }t |dddddg¡ d S r )r r r< rA r r r )r\ Ú
|
pandas_sqlr rT r9 r9 r: Útest_execute_sql s z _TestSQLApi.test_execute_sqlc C sò t d| j¡}t|jjjtjr$t t jd| jdgd}t|jjjtjsNt |j
|
¡ t ddddddt ddddddgkst t jd| jdd id}t|jjjtjs¬t |j
|
¡ t ddddddt ddddddgksÞt t jd| jd
|
gd}t|j jjtjs
|
t |j
|
¡ t d d d dddt ddddddgks>t t jd| jd
|
did}t|j jjtjslt |j
|
¡ t d d d dddt ddddddgks t t jd| jddid}t|j jjtjsÎt |j
|
¡ t dt dgksît d S )NúSELECT * FROM typesro ©Ú parse_datesiÐ r r. r r ú%Y-%m-%d %H:%M:%Srp i é é iÝ rº rq z%Y%m%dz
|
2010-10-10z
|
2010-12-12)r r! r< r ro r7 r r Ú
|
datetime64r Útolistr rp rq ©r\ r¥ r9 r9 r: Útest_date_parsing sh ÿþ
|
ýþ
|
ÿþ ÿþ ýþz _TestSQLApi.test_date_parsingr£ ÚignoreÚraiseÚcoercezread_sql, text, moderà )r6 Úfallbackrm r6 c C s@ | j |kr<| ddi¡}||| jdd|iid}t ||¡ d S )Nro zdatetime64[ns]Úerrors)r¾ rÅ )r r¤ r< r r3 )r\ r re r r£ r¦ rI rK r9 r9 r: Ú test_custom_dateparsing_errorÅ s
|
ÿýz)_TestSQLApi.test_custom_dateparsing_errorc C sH t jd| jdddgd}t|jjjtjs.t t|j
|
jjtjsDt d S )Nrà ro rp )Ú index_colrÅ ) r r! r< r r§ r7 r r rÉ r rp rË r9 r9 r: Útest_date_and_indexã s üz _TestSQLApi.test_date_and_indexc C st t tddgdd ¡ }t t¡ | d| j¡}W 5 Q R X |dksHtt
|
d| j¡}t |d |d d¡¡ d S ) Nz00:00:01z00:00:03ÚfoorÒ Útest_timedeltar z SELECT * FROM test_timedeltar¡ ) r r r¹ r rô rõ rä r< r r r! Úassert_series_equalr] )r\ r¥ rJ rK r9 r9 r: rÖ ð s z_TestSQLApi.test_timedeltac C sH t dddgi}d}tjt|d | d| j¡d ks:tW 5 Q R X d S )Nrë y ð? ð?y @z Complex datatypes not supportedrî Z test_complex)r r¼ r r rä r< r )r\ r¥ r r9 r9 r: Útest_complex_raisesù s z _TestSQLApi.test_complex_raisesz index_name,index_label,expected)NNr§ )NÚ other_labelrÙ )Ú
|
index_nameNrÚ )rÚ rÙ rÙ )r NÚ0)Nr rÛ c C sT t dtdi}||j_d}tj|d| j|d t || j¡}|jd |ksPt d S )NrG r ú SELECT * FROM test_index_labelÚtest_index_label©Ú index_labelr )
|
r Úranger§ rÓ r rä r< r! r² r )r\ rÚ rß rI Ú
|
temp_framer r~ r9 r9 r: Útest_to_sql_index_labelÿ s z#_TestSQLApi.test_to_sql_index_labelc C sª d}t dtdit ddg¡d}t |d| j¡}||ks>tt d| j¡}|j d d ks^t|j d
|
d kspttj|d| jd d dgd}||kstt d| j¡}|j d d
|
¡ d dgksÂtd dg|j _ tj|d| jd d}||ksîtt d| j¡}|j d d
|
¡ d dgks ttj|d| jd ddgd}||ksDtt d| j¡}|j d d
|
¡ ddgksrtd}t jt|d tj|d| jd dd W 5 Q R X d S )Nr rG )ZA0ÚA1)ZB0ZB1r2 rÝ rÜ r r r Zlevel_1rñ r¨ r© )râ rß r r rª r« zALength of 'index_label' should match number of levels, which is 2rî )r rà r Z from_productr rä r< r r! r² rÊ r§ Únamesr¼ r r )r\ Zexpected_row_countrá rK r~ r r9 r9 r: Ú"test_to_sql_index_label_multiindex sb
|
þ û ÿ û ûz._TestSQLApi.test_to_sql_index_label_multiindexc C sV t jddgdddgddgd}| d| j¡ tjd| jddgd }tj||d
|
d d S ) Nr r r¨ r© rª r Útest_multiindex_roundtripz'SELECT * FROM test_multiindex_roundtrip©rÓ T©Zcheck_index_type)r r rä r< r r! r r3 ©r\ r¥ rK r9 r9 r: ræ Q s ý ÿz%_TestSQLApi.test_multiindex_roundtripr7 N©r¨ r© c C s^ t ddgddggddgd}| d| j¡d ks2t| |¡}tjd
|
| j|d }t ||¡ d S ) Nç333333ó?g333333 @çffffff@g333333 @r¨ r© r± Útest_dtype_argumentr z$SELECT A, B FROM test_dtype_argument©r¾ r7 ) r rä r< r r¤ r r! r r3 )r\ r7 r¥ rI rK r9 r9 r: rí ^ s
|
ÿz _TestSQLApi.test_dtype_argumentc C s4 t ddgddggddgd}tj|d| jdd d S )
|
Nr r r. r r r± Z test_frame_integer_col_namesrñ r )r r rä r< rË r9 r9 r: Útest_integer_col_namess s z"_TestSQLApi.test_integer_col_namesc C s" t j|d| jd}d|ks td S )Nrß r½ ÚCREATE©r Ú
|
get_schemar< r ©r\ rµ Ú
|
create_sqlr9 r9 r: Útest_get_schemaw s z_TestSQLApi.test_get_schemac C s$ t j|d| jdd}d|ks td S )Nrß Úpypi)r¾ rB zCREATE TABLE pypi.rñ ró r9 r9 r: Útest_get_schema_with_schema{ s z'_TestSQLApi.test_get_schema_with_schemac C sj | j dkr ddlm} |}nd}tddgddgd }tj|d
|
| jd |id }d |ksZtd|ksftd S )Nr6 r )rl ÚINTEGERçñ?rë r ç@r0 rß r1 rî rð )r r6 rl r r rò r< r )r\ rl r7 Z float_framerô r9 r9 r: Útest_get_schema_dtypes s
|
ÿ z"_TestSQLApi.test_get_schema_dtypesc C sf t ddgddgd}tj|d| jdd}d }||ks:ttj|d| jd
|
d gd}d }||ksbtd S ) Nrù rë r rú )ÚCol1ZCol2rß rü )r¾ r£ z'CONSTRAINT test_pk PRIMARY KEY ("Col1")r¨ r© z)CONSTRAINT test_pk PRIMARY KEY ("A", "B"))r r rò r< r )r\ rµ r~ rô Zconstraint_sentencer9 r9 r: Útest_get_schema_keys s z _TestSQLApi.test_get_schema_keysc C s t tj dd¡tdd}|jd| jdd t d| j¡}t }d }ddddd
|
g}tjd| jdd D ]0}t ||gd d }t
|
||| kst |d7 }qdt ||¡ | jdkrt }d }ddddd
|
g}tjd| jdd D ]2}t ||gd d }t
|
||| kst |d7 }qØt ||¡ d S )Né r
Zabcder± Ztest_chunksizeFr2 z select * from test_chunksizer r r T©Z ignore_indexr r6 )r r ÚrandomZrandnÚlisträ r< r r! r rþ r r r3 r r" )r\ r¥ r5 r6 ÚiÚsizesÚchunkÚres3r9 r9 r: Útest_chunksize_read s0 ÿ
|
|
|
z _TestSQLApi.test_chunksize_readc C sd t dddgdddgd}| ¡ }|d d¡|d< |jd | jd
|
d t d | j¡}t ||¡ d S ) Nr r r. z John P. Doez Jane Dove)Z person_idÚ person_namer ÚcategoryÚtest_categoricalFr2 z SELECT * FROM test_categorical) r Úcopyr¤ rä r< r r! r r3 )r\ r¥ Údf2r¥ r9 r9 r: r ¼ s þÿz _TestSQLApi.test_categoricalc C s2 t ddgddggddgd}|jd| jd d
|
d S ) Nr r r. r õ ér1 r± Z test_unicodeFr2 )r rä r< rË r9 r9 r: Útest_unicode_column_nameÍ s z$_TestSQLApi.test_unicode_column_namec C sL t dddgdtjdgd}|jd| jdd t d
|
| j¡}t ||¡ d S ) Nr r r r rì rê zd1187b08-4943-4c8d-a7f6Fr2 z'SELECT * FROM `d1187b08-4943-4c8d-a7f6`) r r Únanrä r< r r! r r3 ©r\ r¥ r¥ r9 r9 r: Útest_escaped_table_nameÒ s z#_TestSQLApi.test_escaped_table_name)/r` ra rb r§ r~ Ú__annotations__r¼ Úfixturer¬ rª r® r¯ r r² r´ r¶ r· r¼ r¿ rÀ r rÌ ÚmarkÚ parametrizer r r! r" rÒ rÔ rÖ rØ râ rå ræ rì r¢ rí rï rõ r÷ rû rý r r r r r9 r9 r9 r: r¨ s
|
|
|
7
|
|
ý
|
øþ ôþ
|
9 üþ
|
"r¨ zSQLAlchemy not installed©Úreasonc @ sô e Zd ZdZdZdZedd Zdd Zdd Z d
|
d Z
|
d d Z dd Z dd Z dd Zej ddddddddd d d d d d!d"ee e¡jd#krd$nd%fg¡d&d' Zej d(d)d*g¡d+d, Zd-d. Ze d/¡d0d1 Zd2d3 Z d4d5 Z d6d7 Z d8S )9Ú
|
TestSQLApiz
|
Test the public API as it would be used directly
|
|
Tests for `read_sql_table` are included here, as this is specific for the
|
sqlalchemy mode.
|
|
r# r6 c C s t d¡| _d S ©Nzsqlite:///:memory:©r6 r r rs r9 r9 r: Ú setup_classé s zTestSQLApi.setup_classc C s@ t |d| j¡ ddg}t jd| j|d}|j ¡ |ks<td S )Nrû r¨ r© r± )r rä r< r" r² rÊ r )r\ rµ ÚcolsrK r9 r9 r: Útest_read_table_columnsí s z"TestSQLApi.test_read_table_columnsc C s¨ t |d| j¡ t jd| jdd}|jjdgks4tt jd| jddgd}|jjddgks^tt jd| jddgddgd}|jjddgkst|j ¡ ddgks¤td S ) Nrû r§ rç r¨ r© rª r« )rÓ r² ) r rä r< r" r§ rä r r² rÊ r r9 r9 r: Útest_read_table_index_colõ s ÿz$TestSQLApi.test_read_table_index_colc C sT t d| j¡}t d| j¡}t ||¡ t d| j¡}t d| j¡}t ||¡ d S )Nr
|
r1 )r r! r< r r r3 r" )r\ Ú iris_frame1Ú iris_frame2r9 r9 r: Útest_read_sql_delegate s z!TestSQLApi.test_read_sql_delegatec C sÌ ddl m} ddlm} |d|dg}|D ]h}t| j|rr| j ¡ $}| ¡ | |¡ W 5 Q R X W 5 Q R X q,| j ¡ | j |¡ W 5 Q R X q,t
|
d ¡" t d| j¡ t d| j¡ W 5 Q R X d S )Nr rd rM z,CREATE TABLE invalid (x INTEGER, y UNKNOWN);z0CREATE TABLE other_table (x INTEGER, y INTEGER);Z other_tablezSELECT * FROM other_table)r6 re r[ rN r] r< r^ r_ rA r rô r r" r! )r\ re rN Z
|
query_listr r< r9 r9 r: Útest_not_reflect_all_tables s þ
|
z&TestSQLApi.test_not_reflect_all_tablesc C sV t jtdd t | j¡ dd¡ W 5 Q R X t d ¡ | d| j¡ W 5 Q R X d S )Nz¶The provided table name 'TABLE1' is not found exactly as such in the database after writing the table, possibly due to case sensitivity issues. Consider using lower case table names.rî ZTABLE1Ú Z CaseSensitive)r rô rõ r r r< Zcheck_case_sensitiverä r r9 r9 r: Ú(test_warning_case_insensitive_table_name$ s ý z3TestSQLApi.test_warning_case_insensitive_table_namec C s6 ddl m} |j | j¡}| d¡}dd |D }|S )Nr )Ú
|
reflectionZtest_index_savedc S s g | ] }|d qS ©Z column_namesr9 ©rS r r9 r9 r: rW 9 s z1TestSQLApi._get_index_columns.<locals>.<listcomp>)r[ r$ Z InspectorZ from_enginer< Ú get_indexes)r\ Útbl_namer$ rÆ Úixsr9 r9 r: r 4 s
|
|
z TestSQLApi._get_index_columnsc C s\ ddl m} tdtddgddi}t | j¡}tjd||d }t|j j
|
d j |sXt d S )
|
Nr )Ú TIMESTAMPr ú2014-12-12 01:54ú2014-12-11 02:54T©ÚutcÚ test_typer} ) r6 r* r r r r r< ÚSQLTabler] r% ro r r )r\ r* r¥ r r% r9 r9 r: Ú test_sqlalchemy_type_mapping< s ÿ z'TestSQLApi.test_sqlalchemy_type_mappingzinteger, expected)Zint8ÚSMALLINT)ZInt8r2 )Zuint8r2 )ZUInt8r2 )Úint16r2 )ZInt16r2 )Zuint16rø )ZUInt16rø )Úint32rø )ZInt32rø )Zuint32ÚBIGINT)ZUInt32r5 )r¡ r5 )ÚInt64r5 r¡ r5 rø c C sP t ddgdg|d}t | j¡}tjd||d}t|jjjj }||ksLt
|
d S )Nr r rë ©r² r7 r/ r} ) r r r r< r0 r) r% ro rë r r )r\ r, rI r¥ r r% rK r9 r9 r: Ú test_sqlalchemy_integer_mappingH s
|
z*TestSQLApi.test_sqlalchemy_integer_mappingr, Zuint64ZUInt64c C sN t ddgdg|d}t | j¡}tjtdd tjd||d W 5 Q R X d S ) Nr r rë r7 z1Unsigned 64 bit integer datatype is not supportedrî r/ r} )r r r r< r¼ r r r0 )r\ r, r¥ r r9 r9 r: Ú(test_sqlalchemy_integer_overload_mappinge s ÿz3TestSQLApi.test_sqlalchemy_integer_overload_mappingc C s t ¡ L}d| }d}|j||ddd t ||¡}t ||¡}d}t ||¡}W 5 Q R X t ||¡ t ||¡ t ||¡ d S )NrÐ r1 rñ Frá r
|
)r rÑ rä r r r" r! r3 ) r\ rµ rÓ Údb_urir% r° r¶ r rµ r9 r9 r: Útest_database_uri_stringo s
|
z#TestSQLApi.test_database_uri_stringÚpg8000c C s. d}t jtdd t d|¡ W 5 Q R X d S )Nz)postgresql+pg8000://user:pass@host/dbnamer< rî zselect * from table©r¼ r Ú ImportErrorr r )r\ r: r9 r9 r: Ú(test_pg8000_sqlalchemy_passthrough_error s z3TestSQLApi.test_pg8000_sqlalchemy_passthrough_errorc C sH ddl m} |d}tj|| jddid}t|d }|dhksDtd S )Nr rd z#select * from iris where name=:namerÓ zIris-versicolorr r4 )r6 re r r r< Úsetr )r\ re Z name_textÚiris_dfÚ all_namesr9 r9 r: Útest_query_by_text_obj s
|
z!TestSQLApi.test_query_by_text_objc C sh ddl m}m} t| j}|| |jj|dk¡}tj || j
|
ddid}t |d }|dhksdt d S )Nr )Ú bindparamr rÓ r r r4 ) r6 rD r r; r~ Úwherero r4 r r r< r@ r )r\ rD r r1 Z name_selectrA rB r9 r9 r: Útest_query_by_select_obj s
|
z#TestSQLApi.test_query_by_select_objc C sJ t dddgdddgd}|jd| jd d
|
t d| j¡}t ||¡ d S ) Nr r r r. r r
)r¨ z %_variationZtest_column_percentageFr2 )r rä r< r r" r r3 r r9 r9 r: Útest_column_with_percentage s z&TestSQLApi.test_column_with_percentageN) r` ra rb r§ r~ r r| r r r r r! r# r r1 r¼ r r rì r r7 rÓ r8 r9 r; ÚtdZskip_if_installedr? rC rF rG r9 r9 r9 r: r Ü sP
|
ñþ
|
|
|
r c @ sz e Zd ZdZdZdZdddZdd Zej j
|
e d d
|
d d Z ej j
|
e d d
|
d d Z dd Zdd Zdd Zdd ZdS )ÚTestSQLiteFallbackApiz9
|
Test the public sqlite connection fallback API
|
|
r# rÐ rÛ c C s
|
t |¡S rÉ rd )r\ Zdatabaser9 r9 r: r^ ± s z TestSQLiteFallbackApi.connectc
|
C s~ t ¡ `}t| |¡ }tj|d|dddks4tW 5 Q R X t| |¡}t d|¡}W 5 Q R X W 5 Q R X t ||¡ d S )NZtest_frame3_legacyFr2 r z!SELECT * FROM test_frame3_legacy;) r rÑ r r^ r rä r r! r3 )r\ r¶ rÓ r< rK r9 r9 r: Útest_sql_open_close´ s
|
ÿÿ z)TestSQLiteFallbackApi.test_sql_open_closezSQLAlchemy is installedr c C s2 d}d}t jt|d t d|¡ W 5 Q R X d S )Nz mysql://root@localhost/pandasz-Using URI string without sqlalchemy installedrî r
|
r= )r\ r< r r9 r9 r: Ú test_con_string_import_errorÄ s z2TestSQLiteFallbackApi.test_con_string_import_errorc
|
C sN G dd d}t |d¡(}t t¡ t d|¡ W 5 Q R X W 5 Q R X d S )Nc @ s* e Zd ZddddZdd Zdd Zd S )
|
z~TestSQLiteFallbackApi.test_con_unknown_dbapi2_class_does_not_error_without_sql_alchemy_installed.<locals>.MockSqliteConnectionÚNone©Úreturnc _ s t j||| _d S rÉ )rf rg r< )r\ ÚargsÚkwargsr9 r9 r: Ú__init__Ð s zTestSQLiteFallbackApi.test_con_unknown_dbapi2_class_does_not_error_without_sql_alchemy_installed.<locals>.MockSqliteConnection.__init__c S s t | j|S rÉ )Úgetattrr< )r\ rÓ r9 r9 r: Ú __getattr__Ó s zTestSQLiteFallbackApi.test_con_unknown_dbapi2_class_does_not_error_without_sql_alchemy_installed.<locals>.MockSqliteConnection.__getattr__c S s | j ¡ d S rÉ )r< rU re r9 r9 r: rU Ö s zTestSQLiteFallbackApi.test_con_unknown_dbapi2_class_does_not_error_without_sql_alchemy_installed.<locals>.MockSqliteConnection.closeN)r` ra rb rQ rS rU r9 r9 r9 r: ÚMockSqliteConnectionÏ s rT rÛ zSELECT 1)rÜ r r rô rõ r r )r\ rT r< r9 r9 r: ÚJtest_con_unknown_dbapi2_class_does_not_error_without_sql_alchemy_installedË s
|
z`TestSQLiteFallbackApi.test_con_unknown_dbapi2_class_does_not_error_without_sql_alchemy_installedc C sZ t d| j¡}t d| j¡}t ||¡ d}tjt j|d t d| j¡ W 5 Q R X d S )Nr
|
z9Execution failed on sql 'iris': near "iris": syntax errorrî r1 ) r r! r< r r r3 r¼ r Ú DatabaseError)r\ r r r r9 r9 r: r Ý s z,TestSQLiteFallbackApi.test_read_sql_delegatec C s t |d¡}d|kstd S )Nrß rð )r rò r ró r9 r9 r: Útest_get_schema2æ s z&TestSQLiteFallbackApi.test_get_schema2c C sJ | d¡D ]*}| ¡ d d¡|kr
|
| ¡ d S q
|
td| dd S )NÚ
|
r r9 r zColumn ú
|
not found)ÚsplitÚstripr )r\ rB ÚcolumnÚcolr9 r9 r: Ú_get_sqlite_column_typeë s z-TestSQLiteFallbackApi._get_sqlite_column_typec C sT t dtddgddi}t | j¡}tjd||d}| ¡ }| |d¡dksPtd S ) Nr r+ r, Tr- r/ r} r* ) r r r r r< Z SQLiteTableZ
|
sql_schemar^ r )r\ r¥ r r% rB r9 r9 r: Útest_sqlite_type_mappingñ s ÿ z.TestSQLiteFallbackApi.test_sqlite_type_mappingN)rÛ )r` ra rb r§ r~ r r^ rJ r¼ r ÚskipifÚSQLALCHEMY_INSTALLEDrK rU r rW r^ r_ r9 r9 r9 r: rI ¨ s
|
|
|
rI c @ sp e Zd ZU dZded< edd Zejdddd Z ed
|
d Z
|
ed d Z dd Z dd Z dd Zdd Zdd Zdd Zdd Zd d Zd d Zd d! Zd"d# Zd$d% Zd&d' Zd(d) Zd*d+ Zd,d- Zd.d/ Z d0d1 Z d2d3 Z d4d5 Z d6d7 Z d8d9 Z!d:d; Z"d<d= Z#d>d? Z$d@dA Z%dBdC Z&dDdE Z'dFdG Z(dHdI Z)dJdK Z*dLdM Z+dNdO Z,dPdQ Z-ej. /dRdSe0j1gidSe0j1 gie0j1 gdTgdUg¡dVdW Z2dXdY Z3dZd[ Z4d\d] Z5d^d_ Z6d`da Z7dbdc Z8ej. /dddedfg¡dgdh Z9ej. /dddedig¡djdk Z:ej. /dddedidfg¡dldm Z;dndodpdqZ<dndodrdsZ=dtdu Z>ej. /dve?j@dwg¡ej. /dddedfg¡dxdy ZAdzS ){Ú_TestSQLAlchemyz¶
|
Base class for testing the sqlalchemy backend.
|
|
Subclasses for specific database types are created below. Tests that
|
deviate for each flavor are overwritten there.
|
|
r) r~ c C s | ¡ | ¡ d S rÉ )Ú setup_driverÚ setup_enginers r9 r9 r: r s z_TestSQLAlchemy.setup_classTr© c C sd z | j ¡ | _t | j¡| _W n, tjjk
|
rJ t
|
d| j d¡ Y nX | |¡ | |¡ d S )NzCan't connect to z server)r r^ r< r r rÿ r6 rV rW r¼ Úskipr~ r r r« r9 r9 r: r¬ s
|
z _TestSQLAlchemy.setup_methodc C s
|
t d S rÉ ©ÚNotImplementedErrorrs r9 r9 r: rc s z _TestSQLAlchemy.setup_driverc C s
|
t d S rÉ rf rs r9 r9 r: rd s z _TestSQLAlchemy.setup_enginec C s | ¡ d S rÉ ©r
re r9 r9 r: Útest_read_sql_parameter# s z'_TestSQLAlchemy.test_read_sql_parameterc C s | ¡ d S rÉ ©r re r9 r9 r: Ú test_read_sql_named_parameter& s z-_TestSQLAlchemy.test_read_sql_named_parameterc C s | |¡ d S rÉ ©r r r9 r9 r: Útest_to_sql_empty) s z!_TestSQLAlchemy.test_to_sql_emptyc C sz ddl m} | ¡ }tddddgddddgd}tj|dd }| |d
|
¡d ksVtW 5 Q R X ||}| d
|
¡svtd S ) Nr rz ç ð?ç @ç @ç @©ZoneZtwoTrú rá r ) r6 rà r^ r r r rä r rÄ ©r\ rà Z temp_connrá rÿ rÆ r9 r9 r: Útest_create_table, s ÿ z!_TestSQLAlchemy.test_create_tablec C s¤ ddl m} | ¡ }tddddgddddgd}t |¡}| |d¡d ksPt||}| d¡sft| d¡ z |
|
¡ W n t k
|
r Y nX | d¡r td S )
|
Nr rz rn ro rp rq rr rá r ) r6 rà r^ r r r rä r rÄ r[ Ú clear_cacheÚAttributeErrorrs r9 r9 r: Útest_drop_table9 s ÿ
|
|
z _TestSQLAlchemy.test_drop_tablec C s | |¡ d S rÉ ©r r r9 r9 r: r¿ M s z _TestSQLAlchemy.test_roundtripc C s | ¡ d S rÉ ©r re r9 r9 r: r P s z _TestSQLAlchemy.test_execute_sqlc C s t jd| jd}t| d S )Nr1 r½ )r r" r< r r r9 r9 r: Útest_read_tableS s z _TestSQLAlchemy.test_read_tablec C s. t jd| jddgd}t |jjddg¡ d S )Nr1 r2 )r¾ r² )r r" r< r r r² r\ r r9 r9 r: r W s ÿz'_TestSQLAlchemy.test_read_table_columnsc C s2 d}t jt|d tjd| jd W 5 Q R X d S )Nz!Table this_doesnt_exist not foundrî Zthis_doesnt_existr½ )r¼ r r r r" r< )r\ r r9 r9 r: Ú test_read_table_absent_raises] s z-_TestSQLAlchemy.test_read_table_absent_raisesc C s~ t d| j¡}t|jjjtjs$t t|j
|
jjtj s:t t|j jjtj sPt t|jjjtjsft t|jjjtszt d S r+ )r r" r< r rr r7 r r r r rs r, rt Zbool_ru rv ÚobjectrË r9 r9 r: r- b s z,_TestSQLAlchemy.test_default_type_conversionc C sH t ddgid}|jd| jdddks*tt d| j¡}t ||¡ d S )NÚi64l )r Ú test_bigintFr2 r )r rä r< r r r" r r3 ré r9 r9 r: r~ n s z_TestSQLAlchemy.test_bigintc C s( t d| j¡}t|jjjtjs$t d S r+ ©
|
r r" r< r ro r7 r r rÉ r rË r9 r9 r: Útest_default_date_loadv s z&_TestSQLAlchemy.test_default_date_loadc C s@ dd }t d| j}t|ds4|j tjjdd¡ |j}t |j
|
sHt t d| jdgd}t|dsz|j tjjdd¡ |j}t |j
|
st t |j jdks¢t ||j ttt d| jd d
|
d d }|j}t |j
|
sÜt t |j jdksðt t d | j¡}|j}t |j
|
st t |j|j¡ t d | j¡}||j d S )Nc S s t | jr4| d tdks t| d tdkstnbt| jrt| jjdksRttdddtdddg}t|| j d}t
|
| |¡ ntd| j d S ) Nr z2000-01-01 08:00:00r z2000-06-01 07:00:00ÚUTC)ÚtzrÒ z&DateCol loaded with incorrect type -> ) r r7 r r r r) Údtr r rÓ r r× )r] Z expected_datarI r9 r9 r: r( s
|
|
|
|
þ
|
ÿz:_TestSQLAlchemy.test_datetime_with_timezone.<locals>.checkzselect * from typesrw z&no column with datetime with time zoner rÄ r r r Trÿ rm )r! r< rT ÚnodeÚ
|
add_markerr¼ r Úxfailrw r r7 r r) r r r r r r" r r× )r\ rå r( r¥ r] rI r9 r9 r: Útest_datetime_with_timezone} sD !
|
ÿ ÿ
|
ÿ
|
þz+_TestSQLAlchemy.test_datetime_with_timezonec C sÊ t dtddddi}|jd| jdddks0t| jd krP|d j d
|
¡|d< n|d j d ¡|d< t
|
d| j¡}t ||¡ t d | j¡}| jd krºt|jd tsªtt|d |d< t ||¡ d S )Nr¨ ú2013-01-01 09:00:00r. ú
|
US/Pacific©Úperiodsr Ztest_datetime_tzFr2 r% r z SELECT * FROM test_datetime_tzr# ©r r¨ )r r rä r< r r~ r Z
|
tz_convertZ tz_localizer r" r r3 r! r] Úlocr) r )r\ rI rK r9 r9 r: Ú%test_datetime_with_timezone_roundtripÌ s ÿ
|
|
z5_TestSQLAlchemy.test_datetime_with_timezone_roundtripc C sd t dtdddidgd}|jd| jdddks4tt d| j¡}t tjgdgd}t
|
||¡ d S ) Nr i' r r r2 Ztest_datetime_obbFr± ) r r rä r< r r r" rò ZNaTr r3 )r\ r rK rI r9 r9 r: Útest_out_of_bounds_datetimeç s
|
z+_TestSQLAlchemy.test_out_of_bounds_datetimec C sh t dddd d ¡}tdtdi|d}|jd| jdd dksBttjd| jdd
|
}t j
|
||d d d S ) Nz
|
2018-01-01r
Z6H)r ÚfreqÚnumsr2 Z foo_tableZ info_daterÞ rç F)Z check_names) r Z
|
_with_freqr rà rä r< r r r" r r3 )r\ ÚdatesrI rK r9 r9 r: Ú"test_naive_datetimeindex_roundtripï s
|
z2_TestSQLAlchemy.test_naive_datetimeindex_roundtripc C sJ t d| j¡}| jdkr tntj}t|jj j
|
|s6t t jd| jdgd}t|jj j
|
tjs`t t jd| jddid}t|jj j
|
tjst t jd| jdddiid}t|jj j
|
tjs¼t t jd| jdgd}t|j j j
|
tjsæt t jd| jddid}t|j j j
|
tjst t jd| jdd diid}t|j j j
|
tjsFt d S )
|
Nrm r# ro rÄ rÆ Úformatrp rº Úunit) r r" r< r~ r| r rÉ r ro r7 r r rp )r\ r¥ Ú expected_typer9 r9 r: rÌ ù s6 ÿ
|
ý
|
ÿz!_TestSQLAlchemy.test_date_parsingc C s t tdddt d¡d}| d| j¡dks2tt d| j¡}|j ddd }t
|
||¡ t d
|
| j¡}|j ddd }| j d kr²t|jd tstt|d |d < t
|
||¡ n t
|
||¡ d S )Nr r. ©r rp rê Ú test_datetimer§ r )ZaxisúSELECT * FROM test_datetimer# r r¨ )r r r r¸ rä r< r r r" r` r r3 r! r~ r] r r) r ré r9 r9 r: r s ÿ
|
z _TestSQLAlchemy.test_datetimec C sº t tdddt d¡d}tj|jd< |jd| jdd dksBtt
|
d| j¡}t ||¡ t d
|
| j¡}| jd krªt|jd tstt|d dd|d < t ||¡ n t ||¡ d S )Nr r. r rp rê ©r r¨ r Fr2 r r# r r¨ rÏ )rÑ )r r r r¸ r r rä r< r r r" r r3 r! r~ r] r) r ré r9 r9 r: Útest_datetime_NaT. s ÿ
|
z!_TestSQLAlchemy.test_datetime_NaTc C sl t tdddtdddgdgd}|jd| jdddks<ttd| j}|d }t|d }t ||¡ d S ) NéÞ r r rë r± Ú test_dateFr2 ) r r rä r< r r" r r r× )r\ r¥ r¥ rK rI r9 r9 r: Útest_datetime_dateB s " z"_TestSQLAlchemy.test_datetime_datec C s t tdddtdddgdgd}|jd| jdd d
|
ks<ttd| j}t ||¡ |}tj|d |dd d
|
ksrtt d |¡}|
|
d d ¡}t ||¡ tj|d| jdd d
|
ks´t| j dkræt d| j¡}|
|
dd ¡}t ||¡ t d| j¡}t ||¡ d S )Né r r é rë r± Ú test_timeFr2 r Z
|
test_time2zSELECT * FROM test_time2c S s
|
| d¡S ©Nz %H:%M:%S.%f©Ústrftime©Ú_r9 r9 r: Ú<lambda>X ó z4_TestSQLAlchemy.test_datetime_time.<locals>.<lambda>Z
|
test_time3r# zSELECT * FROM test_time3c S s
|
| d¡S r¢ r£ r¥ r9 r9 r: r§ ^ r¨ ) r r rä r< r r" r r3 r r! Úapplymapr~ )r\ rÝ r¥ r¥ rÖ Úrefr9 r9 r: Útest_datetime_timeL s "
|
z"_TestSQLAlchemy.test_datetime_timec C sh t dtjd}t dtjd}t||d}|jd| jdddksDtt d| j¡}t
|
j ||dd d
|
d S ) Ni rè g )Ús1r» Ztest_read_writeFr2 r T)Z check_dtypeZ check_exact) r r r4 Úfloat32r rä r< r r r" r r3 )r\ r¬ r» r¥ r r9 r9 r: Útest_mixed_dtype_insertc s z'_TestSQLAlchemy.test_mixed_dtype_insertc C sn t dddgdtjdgd}|jd| jdd d
|
ks6tt d| j¡}t ||¡ t
|
d | j¡}t ||¡ d S ) Nr r r r rì rê Útest_nanFr2 r. úSELECT * FROM test_nan) r r r rä r< r r r" r r3 r! ré r9 r9 r: Útest_nan_numerico s z _TestSQLAlchemy.test_nan_numericc C s t dddgtjtjtjgd}|jd| jdddks:tt d| j¡}t ||¡ |d
|
d
|
¡|d < d |d < t d | j¡}t ||¡ d S ) Nr r r rê r¯ Fr2 r. r© r| r° ) r r r rä r< r r r" r r3 r¤ r! ré r9 r9 r: Útest_nan_fullcolumn| s z#_TestSQLAlchemy.test_nan_fullcolumnc C sx t dddgddtjgd}|jd| jdd d
|
ks6td |jd < t d| j¡}t
|
||¡ t d | j¡}t
|
||¡ d S ) Nr r r rë r1 rê r¯ Fr2 r. ©r r© r° ) r r r rä r< r r r r" r r3 r! ré r9 r9 r: Útest_nan_string s
|
z _TestSQLAlchemy.test_nan_stringc C s2 ddl m} || j}| |¡}dd |D }|S )Nr rz c S s g | ] }|d qS r% r9 r& r9 r9 r: rW ¡ s z6_TestSQLAlchemy._get_index_columns.<locals>.<listcomp>)r6 rà r< r' )r\ r( rà rÆ r) r9 r9 r: r s
|
|
|
z"_TestSQLAlchemy._get_index_columnsc C s | ¡ d S rÉ ©r¡ re r9 r9 r: Útest_to_sql_save_index¤ s z&_TestSQLAlchemy.test_to_sql_save_indexc C s | ¡ d S rÉ ©r¦ re r9 r9 r: Útest_transactions§ s z!_TestSQLAlchemy.test_transactionsc
|
C sæ ddl m} ddlm} d}tj||| jd}|jd d
}| || j¡ ||}t | j|r| j
|
¡ $}| ¡ | |¡ W 5 Q R X W 5 Q R X n"| j ¡ | j |¡ W 5 Q R X t || j¡}tj||dd | || j¡ d S )Nr rd rM Ú test_get_schema_create_tabler½ Frè )r6 re r[ rN r rò r< r r[ r] r^ r_ rA r" r r3 ) r\ r¶ re rN r^ rô Z blank_test_dfr< Z returned_dfr9 r9 r: r¹ ª s
|
z,_TestSQLAlchemy.test_get_schema_create_tablec C s¨ ddl m}m} ddlm} ddg}ddg}t||d}| d | j¡d
|
ksNt|jd | jd|id d
|
kslt| }|j | jd |j
|
d j d j }t ||s td} tjt| d |jd| jdtid W 5 Q R X |jd| jd|did d
|
ksöt|j | jd |j
|
d j d j }t ||s&t|jdks6t|jd| j|d d
|
ksRt|j | jd |j
|
d j d j }
|
|j
|
d j d j } t |
|
|stt | |s¤td S )Nr )ri r/ ©r. r¨ r© ©gé?T©gÍÌÌÌÌÌì?Nr± Ú
|
dtype_testr Ú dtype_test2rè rY z&The type of B is not a SQLAlchemy typerî r£ Z dtype_test3é
|
Úsingle_dtype_test)r6 ri r/ Úsqlalchemy.schemar. r rä r< r ÚreflectÚtablesr² r r] r¼ r r r) r ) r\ ri r/ r. r r r¥ ÚmetaZsqltyper ZsqltypeaZsqltypebr9 r9 r: Ú
|
test_dtype s2 " z_TestSQLAlchemy.test_dtypec C s ddl m}m}m}m} ddlm} tdd gttdddd gtdd gdd td
|
d gd }t |}d }|
|
|| j ¡d kst t || j ¡} | }
|
|
|
j| j d | jdkr°|n|} |
|
j| j} t| d j| sÔt t| d j|sèt t| d j|süt t| d j|st d S )Nr )rj rk r- rl rº TéÜ r
r r| rè rù ©ÚBoolÚDateÚIntr- Únotna_dtype_testr rY r$ rÈ rÉ rÊ r- )r6 rj rk r- rl rÁ r. r r r rä r< r r r" r r~ rà r² r] r ) r\ rj rk r- rl r. r r¥ r^ r¦ rÄ Zmy_typeÚcol_dictr9 r9 r: Útest_notna_dtypeå s&
|
|
ü z _TestSQLAlchemy.test_notna_dtypec
|
C sn ddl m}m}m} ddlm} d}tt|gddt|gddt|gddtdgd dtdgd
|
dd }|jd | j d dd|ddiddkst
|
t d | j ¡}t |d jd d¡t |d jd d¡ksÐt
|
| }|j| j d |jd j} t| d jt| d jkst
|
t| d j|s(t
|
t| d j|s>t
|
t| d j|sTt
|
t| d j|sjt
|
d S )Nr )Ú
|
BigIntegerr- rl rº gñYÉBÊÀó?r rè Úfloat64r
r4 r¡ )Úf32Úf64Ú
|
f64_as_f32Úi32r} Z test_dtypesFrñ rÒ é )Z precision)r§ râ r7 r rÑ é rY rÐ rÓ r} )r6 rÎ r- rl rÁ r. r r rä r< r r r" r Úroundr r rà r² r) r r] )
|
r\ rÎ r- rl r. ÚVr¥ r¥ rÄ rÌ r9 r9 r: Útest_double_precision s> ûÿ ûùÿ
|
0 "z%_TestSQLAlchemy.test_double_precisionc sj ddl m dd dd fdd fd d
|
}td dd d gi d | j¡dks\t|| j d S )Nr rM c S s d}t j|| dS )Nz'SELECT test_foo_data FROM test_foo_datar½ )r r! )r= r r9 r9 r: Ú test_select4 s zC_TestSQLAlchemy.test_connectable_issue_example.<locals>.test_selectc S s |j d| dd d S )NÚ test_foo_datarà )rÓ r¾ râ )rä )r= r r9 r9 r: Ú test_append8 s zC_TestSQLAlchemy.test_connectable_issue_example.<locals>.test_appendc s | } | | d S rÉ r9 )r< Zfoo_data)rÛ rÙ r9 r: Útest_connectable; s zH_TestSQLAlchemy.test_connectable_issue_example.<locals>.test_connectablec
|
sH t | r<| ¡ "}| ¡ | W 5 Q R X W 5 Q R X n| d S rÉ )r] r^ r_ )Z connectabler< )rN rÜ r9 r: ÚmainA s
|
|
|
|
z<_TestSQLAlchemy.test_connectable_issue_example.<locals>.mainrÚ r r r. )r[ rN r rä r< r )r\ rÝ r9 )rN rÛ rÜ rÙ r: Ú test_connectable_issue_example/ s ÿÿz._TestSQLAlchemy.test_connectable_issue_exampleÚinputrÕ Úbar)rÕ Úinfe0c C s¼ t |}| jdkrt d¡}t|jtdk rRd|jkrRtjjdd}|j
|
|¡ d}tj t |d |j d | jd
|
d W 5 Q R X n4|j d | jd
|
d d kstt d | j¡}t ||¡ d S ) Nr$ r· z1.0.3rá zGH 36465r z inf cannot be used with MySQLrî ZfoobarFr2 r )r r~ r¼ r½ r Ú __version__r² r r r r
r r rä r< r r r" r Z assert_equal)r\ rß rå r¥ r· r r r¥ r9 r9 r: Ú test_to_sql_with_negative_npinfO s
|
|
ÿþ z/_TestSQLAlchemy.test_to_sql_with_negative_npinfc
|
sÒ ddl m mmm} ddlm}m} d}td|gi}| }G fddd|}|| j Z}|
|
¡ F | ¡ } |j | ¡ | ||d¡ | ¡ tj||j| d }
|
W 5 Q R X W 5 Q R X t |
|
|¡ d S )
|
Nr )r, rl ÚUnicoder )ÚSessionÚdeclarative_basez Hello, World!Úspamc s6 e Zd ZdZddgiZ ddZ dddZd S )
|
z7_TestSQLAlchemy.test_temporary_table.<locals>.TemporaryZ temp_testÚprefixesÚ TEMPORARYT©Z primary_keyr F)ZnullableN)r` ra rb Ú __tablename__Z__table_args__Úidrç r9 ©r, rl rä r9 r: Ú Temporary~ s
|
rî )rç )r r¾ )r6 r, rl rä r Úsqlalchemy.ormrå ræ r r< r_ r= Z __table__ra ÚaddÚflushr r! rç r r3 ) r\ r rå ræ Z test_datarI ZBaserî Úsessionr< r¥ r9 rí r: Útest_temporary_tablen s
|
(z$_TestSQLAlchemy.test_temporary_tablec C s. d}t jt|d | |d¡ W 5 Q R X d S )Nz*engine must be one of 'auto', 'sqlalchemy'rî Z
|
bad_engine)r¼ r r r r± r9 r9 r: Útest_invalid_engine s z#_TestSQLAlchemy.test_invalid_enginec C s& t dd¡ | |¡ W 5 Q R X d S )Nú io.sql.enginer6 ©rò Úoption_contextr r r9 r9 r: Útest_options_sqlalchemy s z'_TestSQLAlchemy.test_options_sqlalchemyc C s& t dd¡ | |¡ W 5 Q R X d S )Nrõ r rö r r9 r9 r: Útest_options_auto s z!_TestSQLAlchemy.test_options_autoc C s t tdtstt dd¡* t tdts2tt tdtsDtW 5 Q R X t dd¡* t tdtsntt tdtstW 5 Q R X d S )Nr6 rõ r )r] r r r rò r÷ re r9 r9 r: Útest_options_get_engine s z'_TestSQLAlchemy.test_options_get_enginec C s d S rÉ r9 re r9 r9 r: Ú"test_get_engine_auto_error_message¨ s z2_TestSQLAlchemy.test_get_engine_auto_error_messageÚfuncr r! c C sÈ d}| ¡ }|j|| jddd t d|¡" tt|d| | j|d}W 5 Q R X | ||¡}t ||¡ t d|¡F tt|d| | j|dd }| ||¡}|D ]}t ||¡ q¨W 5 Q R X d S )
|
Nrß Frñ ©r§ râ úmode.string_storageúSelect * from ©Ú dtype_backendr. ©r r © Údtype_backend_datarä r< rò r÷ rR Údtype_backend_expectedr r3 © r\ Ústring_storagerü r r% r¥ rK rI Úiteratorr9 r9 r: Útest_read_sql_dtype_backend® s* ÿ ü z+_TestSQLAlchemy.test_read_sql_dtype_backendr" c C s¼ d}| ¡ }|j|| jddd t d|¡ tt||| j|d}W 5 Q R X | ||¡}t ||¡ t d|¡@ tt||| j|dd}| ||¡}|D ]}t ||¡ qW 5 Q R X d S ) Nrß Frñ rý rþ r r. r r r r9 r9 r: Ú!test_read_sql_dtype_backend_tableÇ s" ü z1_TestSQLAlchemy.test_read_sql_dtype_backend_tablec C sX d}| ¡ }|j|| jddd d}tjt|d tt||| jdd W 5 Q R X d S ) Nrß Frñ rý zPdtype_backend numpy is invalid, only 'numpy_nullable' and 'pyarrow' are allowed.rî Únumpyr )r rä r< r¼ r r rR rò )r\ rü r% r¥ r r9 r9 r: Ú)test_read_sql_invalid_dtype_backend_tableÞ s ÿz9_TestSQLAlchemy.test_read_sql_invalid_dtype_backend_tabler rM c C sn t tdtjdgddtdddgddtdtjdgddtdd dgddd
|
d d gd
|
d d
|
gd d dgd d d gdS )Nr r. r6 rè r r ç @ÚFloat64ro TFrë r1 ro ©rë r1 ro ÚdÚeÚfÚgÚh)r r r r re r9 r9 r: r ë s øÿz"_TestSQLAlchemy.dtype_backend_datac
|
s& |dkr@t tjdddgtjd}t tjddtjgtjd}n2t d¡t dddg¡}t ddd g¡}t t
|
dtj dgd dt
|
dd
|
dgd dt
|
d tj d gd dt
|
d dd gd dt
|
ddtjgddt
|
dddgdd||d|dkr"t d¡ddl m t fddjD S )NÚpythonrë r1 ro rè rç r r. r6 r r r r ro TFÚbooleanr r )ÚArrowExtensionArrayc s$ i | ] }| j | d dqS )T)Z from_pandas)ró )rS r] ©r r¥ Úpar9 r: Ú
|
<dictcomp>
|
s ÿ z:_TestSQLAlchemy.dtype_backend_expected.<locals>.<dictcomp>)r r ró Zobject_rò ZNAr¼ r½ r r r r Z pandas.arraysr r² )r\ Ústorager Z string_arrayZstring_array_nar9 r r: r ù s4
|
øÿ
|
|
þÿz&_TestSQLAlchemy.dtype_backend_expectedc C sb ddd}t ddgd |¡}| ¡ }|jd| jdd d
|
td | j|d d D ]}t ||¡ qLd S )Nr¡ r| r0 rë r1 r± rß Frñ rý zSELECT * FROM testr )r7 r )r r¤ r
|
rä r< r! r r3 )r\ r r¥ rI rK r9 r9 r: Útest_chunksize_empty_dtypes
|
s
|
ü
|
z+_TestSQLAlchemy.test_chunksize_empty_dtypesr Únumpy_nullablec C s d}t dddgdd}|j|| jddd tt|d
|
| | jd tji|d }t tdddgtjd tdddg|dkszdndd d}t ||¡ d S )Nrß r r r. r
r0 Frñ rý rÿ rë )r7 r rè r r¡ r6 )
|
r rä r< rR rò r rÏ r r r3 )r\ rü r r% r¥ rK rI r9 r9 r: Útest_read_sql_dtype-
|
s$ üþþÿ z#_TestSQLAlchemy.test_read_sql_dtypeN)Br` ra rb r§ r r| r r¼ r r¬ rc rd ri rk rm rt rw r¿ r rz 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Þ r r r Úinfrã ró rô rø rù rú rû r r
|
r r r r r Z
|
no_defaultr r9 r9 r9 r: rb s
|
|
|
|
|
|
O
|
|
# - (þ
|
|
|
|
%rb c s| e Zd ZdZdZedd Zedd Zdd Zd d
|
Z d d Z
|
d d Z dd fdd Z e j dddg¡dd Z ZS )ÚTestSQLiteAlchemyzL
|
Test the sqlalchemy backend against an in-memory sqlite database.
|
|
r# c C s t d¡| _d S r r rs r9 r9 r: rd O
|
s z TestSQLiteAlchemy.setup_enginec C s
|
d | _ d S rÉ )Údriverrs r9 r9 r: rc S
|
s z TestSQLiteAlchemy.setup_driverc C s t d| j¡}t|jjjtjs$t t|j
|
jjtj s:t t|j jjtj sPt t|j jjtjsft t|jjjtjs|t d S r+ )r r" r< r rr r7 r r r r rs r, rt ru rv rË r9 r9 r: r- X
|
s z.TestSQLiteAlchemy.test_default_type_conversionc C s( t d| j¡}t|jjjtjr$t d S r+ r rË r9 r9 r: r g
|
s z(TestSQLiteAlchemy.test_default_date_loadc C sV t dddgidd}|jd| jdddks.tt d ¡ t d| j¡ W 5 Q R X d S ) Nrë r r r¡ rè Ztest_bigintwarningFr2 )r rä r< r r rô r r" rË r9 r9 r: Útest_bigint_warningm
|
s z%TestSQLiteAlchemy.test_bigint_warningc sð ddl m mm ddlm}m} | }G fddd|}| j ¡ |j
|
| j¡ W 5 Q R X || jd}| X}t ddgdd gd
|
}|j d | jd d ddks¬t | ¡ | |j|j¡}t |}W 5 Q R X t|jddgksìt d S )Nr ©r, rl r/ )ræ Ú sessionmakerc s( e Zd ZdZ ddZ dZdS )z>TestSQLiteAlchemy.test_row_object_is_named_tuple.<locals>.Testrû Trê é2 N)r` ra rb rë rì Ú string_columnr9 r# r9 r: ÚTest
|
s r' rY r ZhelloZworld)rì r& rû Frñ )r¾ r§ râ r rì r& )r6 r, rl r/ rï ræ r$ r< r_ r8 Z
|
create_allr rä r ri r rì r& r r² ) r\ ræ r$ Z BaseModelr' rå rò r¥ Z
|
test_queryr9 r# r: Ú test_row_object_is_named_tupleu
|
s" ÿÿz0TestSQLiteAlchemy.test_row_object_is_named_tupler rM c s< t ||¡}|dkr(| ddd¡}n| ddd¡}|S ©Nr r6 )r r zint64[pyarrow]©Úsuperr r¤ ©r\ r r r¥ ©Ú __class__r9 r: r
|
s
|
z(TestSQLiteAlchemy.dtype_backend_expectedrü r r" c C s d S rÉ r9 )r\ r rü r9 r9 r: r
|
£
|
s z3TestSQLiteAlchemy.test_read_sql_dtype_backend_table)r` ra rb r§ r~ r| rd rc r- r r" r( r r¼ r r r
|
Ú __classcell__r9 r9 r- r: r G
|
s
|
|
% r c sN e Zd ZdZdZdZedd Zedd Zdd Z d
|
d fd d Z
|
Z S )ÚTestMySQLAlchemyzA
|
Test the sqlalchemy backend against an MySQL database.
|
|
r$ iê c C s( t jd| j d| j d| jd| _d S )Nzmysql+z://root@localhost:ú/pandas)r¹ )r6 r r! Úportr¹ r rs r9 r9 r: rd ³
|
s þz TestMySQLAlchemy.setup_enginec C s$ t d¡}d| _d|jjji| _d S )Nr· r¸ )r¼ r½ r! r¾ r¿ rÀ r¹ )rt r· r9 r9 r: rc º
|
s
|
z TestMySQLAlchemy.setup_driverc C s d S rÉ r9 re r9 r9 r: r- À
|
s z-TestMySQLAlchemy.test_default_type_conversionr rM c s< t ||¡}|dkr(| ddd¡}n| ddd¡}|S r) r* r, r- r9 r: r Ã
|
s
|
z'TestMySQLAlchemy.dtype_backend_expected) r` ra rb r§ r~ r2 r| rd rc r- r r/ r9 r9 r- r: r0 ©
|
s
|
|
r0 c @ s8 e Zd ZdZdZdZedd Zedd Zdd Z d
|
S ) ÚTestPostgreSQLAlchemyzF
|
Test the sqlalchemy backend against an PostgreSQL database.
|
|
r% i8 c C s" t d| j d| j d¡| _d S )Nz postgresql+z ://postgres:postgres@localhost:r1 )r6 r r! r2 r rs r9 r9 r: rd ×
|
s ÿz"TestPostgreSQLAlchemy.setup_enginec C s t d¡ d| _d S )NrÌ )r¼ r½ r! rs r9 r9 r: rc Ý
|
s
|
z"TestPostgreSQLAlchemy.setup_driverc C s ddl m} tddgddgddgd }| j ¡ | j d
|
¡ | j d ¡ W 5 Q R X |jd | jd ddkspt|jd| jd dddkst|jd| jd dddks¨tt d | j¡}t
|
||¡ t d| j¡}t
|
||¡ tj d| jdd}t
|
||¡ tj d| jdd}t
|
||¡ d}t j t|d tj d| jdd W 5 Q R X | j ¡ | j d
|
¡ | j d ¡ W 5 Q R X |jd| jdd ddkst|jd| jdd dd |jd| jdd dddksÊttj d| jdd}t
|
t||gdd |¡ t| j|r| ¡ } tj| dd}
|
|
|
j|d d ddks2t|
|
j|d d dd dksNt|
|
j|d d dd dksjttj d | jdd}|
|
d ¡}t
|
||¡ d S ) Nr rM r r r/ r rë rE rF z$DROP SCHEMA IF EXISTS other CASCADE;zCREATE SCHEMA other;Ztest_schema_publicFr2 Ztest_schema_public_explicitÚpublic)r§ rB Ztest_schema_otherÚother)rB z!Table test_schema_other not foundrî )rB r§ rñ )rB r§ râ rà Trÿ Ztest_schema_other2rý )r[ rN r r< r_ r rä r r r" r r3 r¼ r r r r] r^ r Z
|
read_table) r\ rN r¥ r5 r6 r Zres4r r¥ Zengine2Zpdsqlr9 r9 r: Útest_schema_supportâ
|
s ÿýÿÿ ÿ ÿûûùÿ
|
ÿÿÿÿ
|
z)TestPostgreSQLAlchemy.test_schema_supportN)
|
r` ra rb r§ r~ r2 r| rd rc r6 r9 r9 r9 r: r3 Í
|
s
|
|
r3 c @ s² e Zd ZdZdZejdddd Zdd Zd d
|
Z d d Z
|
d d Z dd Z dd Z dd Zej dddg¡dd Zdd Zdd Zd d Zd d Zd!d" Zd#d$ Zd%d& Zd'S )(ÚTestSQLiteFallbackzG
|
Test the fallback mode against an in-memory sqlite database.
|
|
r# Tr© c C s0 | ¡ | _| |¡ | |¡ t | j¡| _d S rÉ )r^ r< r r r r rÿ r« r9 r9 r: r¬ I s
|
|
|
z TestSQLiteFallback.setup_methodc C s | ¡ d S rÉ rh re r9 r9 r: ri P s z*TestSQLiteFallback.test_read_sql_parameterc C s | ¡ d S rÉ rj re r9 r9 r: rk S s z0TestSQLiteFallback.test_read_sql_named_parameterc C s | |¡ d S rÉ rl r r9 r9 r: rm V s z$TestSQLiteFallback.test_to_sql_emptyc C sd t ddddgddddgd}| j |d¡dks4t| j d¡sDt| j d¡ | j d¡r`td S )Nrn ro rp rq rr Zdrop_test_framer )r rÿ rä r rÄ r[ )r\ rá r9 r9 r: Útest_create_and_drop_tableY s ÿ z-TestSQLiteFallback.test_create_and_drop_tablec C s | |¡ d S rÉ rx r r9 r9 r: r¿ f s z!TestSQLiteFallback.test_roundtripc C s | ¡ d S rÉ ry re r9 r9 r: r i s z#TestSQLiteFallback.test_execute_sqlc C s t tdddtdddgdgd}|jd| jdddks<ttd | j}| jd
|
krft || t
|
¡¡ n| jd kr|t ||¡ d S ) Nr r r rë r± r Fr2 zSELECT * FROM test_dater# r$ ) r r rä r< r r! r~ r r3 r¤ r) r r9 r9 r: r l s "
|
|
z%TestSQLiteFallback.test_datetime_dateÚtz_awareFc C s |s t dddt dddg}n$tdddd}t| ¡ d d
|
¡}t|d gd }|jd | jdddksjtt d| j}| j
|
dkr| dd
|
¡}t ||¡ d S )Nr r r r r r r r c S s | ¡ S rÉ )Útimetz)r r9 r9 r: r§ ~ r¨ z7TestSQLiteFallback.test_datetime_time.<locals>.<lambda>rë r± r¡ Fr2 zSELECT * FROM test_timer# c S s
|
| d¡S r¢ r£ r¥ r9 r9 r: r§ r¨ )r r r Z to_pydatetimeÚmapr rä r< r r! r~ r© r r3 )r\ r9 Ztz_timesZtz_dtr¥ r¥ rI r9 r9 r: r« w s
|
z%TestSQLiteFallback.test_datetime_timec C sT t dd| d | j¡}g }|jD ]*}t d| d| j¡}| |j ¡ ¡ q$|S )Nz1SELECT * FROM sqlite_master WHERE type = 'index' zAND tbl_name = 'ú'zPRAGMA index_info(ú))r r! r< rÓ rà rÊ )r\ r( r) r Zix_nameZix_infor9 r9 r: r s
|
ÿý
|
z%TestSQLiteFallback._get_index_columnsc C s | ¡ d S rÉ rµ re r9 r9 r: r¶ s z)TestSQLiteFallback.test_to_sql_save_indexc C s | ¡ d S rÉ r· re r9 r9 r: r¸ s z$TestSQLiteFallback.test_transactionsc
|
C sT | j d| d¡}|D ] \}}}}}} ||kr| S qtd| d| dd S )NzPRAGMA table_info(r= zTable z , column rY )r< rA r )
|
r\ r% r\ ZrecsZcidrÓ ÚctypeZnot_nullÚdefaultÚpkr9 r9 r: r^ s
|
|
z*TestSQLiteFallback._get_sqlite_column_typec C s | j dkrt d¡ ddg}ddg}t||d}| d| j¡d ksFt|jd
|
| jdd id d ksdt| dd¡d ksxt| d
|
d¡d kstd}tjt |d |jd| jdt
|
id W 5 Q R X |jd| jd d d ksÚt| dd¡d ksît| dd¡d kstd S )Nr$ ú Not applicable to MySQL legacyr¨ r© r» r¼ r± r½ r r¾ ÚSTRINGrè rø z!B \(<class 'bool'>\) not a stringrî r£ rÀ ) r~ r¼ re r rä r< r r^ r r Úbool)r\ r r r¥ r r9 r9 r: rÅ ¢ s
|
|
z TestSQLiteFallback.test_dtypec C sÄ | j dkrt d¡ tdd gttdddd gtdd gddtd d gd
|
}t|}d }| || j¡d kspt| |d ¡dkst| |d¡dkst| |d¡dks¬t| |d¡dksÀtd S )Nr$ rA TrÆ r
r r| rè rù rÇ rË r rÈ rø rÉ r* rÊ r- r+ )
|
r~ r¼ re r r r rä r< r r^ )r\ r r¥ r^ r9 r9 r: rÍ ¸ s
|
|
|
|
üz#TestSQLiteFallback.test_notna_dtypec C sâ t ddgddggddgd}d}tjt|d | d
|
| j¡ W 5 Q R X td d d dddddddg
|
D ]x\}}| || j¡dkstt || j¡ t ddgddggd|gd}d|d}| || j¡dksÎtt || j¡ qdd S )Nr r r. r rë r1 r± z$Empty table or column name specifiedrî r" ztest_weird_name]ztest_weird_name[ztest_weird_name`ztest_weird_name"ztest_weird_name'z_b.test_weird_name_01-30z"_b.test_weird_name_01-30"Z99beginswithnumberZ12345r Ztest_weird_col_namer )
|
r r¼ r r rä r< Ú enumerater r Z table_exists)r\ r¥ r ZndxZ
|
weird_namer Zc_tblr9 r9 r: Útest_illegal_namesÌ s. öÿ z%TestSQLiteFallback.test_illegal_namesN)r` ra rb r§ r~ r¼ r r¬ ri rk rm r8 r¿ r r r r r« r r¶ r¸ r^ rÅ rÍ rE r9 r9 r9 r: r7 A s(
|
|
|
r7 z'{}'z{:.8f}z{:d}c C s dS )NÚNULLr9 )Úxr9 r9 r: r§ ö r¨ r§ z{:.10f}z'{!s}'c G sJ g }|D ]4}t |tr"t|r"d }tt| }| ||¡ q| t| S rÉ )r] r¢ r Ú _formattersr rà r )r rO Zprocessed_argsÚargÚ formatterr9 r9 r: Ú format_queryü s rK c C s8 t |¡}| | ¡ ¡ }W 5 Q R X |dkr0dS t|S )z#Replace removed sql.tquery functionN)r r rA rm r )r r¾ rÁ r¥ r9 r9 r: Útquery s rL c @ s\ e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z dS )Ú TestXSQLitec C s* | ¡ }| dt |¡ ¡ | ¡ d S rf )r@ rA r rh ri )r\ r r< rG r9 r9 r: r[ s zTestXSQLite.drop_tablec C sØ t ¡ }tj|d|dddks"tt d|¡}|j|_|}t ||¡ dgt| |d< | ¡ }t
|
t t|¡t j dd }| ¡ |d
|
< tj|d |dddks¢ttjd |d
|
d }| ¡ }||_d
|
|j_t ||¡ d S )NZ
|
test_tableF)rÓ r¾ r§ r zselect * from test_tablerë Útxtrè r¿ ZIdxZ test_table2zselect * from test_table2rç )r ÚmakeTimeDataFramer rä r r r§ r3 rþ r
|
r r r¸ r¡ rÓ )r\ rÝ r~ rK rI Zframe2Znew_idxr9 r9 r: Ú
|
test_basic s0 ÿ ÿÿ ÿzTestXSQLite.test_basicc
|
C s t ¡ }tj|jd< t |d¡}| ¡ }| |¡ d}| ¡ D ]"\}}t
|
|f| }t ||d q>| ¡ tj d|d} |j| _t j| |dd d S )N)r r rß z(INSERT INTO test VALUES (%s, %s, %s, %s)r½ úselect * from testgü©ñÒMbP?)Zrtol)r rO r r r r rò r@ rA ZiterrowsrK rL ri r r§ r3 )
|
r\ rÝ r~ rô rG Úinsr¦ rT Zfmt_sqlrK r9 r9 r: Útest_write_row_by_row2 s
|
z!TestXSQLite.test_write_row_by_rowc C s t ¡ }t |d¡}| ¡ }| |¡ d}|jd }t |¡}| |t|¡ W 5 Q R X | ¡ t
|
d|¡}|j d d
|_ t ||d d
¡ d S )Nrß z$INSERT INTO test VALUES (?, ?, ?, ?)r rQ r ) r rO r rò r@ rA r r r ri r r§ r3 ) r\ rÝ r~ rô rG rR rT rÁ rK r9 r9 r: Ú test_executeD s
|
|
zTestXSQLite.test_executec C s t ¡ }t |d¡}| ¡ }|D ]6}| d¡}t|dkr |d dkr |d dks tq tj|dddgd }| ¡ }d
|
|kst| ¡ }| |¡ d S ) Nrß ú r r r¨ r ZDATETIMEr© r$ zPRIMARY KEY ("A", "B"))
|
r rO r rò Ú
|
splitlinesrZ rþ r r@ rA )r\ rÝ r~ rô ÚlinesÚlineÚtokensrG r9 r9 r: Ú test_schemaT s
|
zTestXSQLite.test_schemac
|
C sj d}| ¡ }| |¡ t |¡@}| d¡ | d¡ tjtjdd | d¡ W 5 Q R X W 5 Q R X d S )Nú
|
CREATE TABLE test
|
(
|
a TEXT,
|
b TEXT,
|
c REAL,
|
PRIMARY KEY (a, b)
|
);
|
ú,INSERT INTO test VALUES("foo", "bar", 1.234)z,INSERT INTO test VALUES("foo", "baz", 2.567)zExecution failed on sqlrî z(INSERT INTO test VALUES("foo", "bar", 7))r@ rA r r r¼ r rV )r\ rÝ rô rG rÁ r9 r9 r: Útest_execute_failc s
|
|
|
z TestXSQLite.test_execute_failc
|
C s d}t t d¡¡8}| ¡ }| |¡ t |¡}| d¡ W 5 Q R X W 5 Q R X d}tj tj
|
|d t d|d W 5 Q R X d S )Nr[ rÛ r\ z$Cannot operate on a closed database.rî rQ r½ ) rÜ r rf r^ r@ rA r r r¼ r ZProgrammingErrorrL )r\ rô r< rG rÁ r r9 r9 r: Ú test_execute_closed_connectionw s
|
z*TestXSQLite.test_execute_closed_connectionc C s0 t dt d¡i}tj||ddddks,td S )NÚFromr
Z testkeywordsF©r¾ rÓ r§ )r r Zonesr rä r )r\ rÝ r¥ r9 r9 r: Ú test_keyword_as_column_names s z(TestXSQLite.test_keyword_as_column_namesc C sp t ddgdgd}tj||ddddks,t|}tdd | d
|
¡D }|d ksTtt d
|
|¡}t ||¡ d S ) Nr r Zc0r± Úmono_dfFr` c s s | ]}|d V qdS )r Nr9 )rS Zmy_c0r9 r9 r: Ú <genexpr> s z8TestXSQLite.test_onecolumn_of_integer.<locals>.<genexpr>zselect * from mono_dfr. ) r r rä r ÚsumrA r r r3 )r\ rÝ rb Zcon_xZthe_sumrK r9 r9 r: Útest_onecolumn_of_integer s z%TestXSQLite.test_onecolumn_of_integerc C s´ t ddgddgd}t dddgd d
|
d gd}d }d | }d}tjt|d tj|||dd W 5 Q R X | ||¡ tj|||dd d}tjt|d tj|||dd W 5 Q R X tj|||ddd t||dddgksèttj|||ddddkstt||dddd gks"t| ||¡ tj|||ddddksLtt||dddgksfttj|||d dddkstt||dddddd gks¤t| ||¡ d S ) Nr r r¨ r© )rG rH r. r r
rª r« ÚEZtable_if_existszSELECT * FROM z*'notvalidvalue' is not valid for if_existsrî Z notvalidvalue)r~ r¾ rÓ râ r z&Table 'table_if_exists' already existsrñ F)r~ r¾ rÓ râ r§ r½ r r³ )r. rª )r r« )r
rf rà ) r r¼ r r r rä r[ rL r )r\ rÝ Zdf_if_exists_1Zdf_if_exists_2r Z
|
sql_selectr r9 r9 r: Útest_if_exists s
|
ü ÿüûûùÿ
|
ûùÿ
|
ûùÿ
|
|
û zTestXSQLite.test_if_existsN) r` ra rb r[ rP rS rT rZ r] r^ ra re rg r9 r9 r9 r: rM s rM )N)r§ Ú
|
__future__r rÜ r rC r r r r r r Úpathlibr
|
rf r r r¼ Z pandas._libsr Z pandas.util._test_decoratorsÚutilZ_test_decoratorsrH Zpandas.core.dtypes.commonr r Zpandasrò r r r r r r r r r r Zpandas._testingZ_testingr Zpandas.core.arraysr r Zpandas.util.versionr Z pandas.ior Z pandas.io.sqlr r r r r r! r" r6 ra r> r r; rJ rc rh ry r{ r| r r r r rz r¦ rµ r¶ rÈ rË rÎ rÏ rÔ rÕ rÖ r× rØ rÙ rÝ rÞ Zmysql_connectableZpostgresql_connectableZsqlite_connectableZsqlite_iris_connectableZsqlalchemy_connectableZsqlalchemy_connectable_irisZall_connectableZall_connectable_irisr r r ræ r÷ rø r r r r r r r r" r# r* r- r7 rL rQ rR rS rc rr r} r¨ r` r rI rb r r0 r3 r7 r r) Zstr_Úbytesr¢ rì r rÏ rC rH rK rL rM r9 r9 r9 r: Ú<module> s 0 $
|
|
ýøýí!
|
|
|
|
#
|
|
|
|
|
|
|
|
|
|
|
|
|
þþýý
|
ÿ
|
|
$ %
|
|
" G LX Lb#s 0 ÷
|
|
