
    dp                     ~   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Z ddl	m
c mZ ddlZddlmZmZ ddlmZ d Zej        d             Zd Z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         d                         Z!d Z"ej        j         ej         d                         Z#d Z$ej        j         ej         ej        %                    e&d          d                                     Z'd Z(d Z)d Z*d Z+dS )z1
Testing that we work in the downstream packages
    N)IntCastingNaNError)	DataFrameSeriesc                 ~    	 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    X/var/www/html/t/fyr/venv311/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   c                 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c                     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   7   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        ||           d}t          j	        t          |          5  t          |d           d d d            n# 1 swxY w Y   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            d S # 1 swxY w Y   d S )	Nr   )r   g      @r   z)Trying to coerce float values to integersmatchi8)dtypez9Cannot convert non-finite values \(NA or inf\) to integerr   )r   r   r+   r)   
from_arrayr   r,   r-   r
   raises
ValueErrornanr   )r$   arrdarrresr2   msgs         r   7test_construct_dask_float_array_int_dtype_match_ndarrayrA   K   s     
(;;;

C==D
,,Cc{{H3))) 6C	z	-	-	- ! !t4    ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! GCVCF	)	5	5	5 ! !t4    ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! 
)	5	5	5    s$                                   s6   B  B$'B$C55C9<C9D::D>D>c                 P    t          d          }|                                 J d S )Nxarray)r   	to_xarray)r   rC   s     r   test_xarrayrE   f   s)    8$$F<<>>%%%%%r   cftimerC   z0.21.0c                      dd l } dd l}|                    dd          }|                     ddd          }|                    |gd          }d}||k    sJ d S )	Nr   0001r   )periodsi  r   nearest)method)rF   rC   cftime_rangeDatetimeGregorianget_indexer)rF   rC   timeskeyr1   r2   s         r   test_xarray_cftimeindex_nearestrQ   l   s{     MMMMMM22E

"
"4A
.
.CuY77FHXr   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[   z   s$    3>5$HIIIIIr   c                  J    t          j        t          j        dddg           d S )NrS   rT   z_import pandas as pd, pickle; pickle.loads(pickle.dumps(pd.date_range('2021-01-01', periods=1)))rV   r   r   r   )test_oo_optimized_datetime_index_unpickler]      s7    NU	

 
 
 
 
r   c                      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_rdatasetrb   olsfit)r_   smsmfr   s       r   test_statsmodelsrm      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   )rg   svmgMbP?g      Y@)gammaC)
r   ro   rg   rp   load_digitsSVCrj   rb   targetpredict)ro   rg   rp   digitsclfs        r   test_scikit_learnrz      s    I&&G       
 !!##F
'''
'
'CGGFKfmCRC0111KKBCC !!!!!r   c                  ~    t          d          } |                     d          }|                     dd|           d S )Nseaborntipsday
total_bill)xyrb   )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)r:   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          }|j                            |           }|                                }t	          j        ||            d S )Npyarrow)r   Tabler    	to_pandasr,   assert_frame_equal)r   r   tabler1   s       r   test_pyarrowr      sL    I&&GM%%b))E__F&"%%%%%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\/rT   z$import pandas;print(pandas.__file__)zsite-packagesz pandas installed as site packagez-sSErU   zCommand '\['zD', '-sSE', '-c', 'import pandas'\]' returned non-zero exit status 1.r5   )stderr)numpypytzdateutil)rY   rZ   replacerW   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      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   rW   rY   r   r+   r
   pandas.errorsr   pandas.util._test_decoratorsutil_test_decoratorstdpandasr   r   r   pandas._testing_testingr,   r   fixturer   r&   r3   
skip_if_norA   rE   rQ   r[   r]   marknetworkrm   rz   r   r   xfailr;   r   r   r   r   r   r   r   r   <module>r      s            



      , , , , , , ) ) ) ) ) ) ) ) )                 9 9 9 ' ' '5 5 5$5 5 5( v     4& & & xx""	 	 #" 	J J J
   C C  C
" 
" 
" : :  :- - - D   
L L   L
& & &
' 
' 
'! ! !H5 5 5 5 5r   