tKgu<NdZddlZddlZddlmZmZddlmZm Z m Z m Z m Z m Z mZmZddlmZmZddlmZej*j-Zej1ee e egZej4ej6ej8gZej<dZej<dZ ejBjEd ee e e gd Z#ejBjEd eGd d Z$ejBjEd e egGd dZ%y)z*Test of 1D aspects of sparse array classesN) assert_equalassert_allclose) bsr_array csc_array dia_array lil_array coo_array csr_array dok_arraySparseEfficiencyWarning)supported_dtypesmatrix)ComplexWarningc2tjgddS)N)rrd)nparrayd/home/alanp/www/video.onchill/myenv/lib/python3.12/site-packages/scipy/sparse/tests/test_common1d.pydat1drs 88L# &&rctDcic]}||j|}}tDcic]/}||jDcgc]\}}||||fc}}1c}}}Scc}wcc}}wcc}}}wN) math_dtypesastype spcreatorsitems)rdtype dat_dtypes spcreatordats rdatsp_math_dtypesr$s:EF+%e,,+JF$ #I :CSCSCUVCUZUCUC30CUVV# GV sA'A2A,A2,A2r"cxtjtd5|gddddy#1swYyxYw)Nzarrays don't support 1D inputmatchrrrpytestraises ValueError)r"s rtest_no_1d_support_in_initr.(s& z)H I, J I Is 09ceZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZdZedZedZedZdZdZdZdZdZdZdZdZ dZ!dZ"y )! TestCommon1Dz5test common functionality shared by 1D sparse formatsct|djtjdt|djdt|dj dy)Nrrr)rtoarrayrzerosnnz count_nonzeroselfr"s rtest_create_emptyzTestCommon1D.test_create_empty3sLYt_,,. <Yt_((!,Yt_224a8rcttjtd5|ddddy#1swYyxYw)Nzelements cannot be negativer&)r*r7s rtest_invalid_shapesz TestCommon1D.test_invalid_shapes8s& ]]:-J K e L K Ks .7c&t||yr)reprr8r"rs r test_reprzTestCommon1D.test_repr<s Yu rc&t||yr)strr?s rtest_strzTestCommon1D.test_str?s Ie rcxtjgdd}t| || jy)N) rrrrrrrrr)rrrr3r8r"As rtest_negzTestCommon1D.test_negBs. HH=s CaR9Q<-0023rc:|gd}|jdk(sJy)Nr(r)ndimrIs rtest_1d_supported_initz#TestCommon1D.test_1d_supported_initFs l #vv{{rc|gd}|jdd}gdg}t|j|||}|jd}t|j|d|jd}|j|jk(sJ|jjd}t|j|dy)N rrrrrrr;rrrr rrS)reshaperr3shapeT)r8r"xydesiredy2s r$test_reshape_1d_tofrom_row_or_columnz1TestCommon1D.test_reshape_1d_tofrom_row_or_columnJs ; < IIa 89QYY['* g  IIbMQYY['!*- YYu ww"((""" CCKKOQYY['!*-rc*|gd}|jd}gdgdgdgdg}t|j||jd}||usJ|jd}t|j|jy) NrP)r)rrrQ)rrr)rr;r)rrrRrTrS)rUrr3)r8r"rXrYrZs r test_reshapezTestCommon1D.test_reshape\su ; < IIf iY?QYY['* IIe Av v IIbMQYY[!))+.rcrtjjdtjgd}tjj d}tjg}tj d}||||g}|D]}||}tj d5tj|jsJt|j|jt|jd|jdt|jd|jdt|jd |jd dddjdtj d  y#1swY.xYw) Ni) rrr)rrHrRrQ rR)(ignore)overaxisrrErrgout) rrandomseedrrandr4errstateisscalarsumr) r8r"dat_1dat_2dat_3dat_4arraysr#datsps rtest_sumzTestCommon1D.test_sumhs2 t67 q! u-CcNE(+{{599;/// 599;7T 2EII4I4HIQ1BCR 0%)))2DE ,+  qbhhrl +,+s )CF,,F6 c~tjd}tjgd}||}tjt d5|j ddddtjtd5|j ddddtjtd 5|j d dddtjt d 5|j d | dddy#1swYxYw#1swYxYw#1swYYxYw#1swYyxYw)Nr2rrr)zaxis must be None, -1 or 0r&rrfTuples are not acceptedrraxis must be an integer?dimensions do not matchrrh)rr4rr+r,r-ro TypeErrorr8r"rir#rus rtest_sum_invalid_paramsz$TestCommon1D.test_sum_invalid_params|shhtnhhy!# ]]:-I J II1I K ]]9,E F II6I "G ]]9,E F II3I G ]]:-F G II1#I &H G K J F F F F G Gs0DD<D'2D3DD$'D03D<ctjgd}||}tj|}tj|}t||yNrx)rrror)r8r"r#rudat_sum datsp_sums rtest_numpy_sumzTestCommon1D.test_numpy_sums=hhy!#&&+FF5M +rctjgd}||}t|j|jtj|jdsJt|jd|jdt|jd|jdt|jd|jdt j td5|jddddt j td5|jddddy#1swY@xYw#1swYyxYw) NrxrfrrErgr&r)rrrmeanrnr+r,r-)r8r"r#rus r test_meanzTestCommon1D.test_meanshhy!# EJJL1{{5::4:0111d+UZZTZ-BCa(%**!**<=b)5::2:+>? ]]:V 4 JJAJ 5 ]]:V 4 JJBJ 5 45 4 4 4sE E, E),E5ctjtjd}tjgdgdgdg}||}t j t d5|jddddt j td 5|jd dddt j td 5|jd dddt j t d 5|jd|dddy#1swYxYw#1swYxYw#1swYYxYw#1swYyxYw)N)rrrx)rrHrR)rarQrbzaxis out of ranger&rrfryrzr{r|r}rrh) rasarrayr4rr+r,r-rr~rs rtest_mean_invalid_paramsz%TestCommon1D.test_mean_invalid_paramssjj&)*hh :z:;# ]]:-@ A JJAJ B ]]9,E F JJFJ #G ]]9,E F JJCJ G ]]:-F G JJA3J 'H G B A F F F F G Gs0*D) D5E E )D25D>E  Ectjgd}||}tD]R}|j|}|j|}t ||t |j |j TyNrxr )rrr rorrr )r8r"r#rur rrs rtest_sum_dtypezTestCommon1D.test_sum_dtypes_hhy!#%EggEg*G  .I GY /   8 &rctjgd}||}tD]R}|j|}|j|}t ||t |j |j Tyr)rrr rrrr )r8r"r#rur dat_mean datsp_means rtest_mean_dtypezTestCommon1D.test_mean_dtypesahhy!#%Exxex,H%0J Hj 1 )9)9 : &rcbtjgd}||}tjdg}tjdg}|j|d|j|t|||jd|d|jd|t||y)NrxrT)rikeepdimsri)rgrirrh)rrrr)r8r"r#rudat_out datsp_outs r test_mean_outzTestCommon1D.test_mean_outshhy!#((A3-HHaSM  Wt, y !+ aWt4 y )+rctjgd}||}tj|}tj|}t||t |j |j yr)rrrrrr )r8r"r#rurrs rtest_numpy_meanzTestCommon1D.test_numpy_meansQhhy!#773<WWU^ *-X^^Z%5%56rcBtjgd}t||j|tjgd}t||j|t||dj|j dyN)r)rr^y?@rrEint16r)rrrr3rrIs rtest_from_arrayzTestCommon1D.test_from_arraysp HHY Yq\))+Q/ HH& 'Yq\))+Q/Yq088:AHHWrc||D]?\}}}|j}d|d<|}||z}t|||jzAy)Nrrrrs r test_raddzTestCommon1D.test_radd>sN!29!= E3 AAaDAAA A O , ">rcV||D] \}}}|tjdk(r!t||z gdt||z gdtd|z j| |gdd}t||z ||jz t||z |j|z t|j|z |j|z t||jz ||jz t|dd|z |dd|z #y)Nr)rrrrr)rrHrr)rrr)rr rr3)r8r"r$r r#rurJs r test_rsubzTestCommon1D.test_rsubFs!29!= E3(( #+ 5 %#+ 5 !e),,. 5-s3A #'C!))+$5 6 !c'AIIK#$5 6 u,aiikC.? @ ,cAIIK.? @ Ra5#bq'C- 8!">rctjgd}||}tjgdgj}t||z|j |z|tjgdzj dk(sJ|tjgdzdk(sJ|tjgdzj dk(sJ|tjdgdgdggzj dk(sJt |tgdgjztjsJ|tjgdgjzj dk(sJtjddgtjgdtjdgdggg}|D]7}tjtd 5|j|ddd9tj|j gd}t|tjgdz|t|dgdgdggz|jy#1swYxYw) N)r)rr)rr)rr rr)rrr)rr^zdimension mismatchr&)rrrWrr3rV isinstancerndarrayr+r,r- __matmul__r)r8r"rJAspcolbad_vecsrX dot_results r test_matveczTestCommon1D.test_matvecYs HH[ !lhh {#%%c 3;;=3#67BHHY''.."444RXXi((B...bhhy))00B666bhhaS1#//66"<<<# { 3 5 55rzzBBBbhh {+---44:::HHaV$bhh|&RSAz1EFq!GF VVCKKM95 bhhy11:>sQC!o-z||<GFs II c||}tgd|ztjgd|jtjgdg}t||z||jzy)Nr)rrrr3r)r8r"rMrows r test_rmatveczTestCommon1D.test_rmatvecusU e  q("&&qyy{*KLhh ~&aqyy{!23rc|tjgfD]l}||}t|j|t|j j|t|j |j nyr)rrrr3 transposer )r8r"rrJBs rtest_transposezTestCommon1D.test_transpose{s]"&A! A a ( ..0! 4 !'' * 'rcp||D].\}}}||z}t|||z||z}t|||z0yr)r)r8r"r$r r#rusum1sum2s rtest_add_dense_to_sparsez%TestCommon1D.test_add_dense_to_sparsesF!29!= E3;D sSy )3;D sSy ) ">rctjd}||}|jdvr!t||D]\}}t ||yy)NrR)coodiabsr)rarangeformatzipr)r8r"rrJrXrYs r test_iteratorzTestCommon1D.test_iteratorsH IIaL aL 880 0Aq 1Q"" 1rctjgd}||}|jdJt|j gd|jdt|j gdy)N)rrrr^r2)rrr)rR)rrrrr)rrresizerr3rs r test_resizezTestCommon1D.test_resizesX HH\ " aLxx~%%%QYY[), QYY[/2rN)#__name__ __module__ __qualname____doc__r9r<r@rCrKrNr\r_rvrrrrrrrr sup_complexrrrrrrrrrrrrrrrrr0r0/s?9 4.$ /,( ',   ( 9 ; ,7OO   O O,0\ 0-9&=84 +*#3rr0ceZdZdZdZy) TestGetSet1Dctjgd}||}|jd}t| |D]}t ||||dD]=}t j ttfd5|j|ddd?|ddk(sJt j td5|jddddy#1swYxYw#1swYyxYw) N)r^rrr)rrHindex value out of boundsr&)rr^)r^) rrrVrangerr+r, IndexErrorr~ __getitem__)r8r"rrJNjijs rtest_getelementzTestGetSet1D.test_getelements HHY  aL GGAJr1A 1qt $BY'/J b!w!|| ]]:-H I MM$ J IJ Is9C:C!C !C*ctj}|d|}tjj5}|j t d|d|d<|d|d<|d|d<|d |d <d |d <|d|d <|d |d<dD]8}t jtd5|j|dddd: dddy#1swYNxYw#1swYyxYw)NrTrzChanging the sparsity structurerrrg"@rQrrbrR)i)r) i)r)rr&g^@) rfloat64testingsuppress_warningsfilterr r+r,r __setitem__)r8r"r rJsuprs rtest_setelementzTestGetSet1D.test_setelements  e5 ) ZZ ) ) +s JJ.0Q R8AaD8AaD:AaD!HAbEAaD1XAcF!HAbE-]]:5PQMM"e,RQ., +RQ, +s$A>C*7C C*C' #C**C3N)rrrrrrrrrrs  (-rr)&rr+numpyr numpy.testingrr scipy.sparserrrrr r r r scipy.sparse._sputilsr rscipy._lib._utilrrrrrrint64r complex128rfixturerr$mark parametrizer.r0rrrrrs,0 7;+jj**, >"I . xxR]]3 ''y)Y &RS T j1i3i32i3X y)&<=%-%->%-r