
    Wh:&              	          d Z ddlZddlZddlZddlZddlZddlmc m	Z
 ddlZddlmZmZ ddlmZ ej                            d          Zd Zej        d             Zej                            d          d             Zej                            d          ej                            d	          d
                         Z e
j        d          d             Zd Z e
j        d           e
j        dd          d                         Zd Zd Zej        j         ej         ej                            d          ej                            d          ej                            d          ej                            d          ej                            d          d                                                                                     Z!ej                            d          d             Z"ej        j         ej         ej                            d          d                                     Z#d Z$ej        j         ej         ej        %                    e&d          d                                     Z'd  Z(ej                            d!          ej                            d"          d#                         Z)d$ Z*d% Z+d& Z,d' Z-dS )(z1
Testing that we work in the downstream packages
    N)	DataFrameSerieszDignore:distutils Version classes are deprecated.*:DeprecationWarningc                 ~    	 t          j        |           S # t          $ r t          j        d|  d           Y d S w xY w)Nzskipping as z not available)	importlibimport_moduleModuleNotFoundErrorpytestskip)names    b/var/www/html/movieo_spanner_bot/venv/lib/python3.11/site-packages/pandas/tests/test_downstream.pyr   r      sX    9&t,,, 9 9 9747778888889s    "<<c                  (    t          dg di          S )NA         )r        r   dfr   $   s    c999%&&&r   z,ignore:.*64Index is deprecated:FutureWarningc                 D   t          j        d          }	 t          d          }t          d          }dd lm} |                    | d          }|j        J |                                J 	 t          j        d|           d S # t          j        d|           w xY w)Ncompute.use_numexprtoolzdaskr   r   npartitions)	pd
get_optionr   dask.dataframe	dataframefrom_pandasr   compute
set_option)r   olduser   r   ddddfs         r   	test_daskr&   )   s    
 ]011F
5g&&V$$######nnRQn//u   {{}}((((
+V44444+V4444s   AB Bz,ignore:The __array_wrap__:DeprecationWarningc                     t          j        d          } 	 t          d          }dd lm} dd lm} t          g d          }|                    |d          }|	                    |          
                                }t          j	        |          }t          j        ||           t          j        d|            d S # t          j        d|            w xY w)Nr   r   r   )g      ?gffffff@g@g      @r   r   )r   r   r   
dask.arrayarrayr   r   r   r    fixr!   nptmassert_series_equalr"   )r#   r   dar$   sdsresultexpecteds           r   test_dask_ufuncr3   =   s     ]011F5V$$######'''((^^A1^--##%%6!99
vx000
+V44444+V4444s   BC   Cr   c                     dd l m}  t          j        g d          }|                     |          }t          |          }t          |          }t          j        ||           t          |d          }t          |d          }t          j        ||           d}t          j        |d<   t          j	        t          |          5  t          |d          }d d d            n# 1 swxY w Y   t          j	        t          |          5  t          |d          }d d d            n# 1 swxY w Y   t          j        ||           d S )Nr   )r   g      @r   i8)dtypez>In a future version, passing float-dtype values containing NaNr   match)r   r   r+   r)   
from_arrayr   r,   r-   nanassert_produces_warningFutureWarning)r$   arrdarrresr2   msgs         r   7test_construct_dask_float_array_int_dtype_match_ndarrayrA   V   s     
(;;;

C==D
,,Cc{{H3)))
T
"
"
"Cc&&&H3)))
JCVCF		#M	=	=	= ' 'T&&&' ' ' ' ' ' ' ' ' ' ' ' ' ' '		#M	=	=	= + +#T***+ + + + + + + + + + + + + + +3)))))s$   C&&C*-C*D**D.1D.c                 P    t          d          }|                                 J d S )Nxarray)r   	to_xarray)r   rC   s     r   test_xarrayrE   p   s)    8$$F<<>>%%%%%r   cftimerC   z0.10.4c                     dd l } dd l}|                    dd          }|                     ddd          }t	          j        t          dd	          5  |                    |d
          }d d d            n# 1 swxY w Y   d}||k    sJ d S )Nr   0001r   )periodsi  r   
deprecatedF)r8   check_stacklevelnearest)method)rF   rC   cftime_rangeDatetimeGregorianr,   r;   r<   get_loc)rF   rC   timeskeyr1   r2   s         r   test_xarray_cftimeindex_nearestrS   w   s     MMMMMM22E

"
"4A
.
.C		#\E
 
 
 6 6 s9556 6 6 6 6 6 6 6 6 6 6 6 6 6 6 HXs   A77A;>A;c                  J    t          j        t          j        dddg           d S )N-OO-cimport pandas
subprocess
check_callsys
executabler   r   r   test_oo_optimizabler]      s$    3>5$HIIIIIr   c                  J    t          j        t          j        dddg           d S )NrU   rV   z_import pandas as pd, pickle; pickle.loads(pickle.dumps(pd.date_range('2021-01-01', periods=1)))rX   r   r   r   )test_oo_optimized_datetime_index_unpickler_      s7    NU	

 
 
 
 
