L iPdZddlmZddlmZmZmZddlZddlm Z GddZ y)z)test sparse matrix construction functions) assert_equal) csr_matrix csr_arraysparrayN)_extractc*eZdZdZdZdZdZdZy) TestExtractctddggtddggtddggtdgdggtdgdggtdgdggtddgddggtddgddggtddgddggtddgddggtgdgdgdgtgdgdgdgjg |_y) Nr)r r rrr )rr)rr r)rTcases)selfs e/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/scipy/sparse/tests/test_extract.py setup_methodzTestExtract.setup_method s !ug  !ug  !ug  sA3i sA3i sA3i !uaUm $ !uaUm $ !uaUm $ !uaUm $ {;{; < {;{; < > >   c|jD]^}tj|\}}}t|||ff|j}t |j |j `y)N)shape)rrfindrrrtoarray)rAIJVBs r test_findzTestExtract.test_findsW 3AMM!$EAa1aU)1773A aiik 2 3rc |jD]]}|j}dD]F}ttj||jt j||H_yN)rr r r )k)rrrrtrilnprrr"r)s r test_trilzTestExtract.test_tril![ MA A' MX]]1q199;RWWQ^L M Mrc |jD]]}|j}dD]F}ttj||jt j||H_yr%)rrrrtriur+r,s r test_triuzTestExtract.test_triu'r.rcd|jD]}ttj|tsJttj |tsJt |}ttj|trJttj |tsJy)N)r isinstancerr*rr0r)rrMs rtest_array_vs_matrixz TestExtract.test_array_vs_matrix-s =AhmmA.8 88hmmA.8 881 A!(--"2G< <<!(--"2G< <<  =rN)__name__ __module__ __qualname__rr#r-r1r5rrr r s 3 M M =rr ) __doc__ numpy.testingr scipy.sparserrrnumpyr+rr r9rrr>s!/&77!)=)=r