a CCCf<@sdZddlZddlZddlZddlmZmZddl m Z ej Z e e ejjejjgZejejejgZejddZejddZejd eGd d d Zejd ejjgGd d d ZdS)z*Test of 1D aspects of sparse array classesN)supported_dtypesmatrix)ComplexWarningcCstgddS)N)rrd)nparrayr r \/var/www/html/django/DPS/env/lib/python3.9/site-packages/scipy/sparse/tests/test_common1d.pydat1dsr cs$fddtDfddtDS)Ncsi|]}||qSr )astype).0dtyper r r z%datsp_math_dtypes..cs$i|]fddDqS)csg|]\}}|||fqSr r )rrdatspr r rz0datsp_math_dtypes...)items)r) dat_dtypesrr rs) math_dtypes spcreatorsrr )r rr datsp_math_dtypess r spcreatorc@seZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ ddZddZddZddZddZd d!Zed"d#Zed$d%Zed&d'Zd(d)Zd*d+Zd,d-Zd.d/Zd0d1Zd2d3Zd4d5Zd6d7Zd8d9Z d:d;Z!drrr )r#rrArBrCr r r test_reshapeKs     zTestCommon1D.test_reshapec Cs(tjdtgd}tjd}tg}td}||||g}|D]}||}tjddt|svJt ||sJt |jdd|jddsJt |jdd|jddsJt |jd d|jd dsJWdqJ1s0YqJ|jdtd d dS) Ni) rrrr5r;r: r;)(ignore)Zoveraxisrr2r rNout) rrandomseedr Zrandr!Zerrstateisscalarsumallclose) r#rZdat_1Zdat_2Zdat_3Zdat_4Zarraysrdatspr r r test_sumWs       BzTestCommon1D.test_sumcCstd}tgd}||}tjtdd|jddWdn1sP0Ytjtdd|jddWdn1s0Ytjtd d|jd dWdn1s0Ytjtd d|jd |d Wdn1s0YdS)NrrrrHzaxis must be None, -1 or 0r%rrMTuples are not acceptedrraxis must be an integer?dimensions do not matchrrO)rr!r r(r)r*rT TypeErrorr#rrPrrVr r r test_sum_invalid_paramsks ***z$TestCommon1D.test_sum_invalid_paramscCs>tgd}||}t|}t|}t||s:JdSNrX)rr rTrU)r#rrrVdat_sum datsp_sumr r r test_numpy_sumys   zTestCommon1D.test_numpy_sumcCstgd}||}t||s.Jt|jddsDJt|jdd|jddsdJt|jdd|jddsJt|jdd|jddsJtjtdd|jddWdn1s0Ytjtdd|jddWdn1s0YdS) NrXrMrr2rNr%r)rr rUmeanrSr(r)r*)r#rrrVr r r test_means   *zTestCommon1D.test_meancCsjttd}tgdgdgdg}|jdkrptjtdd||Wdn1sb0YdS||}tjtdd|jd d Wdn1s0Ytjt d d|jd d Wdn1s0Ytjt d d|jdd Wdn1s0Ytjtdd|jd|dWdn1s\0YdS)N)rrrX)rr5r;)rIr:rJuniZzqr%zaxis out of rangerrMrYrZr[r\r]rrO) rZasarrayr!r _formatr(r)r*rfr^r_r r r test_mean_invalid_paramss &**,z%TestCommon1D.test_mean_invalid_paramscCs`tgd}||}tD]@}|j|d}|j|d}t||sFJt|j|jsJqdSNrXr)rr rrTrUrr)r#rrrVrrbrcr r r test_sum_dtypes  zTestCommon1D.test_sum_dtypecCs`tgd}||}tD]@}|j|d}|j|d}t||sFJt|j|jsJqdSrk)rr rrfrUrr)r#rrrVrdat_mean datsp_meanr r r test_mean_dtypes  zTestCommon1D.test_mean_dtypecCstgd}||}tdg}tdg}|j|dd|j|dt||sXJ|jd|dd|jd|dt||sJdS)NrXrT)rPkeepdimsrP)rNrPrqrO)rr rfrU)r#rrrVZdat_outZ datsp_outr r r test_mean_outs   zTestCommon1D.test_mean_outcCsRtgd}||}t|}t|}t||s:Jt|j|jsNJdSra)rr rfrUrr)r#rrrVrnror r r test_numpy_means   zTestCommon1D.test_numpy_meancCsrtgd}t|||s&Jtgd}t|||sLJt||dd|dsnJdSN)rHrrFy?@rr2int16rl)rr rr r r6r r r test_from_arrays zTestCommon1D.test_from_arraycCsrgd}t|||s Jgd}t||t|sFJt||ddt|dsnJdSru)rrr r r r6r r r test_from_listszTestCommon1D.test_from_listcCstgd}tj|}t|||s2J||}t|||sRJtgd}tj|}t|||sJt||dd|dsJ||}t|||sJt||dd|dsJdS)N)rrrrvrwrl)rr rsparse coo_arrayrr r r#rDSr r r test_from_sparses  "zTestCommon1D.test_from_sparsec Cs||}|}t||s J|jj|jjks4J|jdd}t||sPJ|jjs\J|jjshJ|jdd}t||sJ|jjsJ|jjsJtj|j|jd}|j|dt||sJd|d<|j|dt||sJt gd}t ||}t ||}t||s*Jt gd} t || }t || }t||sfJ||t d} | t } t| | sJdS) NC)orderFrlrr?.)r@@g@) r rrflags c_contiguous f_contiguousr!r?rr dotboolr ) r#rr rVZchkrPaZdense_dot_densecheckbZspboolZarrboolr r r test_toarrays:            zTestCommon1D.test_toarrayc Csx||D]j\}}}|}d|d<|}||}t|||sFJ||dd}t|||dsJqdS)Nrrrcopyrrr r#rrrrrVrrcr r r test_add%szTestCommon1D.test_addc CsL||D]>\}}}|}d|d<|}||}t|||sJqdS)Nrrrrr r r test_radd2s zTestCommon1D.test_raddcCs$||D]\}}}|tdkr$qt||gds}tj t dd| |dWdq1s0YqWdn1s0YdS)Nr=rlz?Changing the sparsity structure of a cs[cr]_matrix is expensiverrrg"@r:rerJr;)i)r) i)r)rr%g^@) rfloat64testingsuppress_warningsfilterrrzZSparseEfficiencyWarningr(r)r __setitem__)r#rrr7suprr r r test_setelements"        zTestGetSet1D.test_setelementN)rrrrrr r r r rsr)rr(numpyrZscipyrZscipy.sparse._sputilsrrZscipy._lib._utilrrrrrrzr{Z dok_arrayrZint64rZ complex128rZfixturer rmarkZ parametrizerrr r r r s&      o