x\c @sdZddlmZddlZddlmZmZmZddlZddlZddl Z ddl Z ddl j Z ddl mZmZmZmZddlmZmZddlZddlmZmZmZmZmZmZmZmZmZm Z ddl!j"j#Z$ddl%j&j'Z'ddl%m(Z(m)Z)yTddl*Z*ddl+Z*ddl,j'j-Z-dd l.m/Z/dd l0m1Z2e3Z4Wne5k re6Z4nXiid d 6d d6dd6d6idd 6dd6dd6d6idd 6dd6dd6d6iidd6ddddd d!d"d#d$f d%6d 6id&d6ddddd d!d"d#d$f d%6d6id'd6ddd(ddd d!d"d#d$f d%6d6d)6id*d 6d+d6d,d6d-6id.d 6d/d6d0d6d16id2d 6d36Z7d4e8fd5YZ9d6e9fd7YZ:d8e9fd9YZ;d:e9fd;YZ<d<e8fd=YZ=d>e=fd?YZ>e j?j@d@e<e>fdAYZAdBe8fdCYZBe j?j@dDeBeAfdEYZCe j?j@dFe;e>fdGYZDdHe<e=fdIYZEdJeBeEfdKYZFdLe8fdMYZGdNe8fdOYZHdPe8fdQYZIe j?j@e j?jJdReHeEfdSYZKe j?j@e j?jJdTeHeFfdUYZLe j?j@e j?jJdVeIeEfdWYZMe j?j@e j?jJdXeIeFfdYYZNe j?j@dZeGeEfd[YZOe j?j@d\eGeFfd]YZPe j?j@d^e;e=fd_YZQd`ZRi dae6dbeS6dce jT6dde jU6dee jV6dfeW6dgeX6dheYeZ6die j[6dje\6Z]dkZ^eZeZdlZ_e j?j@dme;fdnYZ`e j?j@e j?jJe j?jadodpdqe:fdrYZbdS(ssSQL 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, `_TestSQLAlchemyConn` tests the API with a SQLAlchemy Connection object. The different tested flavors (sqlite3, MySQL, PostgreSQL) derive from the base class - Tests for the fallback mode (`TestSQLiteFallback`) i(tprint_functionN(tdatetdatetimettime(tPY36tlrangetranget string_types(tis_datetime64_dtypetis_datetime64tz_dtype( t DataFrametIndext MultiIndextSeriest Timestamptconcatt date_rangetisnat to_datetimet to_timedelta(tread_sql_querytread_sql_table(t declarative(tsessionsCREATE TABLE iris ( "SepalLength" REAL, "SepalWidth" REAL, "PetalLength" REAL, "PetalWidth" REAL, "Name" TEXT )tsqlitesCREATE TABLE iris ( `SepalLength` DOUBLE, `SepalWidth` DOUBLE, `PetalLength` DOUBLE, `PetalWidth` DOUBLE, `Name` VARCHAR(200) )tmysqlsCREATE TABLE iris ( "SepalLength" DOUBLE PRECISION, "SepalWidth" DOUBLE PRECISION, "PetalLength" DOUBLE PRECISION, "PetalWidth" DOUBLE PRECISION, "Name" VARCHAR(200) )t postgresqlt create_iriss&INSERT INTO iris VALUES(?, ?, ?, ?, ?)s.INSERT INTO iris VALUES(%s, %s, %s, %s, "%s");s,INSERT INTO iris VALUES(%s, %s, %s, %s, %s);t insert_irissCREATE TABLE types_test_data ( "TextCol" TEXT, "DateCol" TEXT, "IntDateCol" INTEGER, "IntDateOnlyCol" INTEGER, "FloatCol" REAL, "IntCol" INTEGER, "BoolCol" INTEGER, "IntColWithNull" INTEGER, "BoolColWithNull" INTEGER )sCREATE TABLE types_test_data ( `TextCol` TEXT, `DateCol` DATETIME, `IntDateCol` INTEGER, `IntDateOnlyCol` INTEGER, `FloatCol` DOUBLE, `IntCol` INTEGER, `BoolCol` BOOLEAN, `IntColWithNull` INTEGER, `BoolColWithNull` BOOLEAN )sCREATE TABLE types_test_data ( "TextCol" TEXT, "DateCol" TIMESTAMP, "DateColWithTz" TIMESTAMP WITH TIME ZONE, "IntDateCol" INTEGER, "IntDateOnlyCol" INTEGER, "FloatCol" DOUBLE PRECISION, "IntCol" INTEGER, "BoolCol" BOOLEAN, "IntColWithNull" INTEGER, "BoolColWithNull" BOOLEAN )tcreate_test_typesso INSERT INTO types_test_data VALUES(?, ?, ?, ?, ?, ?, ?, ?, ?) tquerytTextColtDateColt IntDateColtIntDateOnlyColtFloatColtIntColtBoolColtIntColWithNulltBoolColWithNulltfieldssz INSERT INTO types_test_data VALUES("%s", %s, %s, %s, %s, %s, %s, %s, %s) s| INSERT INTO types_test_data VALUES(%s, %s, %s, %s, %s, %s, %s, %s, %s, %s) t DateColWithTztinsert_test_typess1SELECT * FROM iris WHERE Name=? AND SepalLength=?s9SELECT * FROM iris WHERE `Name`="%s" AND `SepalLength`=%ss7SELECT * FROM iris WHERE "Name"=%s AND "SepalLength"=%stread_parameterss] SELECT * FROM iris WHERE Name=:name AND SepalLength=:length sy SELECT * FROM iris WHERE `Name`="%(name)s" AND `SepalLength`=%(length)s sw SELECT * FROM iris WHERE "Name"=%(name)s AND "SepalLength"=%(length)s tread_named_parameterss] CREATE VIEW iris_view AS SELECT * FROM iris t create_viewt MixInBasecBseZdZRS(cCsDt|dr@x!|jD]}|j|qW|jndS(Ntconn(thasattrt_get_all_tablest drop_tablet _close_conn(tselftmethodttbl((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pytteardown_methods(t__name__t __module__R7(((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyR.st MySQLMixIncBs#eZdZdZdZRS(cCs:|jj}|jdtj||jjdS(NsDROP TABLE IF EXISTS %s(R/tcursortexecutetsqlt_get_valid_mysql_nametcommit(R4t table_nametcur((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyR2s cCs=|jj}|jdg|jD]}|d^q)S(Ns SHOW TABLESi(R/R;R<tfetchall(R4RAttable((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyR1s cCs9ddlm}y|jjWn|k r4nXdS(Ni(tError(t pymysql.errRDR/tclose(R4RD((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyR3s  (R8R9R2R1R3(((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyR:s  t SQLiteMixIncBs#eZdZdZdZRS(cCs.|jjdtj||jjdS(NsDROP TABLE IF EXISTS %s(R/R<R=t_get_valid_sqlite_nameR?(R4R@((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyR2s cCs3|jjd}g|jD]}|d^qS(Ns1SELECT name FROM sqlite_master WHERE type='table'i(R/R<RB(R4tcRC((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyR1s  cCs|jjdS(N(R/RF(R4((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyR3s(R8R9R2R1R3(((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRGs  tSQLAlchemyMixIncBs#eZdZdZdZRS(cCstj|jj|dS(N(R=t SQLDatabaseR/R2(R4R@((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyR2scCs5tjjd|j}|j|jj}|S(Ntbind(t sqlalchemytschematMetaDataR/treflectttablestkeys(R4tmetat table_list((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyR1s cCsdS(N((R4((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyR3s(R8R9R2R1R3(((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRJs  t PandasSQLTestcBseZdZdZejddgdZdZdZd Z d Z d Z d Z d Z dZdZdZddZdZdZdZdZdZdZdZdZdZRS(sT Base class with common private methods for SQLAlchemy and fallback cases. cCs*t|jdr|jS|jjSdS(NR<(R0R/R;(R4((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyt _get_execstparamstiotdatasiris.csvc Csddl}||j}t|ds7|jn|jd|jjtd|j|j |ddddX}t j |}t |td|j}x$|D]}|jj||qWWdQXdS( NiR/tirisRtmodetrtnewlineR(RXtparamR0t setup_connectR2RVR<t SQL_STRINGStflavortopentNonetcsvtreadertnext( R4tdatapathtrequestRXt iris_csv_filetiris_csvR\tinstrow((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pytload_iris_datas     cCs/|jd|jjtd|jdS(Nt iris_viewR-(R2RVR<R`Ra(R4((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyt_load_iris_views cCs[|jdj}|jd}t|tjs5ttj|j dddddgdS(Nigffffff@g @gffffff?g?s Iris-setosa( tdtypesttypetiloct issubclasstnptfloatingtAssertionErrorttmt equalContentstvalues(R4t iris_frametpytypeRl((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyt_check_iris_loaded_frames cCs@dddddg}ddddg}t|d||_dS(NtindextAtBtCtDs2000-01-03 00:00:00gN\^?g9\x`| @g%TO׿g(nIs2000-01-04 00:00:00g}C?gc^Vg~q"BĿg4?s2000-01-05 00:00:00gGͿ?g2r¹e?g>ݦ4gёR?s2000-01-06 00:00:00g?{mX?g$?gnk3Am?g S?tcolumns(s2000-01-03 00:00:00gN\^?g9\x`| @g%TO׿g(nI(s2000-01-04 00:00:00g}C?gc^Vg~q"BĿg4?(s2000-01-05 00:00:00gGͿ?g2r¹e?g>ݦ4gёR?(s2000-01-06 00:00:00g?{mX?g$?gnk3Am?g S?(R t test_frame1(R4RRY((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyt_load_test1_datas cCsttdddddgdddd d gd d d ddgdttttgdddddg}t|d|d<||_dS(NR~iiiiRtasdtgsqtylttjklRg?g@g@g333333@RtEs 1990-11-22s 1991-10-26s 1993-11-26s 1995-12-12(R tdicttFalsetTrueRt test_frame2(R4tdf((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyt_load_test2_data,scCsCdddg}dddfdddg}t|d||_dS(NR}R~Rs2000-01-03 00:00:00iiigZs2000-01-04 00:00:00igc^Vs2000-01-05 00:00:00i Ng2r¹e?s2000-01-06 00:00:00i͏g$?RIi(s2000-01-04 00:00:00igc^V(s2000-01-05 00:00:00i Ng2r¹e?(s2000-01-06 00:00:00i͏g$?(R t test_frame3(R4RRY((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyt_load_test3_data7s   cCs|jd|jjtd|jtd|j}i dd6dd6dd 6d d 6d d 6dd6dd6td6dd6td6i dd6dd6dd 6dd 6dd 6dd6dd6td6dd6dd6g}xC|D];}|jj|dg|dD]}||^qqWdS(Nttypes_test_dataRR*tfirstRs2000-01-03 00:00:00R s2000-01-01 00:00:00-08:00R)iwR!i2R"g333333$@R#iR$R%R&R's2000-01-04 00:00:00s2000-06-01 00:00:00-07:00i'Pi\2RR((R2RVR<R`RaRRc(R4RkRYtdtfield((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyt _load_raw_sqlAs:    cCs'|jjd|j}|dS(Ns"SELECT count(*) AS count_1 FROM %si(RVR<tfetchone(R4R@tresult((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyt _count_rowsfs cCs#|jjd}|j|dS(NsSELECT * FROM iris(t pandasSQLt read_queryR|(R4Rz((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyt_read_sql_iriskscCsFtd|j}ddg}|jj|d|}|j|dS(NR+s Iris-setosagffffff@RW(R`RaRRR|(R4RRWRz((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyt_read_sql_iris_parameteros cCsNtd|j}idd6dd6}|jj|d|}|j|dS(NR,s Iris-setosatnamegffffff@tlengthRW(R`RaRRR|(R4RRWRz((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyt_read_sql_iris_named_parameteruscCs|jd|jj|jdd||jjdsAtt|j}|jd}||ksqt|jddS(NRR5(R2Rtto_sqlRt has_tableRvtlenR(R4R5t num_entriestnum_rows((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyt_to_sql{s cCs.|jd|jj|jjd ddS(NRi(R2RRRRr(R4((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyt _to_sql_emptys cCsw|jd|jj|jddd|jjdsAttjt|jj|jddd|jddS(NRt if_existstfail( R2RRRRRvtpytesttraisest ValueError(R4((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyt _to_sql_fails  cCs|jd|jj|jddd|jj|jddd|jjds]tt|j}|jd}||kst|jddS(NRRRtreplace(R2RRRRRvRR(R4RR((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyt_to_sql_replaces   cCs|jd|jj|jddd|jj|jddd|jjds]tdt|j}|jd}||kst|jddS(NRRRtappendi(R2RRRRRvRR(R4RR((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyt_to_sql_appends   csgfd}|jd|jj|jdd||jjdsVtdgksktt|j}|jd}||kst|jddS(NcsRjdg|D]}tt||^q}|j|jj|dS(Ni(RRtzipR<RCtinsert(tpd_tableR/RRt data_iterRlRY(tcheck(s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pytsamples (RR5i(R2RRRRRvRR(R4RRR((Rs7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyt_to_sql_method_callables cCsk|jd|jj|jd|jjd}|jddtd|j_ t j ||jdS(Nttest_frame_roundtrips"SELECT * FROM test_frame_roundtriptlevel_0tinplace( R2RRRRt set_indexRRcR}RRwtassert_frame_equal(R4R((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyt _roundtrips    cCsA|jjd}|j}tj|dddddgdS(NsSELECT * FROM irisgffffff@g @gffffff?g?s Iris-setosa(RR<RRwRx(R4t iris_resultsRl((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyt _execute_sqls cCsktjd dgddd d gd dg}|jj|d |jd }|dggksgtdS(Nig@tline1ig?tline2RR~RRR}ttest_to_sql_saves_index(ig@R(ig?R(R t from_recordsRRt_get_index_columnsRv(R4Rtix_cols((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyt_to_sql_save_indexs cCs|jjdd}y5|jj }|j|tdWdQXWntk r^nX|jjd}t|dkst|jj}|j|WdQX|jjd}t|dkstdS(Ns'CREATE TABLE test_trans (A INT, B TEXT)s/INSERT INTO test_trans (A,B) VALUES (1, 'blah')terrorsSELECT * FROM test_transii(RR<trun_transactiont ExceptionRRRv(R4tins_sqlttranstrestres2((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyt_transaction_tests  (RXRYsiris.csvN(R8R9t__doc__RVRtfixtureRmRoR|RRRRRRRRRcRRRRRRRRRR(((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRUs.     %          t _TestSQLApicBsLeZdZdZd"ZdZejde dZ dZ dZ dZ dZd Zd Zd Zd Zd ZejjddZdZdZdZdZdZdZdZdZdZdZ dZ!dZ"dZ#dZ$dZ%dZ&d Z'd!Z(RS(#s 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 RcCs|j|_dS(N(tconnectR/(R4((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyR_stautousecCs|jdS(N(tload_test_data_and_sql(R4Rm((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyt setup_methodscCs6|j|j|j|j|jdS(N(RoRRRR(R4((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRs     cCs&tjd|j}|j|dS(NsSELECT * FROM iris(R=RR/R|(R4Rz((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyttest_read_sql_iris#scCs&tjd|j}|j|dS(NsSELECT * FROM iris_view(R=RR/R|(R4Rz((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyttest_read_sql_view(scCs8tj|jd|jtjd|js4tdS(NR(R=RRR/RRv(R4((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyt test_to_sql-scCsftj|jd|jddtjd|js:ttjttj|jd|jdddS(NRRR( R=RRR/RRvRRR(R4((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyttest_to_sql_fail1s cCstj|jd|jddtj|jd|jddtjd|jsYtt|j}|jd}||kstdS(NRRRR(R=RRR/RRvRR(R4RR((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyttest_to_sql_replace9scCstj|jd|jddtj|jd|jddtjd|jsYtdt|j}|jd}||kstdS(Nt test_frame4RRRi(R=RRR/RRvRR(R4RR((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyttest_to_sql_appendFscCsKtj|jd|jdttjd|j}tj|j|dS(Nt test_frame5R}sSELECT * FROM test_frame5(R=RRR/Rtread_sqlRwR(R4R((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyttest_to_sql_type_mappingTscCslttjddddd}tj|d|jdttjd|j}tj |j |dS( Nitdtypetint64Rtseriest test_seriesR}sSELECT * FROM test_series( R RttarangeR=RR/RRRwRtto_frame(R4tsts2((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyttest_to_sql_seriesZs!signore:\nPanel:FutureWarningcCs/tj}tjttj|d|jdS(Nt test_panel(Rwt makePanelRRtNotImplementedErrorR=RR/(R4tpanel((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyttest_to_sql_panel`s cCstj|jdd|jtjdd|j}|jj|_|jddt|jjt d|j_ t j ||jdS(NRtcons"SELECT * FROM test_frame_roundtripRR(R=RRR/RR}RRtastypetintRcRRwR(R4R((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyttest_roundtripfs   c CsWtj|jdd|jdtddtjdd|j}tj||jdS(NRRR}t chunksizeis"SELECT * FROM test_frame_roundtrip(R=RRR/RRRwR(R4R((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyttest_roundtrip_chunksizets   cCsGtjdd|j}|j}tj|dddddgdS(NsSELECT * FROM irisRgffffff@g @gffffff?g?s Iris-setosa(R=R<R/RRwRx(R4RRl((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyttest_execute_sql|s c Cstjd|j}t|jjjtj s7t tjd|jddg}t|jjjtjsvt |jj t j ddddddt j ddddddgkst tjd|jdid d6}t|jjjtjs t |jj t j ddddddt j ddddddgks[t tjd|jdd g}t|j jjtjst |j j t j d d d dddt j ddddddgkst tjd|jdidd 6}t|j jjtjs.t |j j t j d d d dddt j ddddddgkst tjd|jdidd6}t|jjjtjst |jj t j dt j dgkst dS(NsSELECT * FROM types_test_datat parse_datesR iiiiis%Y-%m-%d %H:%M:%SR!ii iiRs%Y%m%dR"s 2010-10-10s 2010-12-12(R=RR/RsR RRqRtt datetime64RvttolisttpdRR!R"(R4R((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyttest_date_parsings@" ! *! * ! *! *!  cCsmtjd|jdddddg}t|jjjtjsHt t|j jjtjsit dS(NsSELECT * FROM types_test_datat index_colR RR!( R=RR/RsR}RRqRtRRvR!(R4R((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyttest_date_and_indexs !cCsttddgddj}tjt|jd|jWdQXtj d|j}tj |d|dj ddS(Ns00:00:01s00:00:03Rtfoottest_timedeltasSELECT * FROM test_timedeltaR( RR RRwtassert_produces_warningt UserWarningRR/R=Rtassert_series_equalR(R4RR((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRs !cCs9tiddgd6}tjt|jd|jdS(Niy?y@tat test_complexy??(R RRRRR/(R4R((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRscCs titdd6}tj|d|jtjd|j}|jddks]ttj|d|jddd d tjd|j}|jdd kstd |j_ tj|d|jddtjd|j}|jdd ksttj|d|jddd d tjd|j}|jdd ksStd|j_ tj|d|jddtjd|j}|jdd kstd|j_ tj|d|jddd dtjd|j}|jdd kstdS( Nitcol1ttest_index_labelsSELECT * FROM test_index_labeliR}RRt index_labelt other_labelt index_namet0( R RR=RR/RRRvR}RRc(R4t temp_frametframe((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyttest_to_sql_index_labels6      c Cstitdd6dtjddg}tj|d|jtjd |j}|jd d ksrt |jd d kst tj|d|jdddddgtjd |j}|jd j ddgkst ddg|j _ tj|d|jddtjd |j}|jd j ddgksUt tj|d|jdddddgtjd |j}|jd j ddgkst t jttj|d|jdddddS(NiRR}tA0tA1tB0tB1RsSELECT * FROM test_index_labeliRitlevel_1RRRR~RiRR(R R(RR(R RR t from_productR=RR/RRRvRR}tnamesRRR(R4R R ((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyt"test_to_sql_index_label_multiindexs.%%%cCs~tjddgddd d gd dd g}|jd |jtjd |jddd g}tj||dtdS(Nig@Rig?RRR~RRR}ttest_multiindex_roundtrips'SELECT * FROM test_multiindex_roundtripRtcheck_index_type(ig@R(ig?R( R RRR/R=RRwRR(R4RR((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRs  cCsJtddgddggdddg}tj|d|jdd dS( NiiiiRittest_frame_integer_col_namesRR(R R=RR/(R4R((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyttest_integer_col_namess*cCs4tj|jdd|j}d|ks0tdS(NttestRtCREATE(R=t get_schemaRR/Rv(R4t create_sql((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyttest_get_schema"sc Cstiddgd6ddgd6}|jdkr>tjnd}tj|d d |jd i|d6}d |ks~td|kstdS( Ng?g333333?Rg@g@tbRMtINTEGERRRRR(R R[RMtIntegerR=RR/Rv(R4t float_frameRR((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyttest_get_schema_dtypes&s &cCstiddgd6ddgd6}tj|dd|jd d}d }||ks_ttj|jdd|jd d d g}d }||kstdS(Ng?g333333?tCol1g@g@tCol2RRRRs'CONSTRAINT test_pk PRIMARY KEY ("Col1")R~Rs)CONSTRAINT test_pk PRIMARY KEY ("A", "B")(R R=RR/RvR(R4R Rtconstraint_sentence((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyttest_get_schema_keys.s&!cCsttjjdddtd}|jd|jdttj d|j}t}d}ddddd g}xatj d|jd dD]D}t ||gd t }t |||kst |d 7}qWtj|||jd krt}d}ddddd g}xatjd|jd dD]D}t ||gd t }t |||ksvt |d 7}q<Wtj||ndS(NiiRtabcdettest_chunksizeR}sselect * from test_chunksizeiiRt ignore_indexiRM(R RttrandomtrandntlistRR/RR=RRRRRvRwRR[R(R4Rtres1Rtitsizestchunktres3((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyttest_chunksize_read:s,'    cCstidddgd6dddgd6}|j}|djd|d<|jd |jd ttjd |j}tj ||dS( Niiit person_ids John P. Does Jane Dovet person_nametcategoryttest_categoricalR}sSELECT * FROM test_categorical( R tcopyRRR/RR=RRwR(R4Rtdf2R((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyR6\s cCsGtddgddggdddg}|jd|jd tdS( NiiiiRuéubt test_unicodeR}(R RR/R(R4R((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyttest_unicode_column_namejs*cCsqtidddgd6dtjdgd6}|jd|jd ttjd |j}tj ||dS( NiiiR~g?gffffff@Rsd1187b08-4943-4c8d-a7f6R}s'SELECT * FROM `d1187b08-4943-4c8d-a7f6`( R RttnanRR/RR=RRwR(R4RR((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyttest_escaped_table_nameos /  N()R8R9RRaRcR[R_RRRRRRRRRRRRRtmarktfilterwarningsRRRRRRRRR RRRRR"R&R2R6R:R<(((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRsB             .  ( #    "  t TestSQLApicBseZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZRS(s 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. RRMcCs$trtjdStjddS(Nssqlite:///:memory:sSQLAlchemy not installed(tSQLALCHEMY_INSTALLEDRMt create_engineRtskip(R4((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRs cCs_tj|jd|jddg}tjd|jd|}|jj|ks[tdS(Nt test_frameR~RR(R=RRR/RRRRv(R4tcolsR((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyttest_read_table_columnss cCstj|jd|jtjd|jdd}|jjdgksOttjd|jdddg}|jjddgksttjd|jdddgdddg}|jjddgkst|jj ddgkstdS( NRCRR}R~RRRR( R=RRR/RR}RRvRR(R4R((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyttest_read_table_index_cols cCsxtjd|j}tjd|j}tj||tjd|j}tjd|j}tj||dS(NsSELECT * FROM irisRZ(R=RR/RRwRR(R4t iris_frame1t iris_frame2((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyttest_read_sql_delegatescCsd}|jj|d}|jj|tjdtR}tjdtjd|jtjd|jt |dkst WdQXdS(Ns,CREATE TABLE invalid (x INTEGER, y UNKNOWN);s0CREATE TABLE other_table (x INTEGER, y INTEGER);trecordtalwayst other_tablesSELECT * FROM other_tablei( R/R<twarningstcatch_warningsRt simplefilterR=RRRRv(R4tqrytw((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyttest_not_reflect_all_tabless cCsZtjdtB}tjd|jjd|jt|dksPtWdQXdS(NRJRKt CaseSensitivei( RMRNRRORRR/RRv(R4RQ((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyt(test_warning_case_insensitive_table_names cCsUddlm}|jj|j}|jd}g|D]}|d^q;}|S(Ni(t reflectionttest_index_savedt column_names(tsqlalchemy.engineRUt Inspectort from_engineR/t get_indexes(R4ttbl_nameRUtinsptixsR.((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRs cCsxtitddgdtd6}tj|j}tjd|d|}t|jj dj t j stt dS(Nt 201412120154t 201412110254tutcRt test_typeR (R RRR=RKR/tSQLTablet isinstanceRCRIRqtsqltypest TIMESTAMPRv(R4RtdbRC((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyttest_sqlalchemy_type_mappings c Cs|j}tjo}d|}d}|j||dddttj||}tj||}d}tj||}WdQXtj ||tj ||tj ||yddl } t j dWnt k rnXd }t jt d d tjd |WdQXdS( Ns sqlite:///RZRRR}sSELECT * FROM irisispg8000 is installeds)postgresql+pg8000://user:pass@host/dbnametmatchtpg8000sselect * from table(RRwt ensure_cleanRRR=RRRRRjRRBt ImportErrorR( R4RRtdb_uriRCRRRRRj((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyttest_database_uri_strings(    c Cst}|j}|jd||jd|j|jd|j|jd|j|jd|j|jd|j}|S(NRZt SepalLengtht SepalWidtht PetalLengtht PetalWidthtName(RMROtTabletColumntREALtTEXT(R4tsatmetadataRZ((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyt_make_iris_table_metadatas  cCsZtjd}tj||jdidd6}t|d}|dhksVtdS(Ns#select * from iris where name=:nameRWsIris-versicolorRRs(RMttextR=RR/tsetRv(R4t name_texttiris_dft all_names((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyttest_query_by_text_obj s  cCs|j}tj|gj|jjtjdk}tj||j didd6}t |d}|dhkst dS(NRRWs Iris-setosaRs( RzRMtselecttwhereRIRst bindparamR=RR/R|Rv(R4RZt name_selectR~R((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyttest_query_by_select_objs (R8R9RRaR[RRERFRIRRRTRRhRnRzRR(((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyR?zs       t_EngineToConnMixincBs)eZdZejdedZRS(sS A mixin that causes setup_connect to create a conn rather than an engine. Rccstt|j|j}|j}|j|_tj||_ ||_ ||_dV|jj |jj |j |_tj|j |_ dS(N( tsuperRRR/Rtbegint_EngineToConnMixin__txR=RKRt_EngineToConnMixin__enginetrollbackRF(R4RmtengineR/((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyR%s       (R8R9RRRRR(((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyR stTestSQLApiConncBseZRS((R8R9(((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyR9stTestSQLiteFallbackApicBseeZdZdZdZddZdZdZdZdZ d Z d Z d Z RS( s9 Test the public sqlite connection fallback API Rtfallbacks:memory:cCs tj|S(N(tsqlite3R(R4tdatabase((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRGsc Cstjg}|j|}tj|jd|dt|j|j|}tjd|}|jWdQXtj |j|dS(Nttest_frame3_legacyR}s!SELECT * FROM test_frame3_legacy;( RwRkRR=RRRRFRR(R4RR/R((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyttest_sql_open_closeJs    cCs9ts(d}tjttjd|n tjddS(Ns&mysql://root@localhost/pandas_nosetestsSELECT * FROM irissSQLAlchemy is installed(R@RRRlR=RRB(R4R/((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyttest_con_string_import_error\s  cCs]tjd|j}tjd|j}tj||tjtjtjd|jdS(NsSELECT * FROM irisRZ( R=RR/RRwRRRt DatabaseError(R4RGRH((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRIdsc Cs]tddgddggdddg}tj!tj|d|jd tWdQXdS( NiiiiRRsb RR}(R RwRR=RR/R(R4R((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyttest_safe_names_warningks* cCs+tj|jd}d|ks'tdS(NRR(R=RRRv(R4R((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyttest_get_schema2rscCs[xD|jdD]3}|jdjd|kr|jdSqWtd|dS(Ns is""isColumn %s not found(tsplittstripR(R4RNtcolumntcol((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyt_get_sqlite_column_typewscCs}titddgdtd6}tj|j}tjd|d|}|j}|j|ddksyt dS(NR_R`RaRRbR Rf( R RRR=tSQLiteDatabaseR/t SQLiteTablet sql_schemaRRv(R4RRgRCRN((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyttest_sqlite_type_mapping~s  ( R8R9RRaR[RRRRIRRRR(((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyR>s       t_TestSQLAlchemycBseZdZd4ZejdedddZdZ ejdedZ e dZ e dZ e d Zd Zd Zd Zd ZdZdZdZdZdZdZdZdZdZdZdZdZdZdZ dZ!dZ"dZ#dZ$d Z%d!Z&d"Z'd#Z(d$Z)d%Z*d&Z+d'Z,d(Z-d)Z.d*Z/d+Z0d,Z1d-Z2d.Z3d/Z4d0Z5d1Z6d2Z7d3Z8RS(5s Base class for testing the sqlalchemy backend. Subclasses for specific database types are created below. Tests that deviate for each flavor are overwritten there. RtscopetclasscCs.|j|j|j}|jdS(N(t setup_importt setup_driverR(tclsR/((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyt setup_classs   cCs|j|jdS(N(RR(R4((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRs cCs|jdS(N(R(R4Rm((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRscCststjdndS(NsSQLAlchemy not installed(R@RRB(R((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRscCs tdS(N(R(R((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRscCs tdS(N(R(R((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRscCsly5|j|_tj|j|_|jjWn0tjjk rgtj dj |j nXdS(NsCan't connect to {0} server( RR/R=RKRRMtexctOperationalErrorRRBtformatRa(R4((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyR_scCs|jdS(N(R(R4((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyt test_read_sqlscCs|jdS(N(R(R4((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyttest_read_sql_parameterscCs|jdS(N(R(R4((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyttest_read_sql_named_parameterscCs|jdS(N(R(R4((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRscCs|jdS(N(R(R4((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyttest_to_sql_emptyscCs|jdS(N(R(R4((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRscCs|jdS(N(R(R4((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRscCs|jdS(N(R(R4((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRscCs|jdddS(NR5tmulti(R(R4((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyttest_to_sql_method_multiscCs|jdS(N(R(R4((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyttest_to_sql_method_callablescCsv|j}tiddddgd6ddddgd6}tj|}|j|d|jdsrtdS(Ng?g@g@g@tonettwoR (RR R=RKRRRv(R4t temp_connR R((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyttest_create_tables  /cCs|j}tiddddgd6ddddgd6}tj|}|j|d|jdsrt|jd|jd stdS(Ng?g@g@g@RRR (RR R=RKRRRvR2(R4RR R((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyttest_drop_tables / cCs|jdS(N(R(R4((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRscCs|jdS(N(R(R4((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRscCs)tjdd|j}|j|dS(NRZR(R=RR/R|(R4Rz((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyttest_read_tablescCsDtjdd|jdddg}tj|jjddgdS(NRZRRRo(R=RR/RwRxRRy(R4Rz((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyREscCs#tjttjdd|jdS(Ntthis_doesnt_existR(RRRR=RR/(R4((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyttest_read_table_absentscCstjd|j}t|jjjtjs6t t|j jjtj sWt t|j jjtj sxt t|jjjtjst t|jjjtjst dS(NR(R=RR/RsR#RRqRtRuRvR$tintegerR%tbool_R&R'tobject(R4R((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyttest_default_type_conversion s !!!!cCs[tdidgd6}|jd|jdttjd|j}tj||dS(NRYii>ti64t test_bigintR}I@(R RR/RR=RRwR(R4RR((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRscCs:tjd|j}t|jjjtjs6t dS(NR( R=RR/RsR RRqRtRRv(R4R((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyttest_default_date_loadscCsd}tjd|j}t|ds=tjdn|j}t|js[t tjd|jddg}t|dstjdn|j}t|jst t |j j dkst ||jtj ttjd|jddd t}|j}t|js/t t |j j dksMt tjd |j}|j}t|jst tj|j|jtjd |j}||jdS( NcSst|jrJ|dtdks+t|dtdkstnt|jrt|jjdkswttdddtdddg}t|d|j }t j ||ntdj |jdS( Nis2000-01-01 08:00:00is2000-06-01 07:00:00tUTCttzRs)DateCol loaded with incorrect type -> {0}( RRRRvR tstrtdtRR RRwRR(Rt expected_datatexpected((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyR)s sselect * from types_test_dataR)s&no column with datetime with time zoneRRRiR)R(RRR/R0RRBR)R RRvRRRRR,RR=RRwR(R4RRRR((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyttest_datetime_with_timezone$s4        cCstitdddddd6}|jd|jdt|jd krj|djjd |dst test_time3RsSELECT * FROM test_time3cSs |jdS(Ns %H:%M:%S.%f(R(R((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRs(R RRR/RRRwRtTestSQLiteFallbackRR=RtapplymapRa(R4RRt sqlite_conntref((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyttest_datetime_times 3 cCstd dtj}tddtj}ti|d6|d6}|jd|jd ttj d|j}t j ||d td t dS(NiiiRgts1Rttest_read_writeR}t check_dtypet check_exactii( R Rttint32tfloat32R RR/RR=RRwRR(R4RRRR8((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyttest_mixed_dtype_inserts cCstidddgd6dtjdgd6}|jd|jd ttjd|j}tj ||tj d |j}tj ||dS( NiiiR~g?gffffff@Rttest_nanR}sSELECT * FROM test_nan( R RtR;RR/RR=RRwRR(R4RR((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyttest_nan_numerics /cCstidddgd6tjtjtjgd6}|jd|jdttjd|j}tj |||dj d|ds  c Csddg}dtfdg}t|d|}|jd|j|jd|jditjd6tjjd |j}|j |j dj dj }t |tjsttjt|jd |jditd6|jd |jditjd d6|j |j d j dj }t |tjsBt|jd ksWt|jd |jdtjtjjd |j}|j |j d j dj }|j d j dj }t |tjstt |tjstdS(NR~Rg?g?Rt dtype_testt dtype_test2RRLRt dtype_test3i tsingle_dtype_test(g?N(RRcR RR/RMRwRNRORPRQRRqRdRvRRRRtStringR(R4RDRYRRStsqltypetsqltypeatsqltypeb((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyt test_dtypeOs0   # )  cCsqittdgd6ttddddgd6tddgddd6td dgd 6}t|}d }|j||jtj||j}t j j d |j}|j |j d krtj}n tj}|j|j}t|dj|stt|djtjs/tt|djtjsNtt|d jtjsmtdS(NtBooliiitDateRRtIntg?tFloattnotna_dtype_testRLR(R RRcRR RR/R=RRMRNRORPRaReR tBooleanRQRRdRqRvtDateTimeR(R4RDRR6RRStmy_typetcol_dict((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyttest_notna_dtypems$    c Csd}tit|gddd6t|gddd6t|gddd6tdgdd d 6tdgdd d 6}|jd |jdtddditjddd6tjd |j}t j |dj ddt j |dj ddks t tj jd|j}|j|jd j}t|djt|djkskt t|djtjst t|djtjst t|d jtjst t|d jtjst dS(NgYB?RRtf32tfloat64tf64t f64_as_f32iRti32RRt test_dtypesR}RRt precisioniiiRL(R R RR/RRMRR=RRttroundRrRvRNRORPRQRRRqRdReR t BigInteger(R4tVRRRSR ((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyttest_double_precisions&& ,cs^ddfd}tidddgd6jd|j||jdS(NcSsd}tj|d|S(Ns'SELECT test_foo_data FROM test_foo_dataR(R=R(t connectionR((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRscSs |jddd|dddS(NRt test_foo_dataRRR(R(RRY((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pytbarsc sK|j9}|j$|j}|j|WdQXWdQXdS(N(RRt run_callable(t connectableR/tfoo_data(RR(s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pytmains iiiR(R RR/(R4R((RRs7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyttest_connectable_issue_examples    c Csd}ti|gd6}tj}d|fdY}tjd|j}|}|jh|j}|jj ||j |d||j t j dtj|jgd|}WdQXtj||dS(Nu Hello, World!tspamt TemporarycBsTeZdZidgd6ZejejdeZejej dde Z RS(t temp_testt TEMPORARYtprefixest primary_keyitnullable( R8R9t __tablename__t__table_args__RMRuR RtidtUnicodeRR!(((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyR"sRLR=R(R Rtdeclarative_baset sa_sessiont sessionmakerR/t transactionRt __table__tcreatetaddtflushR=RRMRR!RwR( R4t test_dataRtBaseR"tSessionRR/R((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyttest_temporary_tables      N(9R8R9RRcRaRRRRRRt classmethodRRRR_RRRRRRRRRRRRRRRRERRRRRRRRRRRRRRRRRRRRRR RR R7(((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRsd!                    G               t_TestSQLAlchemyConncBseZdZRS(cCstjddS(Ns4Nested transactions rollbacks don't work with Pandas(RRB(R4((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRs(R8R9R(((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyR9st_TestSQLiteAlchemycBsMeZdZdZedZedZdZdZdZ RS(sL Test the sqlalchemy backend against an in-memory sqlite database. RcCs tjdS(Nssqlite:///:memory:(RMRA(R((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRscCs d|_dS(N(Rctdriver(R((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRscCstjd|j}t|jjjtjs6t t|j jjtj sWt t|j jjtj sxt t|j jjtjst t|jjjtjst dS(NR(R=RR/RsR#RRqRtRuRvR$RR%R&R'(R4R((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRs !!!!cCs;tjd|j}t|jjjtj s7t dS(NR( R=RR/RsR RRqRtRRv(R4R((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRscCstiddgd6dd}|jd|jdttjdt?}tjd tj d|jt |d kst WdQXdS( NiiRRRttest_bigintwarningR}RJRKi( R RR/RRMRNRROR=RRRv(R4RRQ((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyttest_bigint_warnings  ( R8R9RRaR8RRRRR=(((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyR:s  t_TestMySQLAlchemycBsDeZdZdZedZedZdZdZRS(sA Test the sqlalchemy backend against an MySQL database. RcCs+d}tj|jd|jd|jS(Ns/mysql+{driver}://root@localhost/pandas_nosetestR;t connect_args(RMRARR;R?(Rturl((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRscCs5tjd}d|_i|jjjd6|_dS(Ntpymysqlt client_flag(Rt importorskipR;t constantstCLIENTtMULTI_STATEMENTSR?(RRA((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRs cCstjd|j}t|jjjtjs6t t|j jjtj sWt t|j jjtj sxt t|j jjtjst t|jjjtjst dS(NR(R=RR/RsR#RRqRtRuRvR$RR%R&R'(R4R((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRs !!!!c Csddl}tidddgd6dddgd 6}|jd |jd td }|jj}|j}y|j|}|jWn!|j k r|j nXt j d |j}t j||t jd |j}t j||dS(NiiiiRg?g?g333333?Rttest_procedureR}sDROP PROCEDURE IF EXISTS get_testdb; CREATE PROCEDURE get_testdb () BEGIN SELECT * FROM test_procedure; ENDsCALL get_testdb();(RAR RR/RRRR<R?RDRR=RRwRR( R4RARtprocRRtr1R-R((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyttest_read_procedure,s  ,  ( R8R9RRaR8RRRRJ(((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyR> s  t_TestPostgreSQLAlchemycBsDeZdZdZedZedZdZdZRS(sF Test the sqlalchemy backend against an PostgreSQL database. RcCs"d}tj|jd|jS(Ns8postgresql+{driver}://postgres@localhost/pandas_nosetestR;(RMRARR;(RR@((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRSscCstjdd|_dS(Ntpsycopg2(RRCR;(R((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRXs c Cstiddgd6ddgd6ddgd 6}|jjd |jjd |jd |jd t|jd|jd tdd|jd|jd tddtjd |j}tj||tjd|j}tj||tjd|jdd}tj||tjd|jdd}tj||t j t tjd|jdd|jjd |jjd |jd|jddd t|jd|jddd tdd|jd|jddd tddtjd|jdd}tjt ||gdt |t|jtjjr |j}tj|dd}tj|d|} | j|dd t| j|dd tdd| j|dd tddtjd|jdd}| jd}tj||ndS(NiiRg?g?tcol2Rtntcol3s$DROP SCHEMA IF EXISTS other CASCADE;sCREATE SCHEMA other;ttest_schema_publicR}ttest_schema_public_explicitRNtpublicttest_schema_othertotherRRRR)RSttest_schema_other2(R R/R<RRR=RRwRRRRRRRdRMRtEngineRRORKt read_table( R4RR-RR1tres4Rtengine2RStpdsql((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyttest_schema_support]sV   "   cCsd}tiddgd6ddgd6dd gd 6}|jd |jd td |tjd |j}tj||dS(Nc Ss|j}|j}tj}tj|}|j||jddjd|D}|j rdj |j |j } n |j } dj | |} |j d| d|WdQXdS(Nis, css|]}dj|VqdS(s"{}"N(R(t.0tk((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pys ss{}.{}s COPY {} ({}) FROM STDIN WITH CSVR=tfile( RR;tcompattStringIORdtwritert writerowstseektjoinRNRRt copy_expert( RCR/RRRt dbapi_connRAts_bufRaRR@t sql_query((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pytpsql_insert_copys       iiRg?g?RMRRNROttest_copy_insertR}R5(R RR/RR=RRwR(R4RiRR((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyt(test_copy_from_callable_insertion_methods  ( R8R9RRaR8RRR[Rk(((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRKLs  @tTestMySQLAlchemycBseZRS((R8R9(((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRlstTestMySQLAlchemyConncBseZRS((R8R9(((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRmstTestPostgreSQLAlchemycBseZRS((R8R9(((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRnstTestPostgreSQLAlchemyConncBseZRS((R8R9(((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRostTestSQLiteAlchemycBseZRS((R8R9(((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRpstTestSQLiteAlchemyConncBseZRS((R8R9(((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRqsRcBseZdZdZedZdZdZej de dZ dZ dZ d Zd Zd Zd Zd ZdZdZdZdZdZdZdZdZdZdZdZdZdZRS(sG Test the fallback mode against an in-memory sqlite database. RcCs tjdS(Ns:memory:(RR(R((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRscCs|j|_dS(N(RR/(R4((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyR_scCs#tj|j|_|jdS(N(R=RR/RR(R4((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRsRcCs|jdS(N(R(R4Rm((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRscCs|jdS(N(R(R4((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRscCs|jdS(N(R(R4((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRscCs|jdS(N(R(R4((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRscCs|jdS(N(R(R4((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRscCs|jdS(N(R(R4((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRscCs|jdS(N(R(R4((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRscCs|jdS(N(R(R4((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyR scCs|jdS(N(R(R4((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyR scCstiddddgd6ddddgd6}|jj|d|jjds]t|jjd|jjd stdS(Ng?g@g@g@RRtdrop_test_frame(R RRRRvR2(R4R ((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyttest_create_and_drop_tables /cCs|jdS(N(R(R4((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRscCs|jdS(N(R(R4((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRscCsttdddtdddgddg}|jd|jdttd|j}|jd krtj||j t n"|jd krtj||ndS( NiiiRRRR}sSELECT * FROM test_dateRR( R RRR/RRRaRwRRR(R4RR((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyR"s3cCsttdddtdddgddg}|jd|jdttd |j}|jd kr|jd }tj ||ndS( Ni iiiRRRR}sSELECT * FROM test_timeRcSs |jdS(Ns %H:%M:%S.%f(R(R((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyR4s( R RRR/RRRaRRwR(R4RRR((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyR-s 3cCsjtjdd||j}g}x@|jD]5}tjd||j}|j|jjq-W|S(Ns1SELECT * FROM sqlite_master WHERE type = 'index' sAND tbl_name = '%s'sPRAGMA index_info(%s)(R=RR/RRR(R4R\R^Rtix_nametix_info((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyR7scCs|jdS(N(R(R4((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRBscCs$trtjdn|jdS(Nsnot working on python > 3.5(RRRBR(R4((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyREsc Csc|jjd|}x0|D](\}}}}}} ||kr|SqWtd||fdS(NsPRAGMA table_info(%s)sTable %s, column %s not found(R/R<R( R4RCRtrecstcidRtctypetnot_nulltdefaulttpk((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRJs  c Cs@|jdkrtjdnddg}dtfdg}t|d|}|jd|j|jd |jd id d6|jddd kst |jd dd kst tj t |jd |jd it d6|jd|jd d |jddd kst |jddd ks<t dS(NRsNot applicable to MySQL legacyR~Rg?g?RRRRtSTRINGRRR(g?N( RaRRBRRcR RR/RRvRRtbool(R4RDRYR((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRQs&    cCs$|jdkrtjdnittdgd6ttddddgd6tddgdd d 6td dgd 6}t|}d }|j||j |j |ddkst |j |ddkst |j |d dkst |j |d dks t dS(NRsNot applicable to MySQL legacyRiiiRRRRg?RRRRfRv( RaRRBR RRcRR RR/RRv(R4RDRR6((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyR ks c Cstddgddggdddg}tjt|jd|jxtd d d d d dddddg D]\}}|j||jtj||jtddgddggdd|g}d|}|j||jtj||jqqWdS(NiiiiRRRtstest_weird_name]stest_weird_name[stest_weird_name`stest_weird_name"stest_weird_name's_b.test_weird_name_01-30s"_b.test_weird_name_01-30"t99beginswithnumbert12345uéstest_weird_col_name%d( R RRRRR/t enumerateR=t table_exists(R4Rtndxt weird_nameR8tc_tbl((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyttest_illegal_names~s* * ( R8R9RRaR8RR_RRRRRRRRRRRRRRsRRRRRRRRRR R(((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRs4                 cCs |jdS(s Returns date in YYYYMMDD format.s%Y%m%d(R(R((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyt date_formatscCsdt|S(Ns'%s'(R(R((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRscCsd|S(Ns'%s'((tx((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRscCsd|S(Ns'%s'((R((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRscCsd|S(Ns'%s'((R((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRscCsd|S(Ns'%s'((R((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRscCsd|S(Ns%.8f((R((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRscCsd|S(Ns%s((R((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRscCsdS(NtNULL((R((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRscCsd|S(Ns%.10f((R((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRscCsd|S(Ns'%s'((R((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRscGslg}xU|D]M}t|tr7t|r7d}ntt|}|j||q W|t|S(s N(RdtfloatRRct _formattersRqRttuple(R=targstprocessed_argstargt formatter((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyt format_querys  cCs?tj|d|d|j}|dkr1dSt|SdS(s#Replace removed sql.tquery functionRRAN(R=R<RBRcR,(RRRAR((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyttquerys! t TestXSQLitecBseZejdedZdZdZdZdZ dZ dZ dZ d Z d Zd Zd ZRS( RccsE|j|_tjd|_dV|j|_tjd|_dS(Ns:memory:(tfunctionR5RRR/(R4RhRg((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRs   cCstj}|j|dS(N(RwtmakeTimeDataFramet_check_roundtrip(R4R ((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyt test_basics c Cstj}tj|jdR ssselect * from mono_dfi( R R=RR/RtsumR<RvRRwR(R4Rtcon_xtthe_sumR((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyttest_onecolumn_of_integerJ s"  c shtiddgd6ddgd6}tiddd gd6d d d gd6}d }d|}fd}tjttjd|djd|dd||tjd|djd|ddtjttjd|djd|ddtjd|djd|dddtt|djddgksNt tjd|djd|dddtt|djdddgkst ||tjd|djd|dddtt|djddgkst tjd|djd|dddtt|djd d!d"d#d$gksZt ||dS(%NiiRR~RRMiiiRRRttable_if_existssSELECT * FROM %scsj|dS(s Drops tables created from individual tests so no dependencies arise from sequential tests N(R2(ttest_table_to_drop(R4(s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pytclean_upa sR RRRt notvalidvalueRRR}R(iR~(iR(iR(iR(iR(iR~(iR(iR~(iR(iR(iR(iR( R RRRR=RR/RRRv(R4tdf_if_exists_1tdf_if_exists_2R@t sql_selectR((R4s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyttest_if_existsZ sJ&)          '   ' (R8R9RRRRRRRRRRRRRRR(((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRs          treasonsFgh-13611: there is no support for MySQL if SQLAlchemy is not installedt TestXMySQLcBseZejdedddZejdedZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZRS(RRRc Cstjd}|jdddddddd y|jd d Wn?|jk rgtd n |jk rtd nXdS(NRAthostt localhosttusertroottpasswdR~Rgtpandas_nosetesttread_default_grouptpandassCreate a group of connection parameters under the heading [pandas] in your system's mysql default file, typically located at ~/.my.cnf or /etc/.my.cnf.sCannot connect to database. Create a group of connection parameters under the heading [pandas] in your system's mysql default file, typically located at ~/.my.cnf or /etc/.my.cnf.(RRCRtProgrammingErrort RuntimeErrorRD(RRA((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyR s c Cstjd}|jdddddddd y|jd d Wn?|jk rgtd n |jk rtd nX|j|_dS(NRARRRRRR~RgRRRsCreate a group of connection parameters under the heading [pandas] in your system's mysql default file, typically located at ~/.my.cnf or /etc/.my.cnf.sCannot connect to database. Create a group of connection parameters under the heading [pandas] in your system's mysql default file, typically located at ~/.my.cnf or /etc/.my.cnf.(RRCRRRRDRR5(R4RhRgRA((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyR s  cCstj}|j|dS(N(RwRR(R4R ((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyR s c Cstj}tj|jd R<RrRyRRR?RR}R(R4R RRRARkRlR((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyR s    cCstj}tj|d}|j}xX|D]P}|jd}t|dkr1|ddkr1|ddkstq1q1Wtj}d}tj|dd dd g}|j}d |kst|jj }|j ||j |dS( NRRiiR~iRsDROP TABLE IF EXISTS testRRRsPRIMARY KEY (`A`, `B`)( RwRR=RRRRRvR/R;R<(R4R RRRRRRA((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyR s   "   cCsd}d}|jj}|j||j|tjd|jtjd|jtjttjd|jWdQXdS(NsDROP TABLE IF EXISTS tests CREATE TABLE test ( a TEXT, b TEXT, c REAL, PRIMARY KEY (a(5), b(5)) ); s,INSERT INTO test VALUES("foo", "bar", 1.234)s,INSERT INTO test VALUES("foo", "baz", 2.567)s(INSERT INTO test VALUES("foo", "bar", 7)(R/R;R<R=RRR(R4RRRA((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyR s   cCsd}d}|jj}|j||j|tjd|j|jjtjttdd|jWdQX|j ||dS(NsDROP TABLE IF EXISTS tests CREATE TABLE test ( a TEXT, b TEXT, c REAL, PRIMARY KEY (a(5), b(5)) ); s,INSERT INTO test VALUES("foo", "bar", 1.234)sselect * from testR( R/R;R<R=RFRRRRR(R4RhRgRRRA((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyR s    cCsdS(N((R4((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyR5 sc Csd}|jj}tj"tjdd|j|WdQXtj|ddd|jdttj d|j}|j |_ |j j |j _ |}t j ||d gt||d <|j}ttt|d }||d R<R=RRRR}RRwRRR7R RR(R4R RRARRRR}((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyR8 s8 "       c CsHtitjdd6}tj|d|jdddddtdS( NiRRRRRRR}(R RtRR=RR/R(R4R((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyR\ sc sntiddgd6ddgd6}tiddd gd6d d d gd6}d }d|}fd}tjttjd|djd|dd||tjd|djd|dddttjttjd|djd|ddtjd|djd|dddtt|djddgksTt tjd|djd|dddtt|djdddgkst ||tjd|djd|dddtt|djddgkst tjd|djd|dddtt|djd d!d"d#d$gks`t ||dS(%NiiRR~RRMiiiRRRRsSELECT * FROM %scsj|dS(s Drops tables created from individual tests so no dependencies arise from sequential tests N(R2(R(R4(s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRh sR RRRRRR}RR(iR~(iR(iR(iR(iR(iR~(iR(iR~(iR(iR(iR(iR( R RRRR=RR/RRRv(R4RRR@RR((R4s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyRa sJ&)          '   ' (R8R9RRRRRRRRRRRRRRRR(((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyR s!         $ (cRt __future__RRdRRRRRMtnumpyRtRt pandas.compatR_RRRRtpandas.core.dtypes.commonRR RRR R R R RRRRRRtpandas.util.testingtutilttestingRwt pandas.io.sqlRXR=RRRMtsqlalchemy.schematsqlalchemy.sql.sqltypesRetsqlalchemy.extRtsqlalchemy.ormRR-RR@RlRR`RR.R:RGRJRURR=tsingleR?RRRRR9R:R>RKRgRlRmRnRoRpRqRRRtstr_t text_typet binary_typeRRRqRcRR}RRRRRBR(((s7lib/python2.7/site-packages/pandas/tests/io/test_sql.pyts     " F                   z   NG0Cp