`L i(>dZddlZddlZddlZddlZddlZdZejjdejjdejje ddd ejjdid Z d Z d Z dd ZdZddZejj%deddZejj%deddZejj%deddZy)z&Test loaders for common functionality.Nc, ddl}y#t$rYywxYw)NrTF)PIL ImportError)rs h/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/sklearn/datasets/tests/test_common.pyis_pillow_installedr s s  z.X is a list and does not have a shape argument)reasonz)fetch_opeml requires a dataset name or idzpillow is not installed)fetch_20newsgroups fetch_openmlfetch_lfw_peopler ) return_X_yas_framec ddl}tjdy#t$rN|j}|d}tj t|5|ddddYy#1swYYyxYwwxYw)Nrz-This test requires pandas to not be installedz# with as_frame=True requires pandas)matchTr )pandaspytestskipr__name__raises) fetch_funcrname expected_msgs rcheck_pandas_dependency_messager)sk & CD &""BC ]];l ; &  % & & & &s'5A3 A%A3%A/ *A3/A3c|d}t|tsJ|dj|jjk(sJ|dj|jjk(sJy)NT)r r) isinstancetupleshapedatatarget)bunch dataset_func X_y_tuples rcheck_return_X_yr$7s`-I i '' ' Q<  !1!1 11 1 Q<  !3!3 33 3ctjd}|d}t|dsJt|j|j sJt|j |j sJ|j j|j jk(sJ|jjdkDr"t|j|j s$Jt|j|jsJ|jjd|jjdk(sJ|.tj|j j|k(sJ|.tj|jj|k(sJ|dd\}}t||j sJ|jdkDrt||j sJyt||jsJy)NrTrframerr)r r )r importorskiphasattrrr' DataFramerrr ndimSeriesnpalldtypes)r!r"expected_data_dtypeexpected_target_dtypepd frame_bunchframe_Xframe_ys rcheck_as_framer6>s   X &B-K ; (( ( k'' 66 6 k&& 55 5    ! !UZZ%5%5 55 5"+,,bll;;;+,,bii888    # #A &%,,*<*Zs ::>>6 < CCr%c #@Ktj|i}tjtj D]\}}tj |st|Dcgc]}|j|c}}|tj|jv}|sh|sktjj|jdxr tdg}||vr|j||tj |||ycc}ww)Nfetchz.Skip because fetcher requires internet network) conditionr)marks)FETCH_PYTEST_MARKERSr<inspect getmemberssklearndatasets isfunctionany startswith signature parametersrmarkskipifr>appendparam) rP dataset_type markers_fetchrobjtis_dataset_typeis_support_paramrBs r_generate_func_supporting_paramrW^s(,,UB7M''(8(897 c!!#& <Hatq1HI G$5$5c$:$E$EE / """oog6P;N;PK#E }$ ]401,,tS6 6#7IsAD!D 9*D$D'A7Dzname, dataset_funcr c*|}t||yN)r$rr"r!s rtest_common_check_return_X_yr[ts NEUL)r%r c*|}t||yrY)r6rZs rtest_common_check_as_framer]|s NE5,'r%ct|yrY)r)rr"s r#test_common_check_pandas_dependencyr_s $L1r%)NN))loadr@)__doc__rDr:numpyr-rsklearn.datasetsrFrrMxfailrNrCrr$r6r>rW parametrizer[r]r_r=r%rrfsT,  %kk//C0  ))>* #KK..#% %.G/    ))>* ( &4JN.8D7,9,G** 9*E(( 9*E22r%