r   z(ignore:pandas.util.testing is deprecatedzignore:can't:ImportWarningzGignore:Using or importing the ABCs from 'collections:DeprecationWarningz5ignore:`np.MachAr` is deprecated.*:DeprecationWarningc                      t          d          } dd lm} dd lmc m} |j                            dd          j        }|                    d|          	                                 d S )Nstatsmodelsr   GuerryHistDataz$Lottery ~ Literacy + np.log(Pop1831))data)
r   statsmodels.apiapistatsmodels.formula.apiformuladatasetsget_rdatasetrd   olsfit)ra   smsmfr   s       r   test_statsmodelsro      s       ..K      )))))))))		!	!(J	7	7	<BGG2G<<@@BBBBBr   c                  .   t          d          } ddlm}m} |                                }|                    dd          }|                    |j        d d         |j        d d                    |	                    |j        dd                     d S )Nsklearnr   )ri   svmgMbP?g      Y@)gammaC)
r   rq   ri   rr   load_digitsSVCrl   rd   targetpredict)rq   ri   rr   digitsclfs        r   test_scikit_learnr|      s     I&&G       
 !!##F
'''
'
'CGGFKfmCRC0111KKBCC !!!!!r   ignorec                  ~    t          d          } |                     d          }|                     dd|           d S )Nseaborntipsday
total_bill)xyrd   )r   load_dataset	stripplot)r   r   s     r   test_seabornr      sE    
 I&&G''DD99999r   c                  P    t          j        dd           t          d          } d S )N
pandas_gbqz0.10.0)
minversion)r	   importorskipr   )r   s    r   test_pandas_gbqr      s)    
::::|,,JJJr   z|The Quandl API key must be provided either through the api_key variable or through the environmental variable QUANDL_API_KEY)raisesreasonc                  T    t          d          } |                     dddd           d S )Npandas_datareaderFquandlz
2017-01-01z
2017-02-01)r   
DataReader)r   s    r   test_pandas_datareaderr      s3     &&9::  hlKKKKKr   c                      t          d          } |                     g d|                     g dg d          d          }|ddg         j        j                            t          g d                    sJ d S )N	geopandasr   )colgeometryr   r   )g      ?g       @g      @)r   GeoDataFramepoints_from_xyr   r   equalsr   )r   gdfs     r   test_geopandasr      s    k**I

 
 		y'?'?			999'U'UVV C z"#,.55f___6M6MNNNNNNNr   z"ignore:can't resolve:ImportWarningz4ignore:RangeIndex.* is deprecated:DeprecationWarningc                     t          d          }|j                            |           }|                                }t	          j        ||            d S )Npyarrow)r   Tabler    	to_pandasr,   assert_frame_equal)r   r   tabler1   s       r   test_pyarrowr      sN     I&&GM%%b))E__F&"%%%%%r   c                     t          d          }|                    dd          }t          |          }| st          j        ||          sJ t          |d                   }t          j        ||          sJ d S )Ntorchi  @   r   )r   randnr   r+   shares_memoryr   )using_array_managerr   
val_tensorr   sers        r   test_torch_frame_constructionr      s    '""ES"%%J	:		B 0J/////
A

CC,,,,,,,r   c                    t          d          }|                    |           }|                    ||j                  }t	          j        | |           |                    ||j                  }t	          j        | |           d S )Nyaml)Loader)r   dumploadr   r,   r   UnsafeLoader)r   r   dumpedloadedloaded2s        r   test_yaml_dumpr     sy      DYYr]]FYYvdkY22F"f%%%iit'8i99G"g&&&&&r   c                     t           j                            dd          } | ddg}t          j        |                                          }d|v rt          j        d           | dddg}d	|  d
}t          j        t          j	        |          5 }t          j        |t          j
                   d d d            n# 1 swxY w Y   |j        j                                        }dD ]}||v sJ 	d S )N\/rV   z$import pandas;print(pandas.__file__)zsite-packagesz pandas installed as site packagez-sSErW   zCommand '\['zD', '-sSE', '-c', 'import pandas'\]' returned non-zero exit status 1.r7   )stderr)numpypytzdateutil)r[   r\   replacerY   check_outputdecoder	   r
   r   CalledProcessErrorSTDOUTvaluestdout)pyexecalloutputr@   excr   s         r    test_missing_required_dependencyr     sb    N""4--E
 4?@D$T**1133F&  6777
 641D	+ 	+ 	+ 	+ 
 
z4C	@	@	@ @CZ->????@ @ @ @ @ @ @ @ @ @ @ @ @ @ @ Y$$&&F-  v~~~~~ s   !B==CCc                     t          j        d          } 	 t          d          }dd lm} |                    ddg          }t          dddgi          }||d<   ||d<   d	|j        d
dgdf<   |j        dgd d f         }t          dgd	gdgddg          }t          j        ||           t          j	        d|            d S # t          j	        d|            w xY w)Nr   r   r   r   r   abcd   FT)r   r   r   )index)
r   r   r   r(   r)   r   locr,   r   r"   )r#   r   r.   ddar   r1   r2   s          r   *test_frame_setitem_dask_array_into_new_colr   5  s   
 ]011F5V$$hh1vc3Z())33%(t}c!"QQQC5uA3??sKKK
fh///
+V44444+V4444s   BC C).__doc__r   rY   r[   r   r+   r	   pandas.util._test_decoratorsutil_test_decoratorstdpandasr   r   r   pandas._testing_testingr,   markfilterwarnings
pytestmarkr   fixturer   r&   r3   
skip_if_norA   rE   rS   r]   r_   networkro   r|   r   r   xfail
ValueErrorr   r   r   r   r   r   r   r   r   r   <module>r      sU            



      ) ) ) ) ) ) ) ) )                  [''J 

9 9 9 ' ' ' JKK5 5 LK5& JKKJKK5 5 LK LK5. v* * *2& & & xx""  #" J J J
   FGG899JKKM  ; C C	 	  LK :9 HG  C 899" " :9" H%%: : &%  :- - - D   
L L   LO O O @AARSS& & TS BA&- - -
' 
' 
'! ! !H5 5 5 5 5r   