o i< @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)nparrayrrZ/home/kim/smarthome/.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+rZonesrrrrtest_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)rtoarrayrzerosZnnzZ count_nonzeroselfrrrrtest_create_empty=szTestCommon1D.test_create_emptycCs<tjtdd |dWddS1swYdS)Nzelements cannot be negativer$)r(r3rrrtest_invalid_shapesBs "z TestCommon1D.test_invalid_shapescCt||dSN)reprr4rrrrr test_reprFzTestCommon1D.test_reprcCr8r9)strr;rrrtest_strIr=zTestCommon1D.test_strcCs*tgdd}t| || dS)N) rrrrrrrrrrrrr1r4rArrrtest_negLszTestCommon1D.test_negcCs|gd}|jdks JdS)Nr&r)ndimrErrrtest_1d_supported_initPs z#TestCommon1D.test_1d_supported_initcCs|gd}|dd}gdg}t||||}|d}t||d|d}|j|jks7J|jd}t||ddS)N rrrrrrr6rrrr rrM)reshaperr1shapeT)r4rxydesiredy2rrr$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) NrJ)r-r)rrrKrrr)rr6r)rrrLrNrM)rOrr1)r4rrRrSrTrrr 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'rrCrLrK rL)(ignore)Zoveraxisrr@rr^out) rrandomseedrZrandr2Zerrstateisscalarsumr) r4rZdat_1Zdat_2Zdat_3Zdat_4Zarraysrdatsprrrtest_sumrs"     zTestCommon1D.test_sumcCstd}tgd}||}tjtdd|jddWdn1s(wYtjtdd|jddWdn1sEwYtjtd d|jd dWdn1sbwYtjtd d|jd |d WddS1swYdS)Nr0rrr'zaxis must be None, -1 or 0r$rr]Tuples are not acceptedrraxis must be an integer?"output parameter.*wrong.*dimensionrr_)rr2rr)r*r+rd TypeErrorr4rr`rrerrrtest_sum_invalid_paramss "z$TestCommon1D.test_sum_invalid_paramscCs8tgd}||}t|}t|}t||dSNrg)rrrdr)r4rrredat_sum datsp_sumrrrtest_numpy_sums   zTestCommon1D.test_numpy_sumcCstgd}||}t||t|jddsJt|jdd|jddt|jdd|jddt|jdd|jddtjtdd|jddWdn1s^wYtjtdd|jddWddS1s|wYdS) Nrgr]rr@r^r$r)rrrmeanrcr)r*r+)r4rrrerrr test_means"zTestCommon1D.test_meancCs$ttd}tgdgdgdg}||}tjtdd|jddWdn1s2wYtjtd d|jd dWdn1sOwYtjtd d|jd dWdn1slwYtjtd d|jd|dWddS1swYdS)N)rrrg)rrCrL)rYrKrZzaxis out of ranger$rr]rhrirjrkrlrr_) rZasarrayr2rr)r*r+rurmrnrrrtest_mean_invalid_paramss"z%TestCommon1D.test_mean_invalid_paramscCTtgd}||}tD]}|j|d}|j|d}t||t|j|jq dSNrgr)rrr rdrrr)r4rrrerrqrrrrrtest_sum_dtype   zTestCommon1D.test_sum_dtypecCrxry)rrr rurrr)r4rrrerdat_mean datsp_meanrrrtest_mean_dtyper|zTestCommon1D.test_mean_dtypecCstgd}||}td}td}|j|d|j|dt|||jd|d|jd|dt||tjtdd|jtdgdWdn1sUwYtjtdd|jtdggdWddS1sxwYdS)Nrgrr`r_zoutput parameter.*dimensionr$)rrrurr)r*r+)r4rrreZdat_outZ datsp_outrrr test_mean_outs       "zTestCommon1D.test_mean_outcCsFtgd}||}t|}t|}t||t|j|jdSrp)rrrurrr)r4rrrer}r~rrrtest_numpy_means    zTestCommon1D.test_numpy_meancCs`tgd}t|||tgd}t|||t||dd|ddSN)r'rr-y?@rr@int16rz)rrrr1rrErrrtest_from_arrays  zTestCommon1D.test_from_arraycCs`gd}t|||gd}t||t|t||ddt|ddSr)rr1rrrrErrrtest_from_listszTestCommon1D.test_from_listcCstgd}t|}t|||||}t|||tgd}t|}t|||t||dd|d||}t|||t||dd|ddS)N)rrrrrrz)rrrrr1rr4rDSrrrtest_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)orderFrzr?.)r@@g@) r1rflags c_contiguous f_contiguousrr2rPrrdotboolr) r4rrreZchkr`aZdense_dot_densecheckbZspboolZarrboolrrr test_toarrays:                   zTestCommon1D.test_toarrayc Csl||D]/\}}}|}d|d<|}||}t|||||dd}t|||dqdS)Nrrrcopyrr1 r4rr#rrrerrcrrrtest_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)rrCrr'rrzr)rrrr1)r4rr#rrrerFrrr test_rsubXs$zTestCommon1D.test_rsubcCs"tgdgdgdg}tgd}||}||}t||||t||||t||||t||||t||||t||||t||||t||||t||||t||||t||||t||||dS)Nr'rrrWrgrrrrD)r4rrFvAspZvsprrrtest_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$) rrrQrr1rP isinstancer Zndarrayr)r*r+r)r4rrFrcolZbad_vecsrRZ dot_resultrrr test_matvecs$$ ,  zTestCommon1D.test_matveccCsT||}tgd|tgd|tgdg}t||||dS)Nr)rrrr1r)r4rrMrowrrr test_rmatvecs"zTestCommon1D.test_rmatveccCsN|tgfD]}||}t||t||t|j|jqdSr9)rrrr1Z transposer)r4rrrFBrrrtest_transposes zTestCommon1D.test_transposecCsD||D]\}}}||}t|||||}t|||qdSr9)r)r4rr#rrreZsum1Zsum2rrrtest_add_dense_to_sparses z%TestCommon1D.test_add_dense_to_sparsecCsBtd}||}|jdvrt||D] \}}t||qdSdS)NrL)ZcooZdiaZbsr)rZarangeformatzipr)r4rrrFrRrSrrr test_iterators   zTestCommon1D.test_iteratorcCsZtgd}||}|ddusJt|gd|dt|gddS)N)rrrr-r0r)rL)rrrrr)rrresizerr1rrrr test_resizes  zTestCommon1D.test_resizeN)'__name__ __module__ __qualname____doc__r5r7r<r?rGrIrVrXrfrorsrvrwr{rrrr)markZ thread_unsafe sup_complexrrrrrrrrrrrrrrrrrrr/9sN         .  r/)%rr)numpyrZ numpy.testingrrZ scipy.sparserrrrrr r Zscipy.sparse._sputilsr r Zscipy._lib._utilr ZtestingZsuppress_warningsrfilterr"Zint64Zfloat64Z complex128r!Zfixturerr#rZ parametrizer,r.r/rrrrs.$