Ugu<HdZddlZddlZddlmZmZddlmZm Z m Z m Z m Z m Z mZmZddlmZmZddlmZejZeee e egZejejejgZejdZejdZ ej!"d ee e e gd Z#ej!"d eGd d Z$ej!"d e egGd dZ%dS)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)ComplexWarningc0tjgddS)N)rrd)nparray_/var/www/surfInsights/venv3-11/lib/python3.11/site-packages/scipy/sparse/tests/test_common1d.pydat1drs 8LLL# & &&rcRfdtDfdtDS)Nc<i|]}||Sr)astype).0dtypers r z%datsp_math_dtypes..s'FFF%e,,FFFrcTi|]#fdD$S)c2g|]\}}|||fSrr)rrdat spcreators r z0datsp_math_dtypes...!s,VVVZUCUC30VVVr)items)rr# dat_dtypess @rrz%datsp_math_dtypes.. sM     VVVV:CSCSCUCUVVV   r) math_dtypes spcreators)rr&s`@rdatsp_math_dtypesr)sHFFFF+FFFJ    #   rr#ctjtd5|gdddddS#1swxYwYdS)Nzarrays don't support 1D inputmatchrrrpytestraises ValueError)r#s rtest_no_1d_support_in_initr3(s z)H I I I   ,,,                  s 7;;ceZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZdZedZedZedZdZdZdZdZdZdZdZdZ dZ!dZ"d S)! TestCommon1Dz5test common functionality shared by 1D sparse formatsct|dtjdt|djdt|dddS)Nrrr)rtoarrayrzerosnnz count_nonzeroselfr#s rtest_create_emptyzTestCommon1D.test_create_empty3ssYYt__,,.. <<<YYt__(!,,,YYt__2244a88888rctjtd5|dddddS#1swxYwYdS)Nzelements cannot be negativer+)r/r<s rtest_invalid_shapesz TestCommon1D.test_invalid_shapes8s ]:-J K K K   Ie                     s 599c6t||dSN)reprr=r#rs r test_reprzTestCommon1D.test_repr<s  YYu  rc6t||dSrC)strrEs rtest_strzTestCommon1D.test_str?s  IIe  rctjgdd}t| || dS)N) rrrrrrrrr)rrrr8r=r#As rtest_negzTestCommon1D.test_negBsH H===s C CaR99Q<<-002233333rc:|gd}|jdksJdS)Nr-r)ndimrOs rtest_1d_supported_initz#TestCommon1D.test_1d_supported_initFs( Illl # #v{{{{{{rc|gd}|dd}gdg}t||||}|d}t||d|d}|j|jksJ|jd}t||ddS)N rrrrrrr@rrrr rrY)reshaperr8shapeT)r=r#xydesiredy2s r$test_reshape_1d_tofrom_row_or_columnz1TestCommon1D.test_reshape_1d_tofrom_row_or_columnJs I;;; < < IIa  8889QYY[['*** Ig   IIbMMQYY[['!*--- YYu  w"("""" CKKOOQYY[['!*-----rcr|gd}|d}gdgdgdgdg}t|||d}||usJ|d}t||dS) NrV)r)rrrW)rrr)rr@r)rrrXrZrY)r[rr8)r=r#r^r_r`s r test_reshapezTestCommon1D.test_reshape\s I;;; < < IIf  99iiiYYY?QYY[['*** IIe  Avvvv IIbMMQYY[[!))++.....rctjdtjgd}tjd}tjg}tjd}||||g}|D]?}||}tjd5tj|sJt||t|d|dt|d|dt|d |d dddn #1swxYwYA|dtjd  dS) Ni) rrr.rrNrXrW rX)(ignore)overaxisrrKrrmout) rrandomseedrrandr9errstateisscalarsumr) r=r#dat_1dat_2dat_3dat_4arraysr"datsps rtest_sumzTestCommon1D.test_sumhs t66677 q!! u- F FCIcNNE(+++ F F{599;;///// 599;;777T 2 2EII4I4H4HIIIQ1B1BCCCR 0 0%)))2D2DEEE  F F F F F F F F F F F F F F F  qbhrll +++++s(DF99F= F= ctjd}tjgd}||}tjt d5|ddddn #1swxYwYtjtd5|ddddn #1swxYwYtjtd 5|d dddn #1swxYwYtjt d 5|d | ddddS#1swxYwYdS)Nr7rrr.zaxis must be None, -1 or 0r+rrlTuples are not acceptedrraxis must be an integer?dimensions do not matchrrn)rr9rr0r1r2ru TypeErrorr=r#ror"r{s rtest_sum_invalid_paramsz$TestCommon1D.test_sum_invalid_params|s<htnnhyyy!! # ]:-I J J J   II1I                   ]9,E F F F # # II6I " " " # # # # # # # # # # # # # # # ]9,E F F F  II3I                  ]:-F G G G ' ' II1#I & & & ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' 'sHA44A8;A8B==CC#DD  D ,EEEctjgd}||}tj|}tj|}t||dSNr~)rrrur)r=r#r"r{dat_sum datsp_sums rtest_numpy_sumzTestCommon1D.test_numpy_sumsRhyyy!! #&++F5MM +++++rcztjgd}||}t||tj|dsJt|d|dt|d|dt|d|dt jtd5|ddddn #1swxYwYt jtd5|dddddS#1swxYwYdS) Nr~rlrrKrmr+r)rrrmeanrtr0r1r2)r=r#r"r{s r test_meanzTestCommon1D.test_meanshyyy!! # EJJLL111{5::4:0011111d++UZZTZ-B-BCCCa((%**!**<*<===b))5::2:+>+>??? ]:V 4 4 4   JJAJ                   ]:V 4 4 4  JJBJ                     s$E&&E*-E* F00F47F4ctjtjd}tjgdgdgdg}||}t jt d5|ddddn #1swxYwYt jtd 5|d dddn #1swxYwYt jtd 5|d dddn #1swxYwYt jt d 5|d|ddddS#1swxYwYdS)N)rrr~)rrNrX)rgrWrhzaxis out of ranger+rrlrrrrrrrn) rasarrayr9rr0r1r2rrrs rtest_mean_invalid_paramsz%TestCommon1D.test_mean_invalid_paramssWj&))**h :::zzz:;; # ]:-@ A A A   JJAJ                   ]9,E F F F $ $ JJFJ # # # $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ ]9,E F F F ! ! JJCJ  ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! ]:-F G G G ( ( JJA3J ' ' ' ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( (sH*B  BB3CCC<DD#&D#E**E.1E.ctjgd}||}tD]X}||}||}t ||t |j|jYdSNr~r)rrr rurrr)r=r#r"r{rrrs rtest_sum_dtypezTestCommon1D.test_sum_dtypeshyyy!! #% 9 9EggEg**G  ..I GY / / /   8 8 8 8  9 9rctjgd}||}tD]X}||}||}t ||t |j|jYdSr)rrr rrrr)r=r#r"r{rdat_mean datsp_means rtest_mean_dtypezTestCommon1D.test_mean_dtypeshyyy!! #% ; ;Exxex,,H%00J Hj 1 1 1 )9 : : : :  ; ;rctjgd}||}tjdg}tjdg}||d||t|||d|d|d|t||dS)Nr~rT)rokeepdimsro)rmrorrn)rrrr)r=r#r"r{dat_out datsp_outs r test_mean_outzTestCommon1D.test_mean_outshyyy!! #(A3--HaSMM  Wt,,, y !!!+++ aWt444 y )))+++++rctjgd}||}tj|}tj|}t||t |j|jdSr)rrrrrr)r=r#r"r{rrs rtest_numpy_meanzTestCommon1D.test_numpy_meansghyyy!! #73<<WU^^ *---X^Z%566666rctjgd}t|||tjgd}t|||t||d|ddSN)r.rrdy?@rrKint16r)rrrr8rrOs rtest_from_arrayzTestCommon1D.test_from_arrays HYYY  YYq\\))++Q/// H&&& ' 'YYq\\))++Q///YYq00088::AHHWsj!29!= - - E3 AAaDAAA A O , , , ,  - -rc||D]i\}}}|tjdkr t||z gdt||z gdtd|z | |gdd}t||z ||z t||z ||z t||z ||z t||z ||z t|dd|z |dd|z kdS)Nr)rrrrr)rrNrr.rrr)rrrr8)r=r#r)rr"r{rPs r test_rsubzTestCommon1D.test_rsubFsl!29!= 9 9 E3(((( #+ 5 5 5 %#+ 5 5 5 !e),,.. 5 5 5 ---s333A #'C!))++$5 6 6 6 !c'AIIKK#$5 6 6 6 u,aiikkC.? @ @ @ ,cAIIKK.? @ @ @ RaR5#bqb'C- 8 8 8 8! 9 9rctjgd}||}tjgdgj}t||z||z|tjgdzjdksJ|tjgdzdksJ|tjgdzjdksJ|tjdgdgdggzjdksJt |tgdgjztjsJ|tjgdgjzjdksJtjddgtjgdtjdgdggg}|D]J}tj td 5| |dddn #1swxYwYKtj |gd}t|tjgdz|t|dgdgdggz|jdS) N)r.rr)rr.rr rr.rrr.rrdzdimension mismatchr+)rrr]rr8r\ isinstancerndarrayr0r1r2 __matmul__r)r=r#rPAspcolbad_vecsr^ dot_results r test_matveczTestCommon1D.test_matvecYs H[[[ ! !illh {##%c 3;;==3#6777BHYYY'''."4444RXiii(((B....bhyyy)))0B6666bhaS1#///6"<<<<# { 3 3 55rzBBBBBbh {++--4::::HaV$$bh|||&<&R>RS " "Az1EFFF " "q!!! " " " " " " " " " " " " " " " VCKKMM99955 bhyyy111:>>>sQC!o-z|<<<<rArFrIrQrTrbrer|rrrrrrrr sup_complexrrrrrrrrrrrrrrrrr5r5/s??999 444...$ / / /,,,( ' ' ',,,     ( ( ( 9 9 9 ; ; ; , , ,777OO[O  [  O O[ O,0,0,0\ 0 0 0---999&===8444 +++***###33333rr5ceZdZdZdZdS) TestGetSet1Dctjgd}||}|jd}t| |D]}t ||||dD]Q}t jttfd5| |dddn #1swxYwYR|ddksJt jtd5| dddddS#1swxYwYdS) N)rdrrr)rrNindex value out of boundsr+)rrd)rd) rrr\rangerr0r1 IndexErrorr __getitem__)r=r#rrPNjijs rtest_getelementzTestGetSet1D.test_getelements HYYY   IaLL GAJr1 % %A 1qt $ $ $ $ " "BY'/J " " b!!! " " " " " " " " " " " " " " " w!|||| ]:-H I I I  MM$                     s$B''B+ .B+ C??DDc,tj}|d|}tj5}|t d|d|d<|d|d<|d|d<|d |d <d |d <|d|d <|d |d<dD]K}t jtd5| |ddddn #1swxYwYL ddddS#1swxYwYdS)NrZrzChanging the sparsity structurerrrg"@rWrrhrX)i)r) i)r)rr+g^@) rfloat64testingsuppress_warningsfilterr r0r1r __setitem__)r=r#rrPsuprs rtest_setelementzTestGetSet1D.test_setelements  Ie5 ) ) ) Z ) ) + + -s JJ.0Q R R R588AaD588AaD5::AaDE!HHAbEAaDU1XXAcFE!HHAbE- - -]:5PQQQ--MM"e,,,--------------- - - - - - - - - - - - - - - - - - - -s6BD  C/# D /C3 3D 6C3 7D  D D N)rrrrr rrrrrs2   (-----rr)&rr0numpyr numpy.testingrr scipy.sparserrrrr r r r scipy.sparse._sputilsr rscipy._lib._utilrrrrrr(int64r complex128r'fixturerr)mark parametrizer3r5rrrrrs)00 77777777;:::::::++++++j**,, >"""I . xR]3 '''y)Y &RSS  TS j11i3i3i3i3i3i3i321i3X y)&<==%-%-%-%-%-%-%->=%-%-%-r