o ֭j< @sdZddlZddlZddlmZmZddlmZm Z m Z m Z m Z m Z mZddlmZmZddlmZejZeee e egZejejejgZejddZejd d Zej !d ee e e gd d Z"ej !d e eee e e gddZ#ej !d eGdddZ$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)ComplexWarningcCstgddS)N)rrd)nparrayrr]/var/www/html/CropPilot/venv/lib/python3.10/site-packages/scipy/sparse/tests/test_common1d.pydat1dsrcs$fddtDfddtDS)Ncsi|]}||qSr)astype).0dtyperrr sz%datsp_math_dtypes..cs$i|]fddDqS)csg|] \}}|||fqSrr)rrdat spcreatorrr !sz0datsp_math_dtypes...)items)r) dat_dtypesrrr s) math_dtypes spcreatorsrr)rr rdatsp_math_dtypess r#rcCs@tjtdd|gdWddS1swYdS)Nzarrays don't support 1D inputmatchrrrpytestraises ValueErrorrrrrtest_no_1d_support_in_init(s"r,cCsBtjtdd|tdWddS1swYdS)Nzarrays don't.*support 3Dr$)rr')r)r*r+ronesrrrrtest_no_nd_support_in_init0s"r/c@s,eZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ ddZddZddZddZddZd d!Zd"d#Zejjed$d%Zejjed&d'Zejjed(d)Zd*d+Zd,d-Zd.d/Zd0d1Zd2d3Z d4d5Z!d6d7Z"d8d9Z#d:d;Z$dd?Z&d@S)A TestCommon1Dz5test common functionality shared by 1D sparse formatscCs>t|dtdt|djdt|dddS)Nrrr)rtoarrayrzerosnnz count_nonzeroselfrrrrtest_create_empty=szTestCommon1D.test_create_emptycCs<tjtdd |dWddS1swYdS)Nzelements cannot be negativer$)r(r6rrrtest_invalid_shapesBs "z TestCommon1D.test_invalid_shapescCt||dSN)reprr7rrrrr test_reprFzTestCommon1D.test_reprcCr;r<)strr>rrrtest_strIr@zTestCommon1D.test_strcCs*tgdd}t| || dS)N) rrrrrrrrrrrrr2r7rArrrtest_negLszTestCommon1D.test_negcCs|gd}|jdks JdS)Nr&r)ndimrHrrrtest_1d_supported_initPs z#TestCommon1D.test_1d_supported_initcCs|gd}|dd}gdg}t||||}|d}t||d|d}|j|jks7J|jd}t||ddS)N rrrrrrr9rrrr rrP)reshaperr2shapeT)r7rxydesiredy2rrr$test_reshape_1d_tofrom_row_or_columnTs      z1TestCommon1D.test_reshape_1d_tofrom_row_or_columncCsv|gd}|d}gdgdgdgdg}t|||d}||us+J|d}t||dS) NrM)r-r)rrrNrrr)rr9r)rrrOrQrP)rRrr2)r7rrUrVrWrrr test_reshapefs     zTestCommon1D.test_reshapec Cstjdtgd}tjd}tg}td}||||g}|D]U}||}tjddAt|s;Jt ||t |jdd|jddt |jdd|jddt |jd d|jd dWdn1suwYq%|jdtd d dS) Ni) rrr'rrFrOrN rO)(ignore)overaxisrrCrrbout) rrandomseedrrandr3errstateisscalarsumr) r7rdat_1dat_2dat_3dat_4arraysrdatsprrrtest_sumrs"     zTestCommon1D.test_sumcCstd}tgd}||}tjtdd|jddWdn1s(wYtjtdd|jddWdn1sEwYtjtd d|jd dWdn1sbwYtjtd d|jd |d WddS1swYdS)Nr1rrr'zaxis must be None, -1 or 0r$rraTuples are not acceptedrraxis must be an integer?"output parameter.*wrong.*dimensionrrc)rr3rr)r*r+rj TypeErrorr7rrdrrprrrtest_sum_invalid_paramss "z$TestCommon1D.test_sum_invalid_paramscCs8tgd}||}t|}t|}t||dSNrr)rrrjr)r7rrrpdat_sum datsp_sumrrrtest_numpy_sums   zTestCommon1D.test_numpy_sumcCstgd}||}t||t|jddsJt|jdd|jddt|jdd|jddt|jdd|jddtjtdd|jddWdn1s^wYtjtdd|jddWddS1s|wYdS) NrrrarrCrbr$r)rrrmeanrir)r*r+)r7rrrprrr test_means"zTestCommon1D.test_meancCs$ttd}tgdgdgdg}||}tjtdd|jddWdn1s2wYtjtd d|jd dWdn1sOwYtjtd d|jd dWdn1slwYtjtd d|jd|dWddS1swYdS)N)rrrr)rrFrO)r\rNr]zaxis out of ranger$rrarsrtrurvrwrrc) rasarrayr3rr)r*r+rrxryrrrtest_mean_invalid_paramss"z%TestCommon1D.test_mean_invalid_paramscCTtgd}||}tD]}|j|d}|j|d}t||t|j|jq dSNrrr)rrr rjrrr)r7rrrprr|r}rrrtest_sum_dtype   zTestCommon1D.test_sum_dtypecCrr)rrr rrrr)r7rrrprdat_mean datsp_meanrrrtest_mean_dtyperzTestCommon1D.test_mean_dtypecCstgd}||}td}td}|j|d|j|dt|||jd|d|jd|dt||tjtdd|jtdgdWdn1sUwYtjtdd|jtdggdWddS1sxwYdS)Nrrrrdrczoutput parameter.*dimensionr$)rrrrr)r*r+)r7rrrpdat_out datsp_outrrr test_mean_outs       "zTestCommon1D.test_mean_outcCsFtgd}||}t|}t|}t||t|j|jdSr{)rrrrrr)r7rrrprrrrrtest_numpy_means    zTestCommon1D.test_numpy_meancCs`tgd}t|||tgd}t|||t||dd|ddSN)r'rr-y?@rrCint16r)rrrr2rrHrrrtest_from_arrays  zTestCommon1D.test_from_arraycCs`gd}t|||gd}t||t|t||ddt|ddSr)rr2rrrrHrrrtest_from_listszTestCommon1D.test_from_listcCstgd}t|}t|||||}t|||tgd}t|}t|||t||dd|d||}t|||t||dd|ddS)N)rrrrrr)rrrrr2rr7rDSrrrtest_from_sparses zTestCommon1D.test_from_sparsec Cs`||}|}t|||jj|jjksJ|jdd}t|||jjs(J|jjs.J|jdd}t|||jjs?J|jjsEJtj|j|jd}|j|dt||d|d<|j|dt||t gd}t ||}t ||}t||t gd} t || }t || }t||||t d} | t } t| | dS) NC)orderFrr?.)r@@g@) r2rflags c_contiguous f_contiguousrr3rSrrdotboolr) r7rrrpchkrdadense_dot_densecheckbspboolarrboolrrr test_toarrays:                   zTestCommon1D.test_toarrayc Csl||D]/\}}}|}d|d<|}||}t|||||dd}t|||dqdS)Nrrrcopyrr2 r7rr#rrrprrcrrrtest_addCszTestCommon1D.test_addc CsF||D]\}}}|}d|d<|}||}t|||qdS)Nrrrrrrr test_raddPszTestCommon1D.test_raddcCs||D]r\}}}|tdkrqt||gdt||gdtd|| |gddd}t||||t||||t||||t||||t|dd||dd|qdS)Nr)rrrrr)rrFrr'rrr)rrrr2)r7rr#rrrprIrrr test_rsubXs$zTestCommon1D.test_rsubcCs"tgdgdgdg}tgd}||}||}t||||t||||t||||t||||t||||t||||t||||t||||t||||t||||t||||t||||dS)Nr'rrrZrrrrrrG)r7rrIvAspvsprrrtest_matmul_basicks zTestCommon1D.test_matmul_basicc Cstgd}||}tgdgj}t|||||tgdjdks-J|tgddks:J|tgdjdksHJ|tdgdgdggjdksZJt|tgdgjtjsjJtddgtgd tdgdggg}|D]}t j t d d  ||Wdn1swYqt |gd}t|tgd|t|dgdgdgg|jdS) Nr)rr'rr rr'r)rrr'rr-zdimension mismatchr$) rrrTrr2rS isinstancer ndarrayr)r*r+r)r7rrIrcolbad_vecsrU dot_resultrrr test_matvecs$$ ,  zTestCommon1D.test_matveccCsT||}tgd|tgd|tgdg}t||||dS)Nr)rrrr2r)r7rrMrowrrr test_rmatvecs"zTestCommon1D.test_rmatveccCsN|tgfD]}||}t||t||t|j|jqdSr<)rrrr2 transposer)r7rrrIBrrrtest_transposes zTestCommon1D.test_transposecCsD||D]\}}}||}t|||||}t|||qdSr<)r)r7rr#rrrpsum1sum2rrrtest_add_dense_to_sparses z%TestCommon1D.test_add_dense_to_sparsecCsBtd}||}|jdvrt||D] \}}t||qdSdS)NrO)coodiabsr)rarangeformatzipr)r7rrrIrUrVrrr test_iterators   zTestCommon1D.test_iteratorcCsZtgd}||}|ddusJt|gd|dt|gddS)N)rrrr-r1r)rO)rrrrr)rrresizerr2rrrr test_resizes  zTestCommon1D.test_resizeN)'__name__ __module__ __qualname____doc__r8r:r?rBrJrLrYr[rqrzr~rrrrrrr)mark thread_unsafe sup_complexrrrrrrrrrrrrrrrrrrr09sN         .  r0)%rr)numpyr numpy.testingrr scipy.sparserrrrrr r scipy.sparse._sputilsr r scipy._lib._utilr testingsuppress_warningsrfilterr"int64float64 complex128r!fixturerr#r parametrizer,r/r0rrrrs.$