_Mh< 8dZddlZddlZddlmZmZddlmZm Z m Z m Z m Z m Z mZddlmZmZddlmZejZeee e egZejejejgZejdZejdZej !d ee e e gd Z"ej !d e eee e e gd Z#ej !d eGd d Z$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_array)supported_dtypesmatrix)ComplexWarningc0tjgddS)N)rrd)nparray`/var/www/html/test/jupyter/venv/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 ValueErrorr"s rtest_no_1d_support_in_initr3(s z)H I I I   ,,,                  s 7;;ctjtd5|tjdddddS#1swxYwYdS)Nzarrays don't.*support 3Dr*)rr-)r/r0r1ronesr2s rtest_no_nd_support_in_initr70s z)C D D D&& "')$$%%%&&&&&&&&&&&&&&&&&&sAA A cheZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZdZejjedZejjedZejjedZdZdZdZdZdZ dZ!dZ"dZ#dZ$dZ%d Z&d!S)" TestCommon1Dz5test common functionality shared by 1D sparse formatsct|dtjdt|djdt|dddS)Nrrr)rtoarrayrzerosnnz count_nonzeroselfr"s rtest_create_emptyzTestCommon1D.test_create_empty=ssYYt__,,.. <<<YYt__(!,,,YYt__2244a88888rctjtd5|dddddS#1swxYwYdS)Nzelements cannot be negativer*)r.r@s rtest_invalid_shapesz TestCommon1D.test_invalid_shapesBs ]:-J K K K   Ie                     s 599c6t||dSN)reprrAr"rs r test_reprzTestCommon1D.test_reprFs  YYu  rc6t||dSrG)strrIs rtest_strzTestCommon1D.test_strIs  IIe  rctjgdd}t| || dS)N) rrrrrrrrrrrrr<rAr"As rtest_negzTestCommon1D.test_negLsH H===s C CaR99Q<<-002233333rc:|gd}|jdksJdS)Nr,r)ndimrTs rtest_1d_supported_initz#TestCommon1D.test_1d_supported_initPs( Illl # #v{{{{{{rc|gd}|dd}gdg}t||||}|d}t||d|d}|j|jksJ|jd}t||ddS)N rrrrrrrDrrrr rr^)reshaperr<shapeT)rAr"xydesiredy2s r$test_reshape_1d_tofrom_row_or_columnz1TestCommon1D.test_reshape_1d_tofrom_row_or_columnTs I;;; < < IIa  8889QYY[['*** Ig   IIbMMQYY[['!*--- YYu  w"("""" CKKOOQYY[['!*-----rcr|gd}|d}gdgdgdgdg}t|||d}||usJ|d}t||dS) Nr[)r5r)rrr\rrr)rrDr)rrr]r_r^)r`rr<)rAr"rcrdres r test_reshapezTestCommon1D.test_reshapefs 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-rrRr]r\ r])(ignore)overaxisrrOrrrout) rrandomseedrrandr=errstateisscalarsumr) rAr"dat_1dat_2dat_3dat_4arraysr!datsps rtest_sumzTestCommon1D.test_sumrs 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)Nr;rrr-zaxis must be None, -1 or 0r*rrqTuples are not acceptedrraxis must be an integer?"output parameter.*wrong.*dimensionrrs)rr=rr/r0r1rz TypeErrorrAr"rtr!rs rtest_sum_invalid_paramsz$TestCommon1D.test_sum_invalid_paramss<htnnhyyy!! # ]:-I J J J   II1I                   ]9,E F F F # # II6I " " " # # # # # # # # # # # # # # # ]9,E F F F  II3I                  ]:-Q R R R ' ' II1#I & & & ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' 'sHA44A8;A8B==CC#DD  D ,EEEctjgd}||}tj|}tj|}t||dSNr)rrrzr)rAr"r!rdat_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) NrrqrrOrrr*r)rrrmeanryr/r0r1)rAr"r!rs 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)rrRr])rlr\rmzaxis out of ranger*rrqrrrrrrrs) rasarrayr=rr/r0r1rrrs 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  ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! ]:-Q R R R ( ( JJA3J ' ' ' ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( (sH*B  BB3CCC<DD#&D#E**E.1E.ctjgd}||}tD]X}||}||}t ||t |j|jYdSNrr)rrr rzrrr)rAr"r!rrrrs 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)rAr"r!rrdat_mean datsp_means rtest_mean_dtypezTestCommon1D.test_mean_dtypeshyyy!! #% ; ;Exxex,,H%00J Hj 1 1 1 )9 : : : :  ; ;rctjgd}||}tjd}tjd}||||t|||d||d|t||t jt d5|tjdgdddn #1swxYwYt jt d5|tjdggddddS#1swxYwYdS)Nrrrtrszoutput parameter.*dimensionr*)rrrrr/r0r1)rAr"r!rdat_out datsp_outs r test_mean_outzTestCommon1D.test_mean_outshyyy!! #(1++HQKK  W y !!!+++ aW%%% y )))+++ ]:-J K K K * * JJ28QC==J ) ) ) * * * * * * * * * * * * * * * ]:-J K K K , , JJ28aSE??J + + + , , , , , , , , , , , , , , , , , ,s$*DDD;+E33E7:E7ctjgd}||}tj|}tj|}t||t |j|jdSr)rrrrrr)rAr"r!rrrs rtest_numpy_meanzTestCommon1D.test_numpy_meansghyyy!! #73<<WU^^ *---X^Z%566666rctjgd}t|||tjgd}t|||t||d|ddSN)r-rr5y?@rrOint16r)rrrr<rrTs rtest_from_arrayzTestCommon1D.test_from_arrays HYYY  YYq\\))++Q/// H&&& ' 'YYq\\))++Q///YYq00088::AHHW>>># { 3 3 55rzBBBBBHaV$$bh|||&<&R>RS  Az1EFFF  a                VCKKMM99955 bhyyy111:>>>sQC!o-z|<<<<r s00 77777777;:::::::++++++j**,, >"""I . xR]3 '''y)Y &RSS  TS )Y 9iS&&& j11E3E3E3E3E3E3E321E3E3E3r