L i"dZddlZddlZddlmZmZddlmZm Z m Z m Z ddlm Z m Z mZmZddlmZdZe ee e gZee gZej*j-deGd d Zej*j-deGd d Zy) z3Test of min-max 1D features of sparse array classesN) assert_equalassert_array_equal) coo_array csr_array csc_array bsr_array) coo_matrix csr_matrix csc_matrix bsr_matrix) isscalarlikecpt|tjs t|r|S|j S)N) isinstancenpndarrayr toarray)as f/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/scipy/sparse/tests/test_minmax1d.pyrrs'!RZZ LO 99; spcreatorc$eZdZdZdZdZdZy)Test_MinMaxMixin1Dctjd}||}t|jdt|j dt| jdt| j dy)Nr)rarangerminmax)selfrDXs r test_minmaxzTest_MinMaxMixin1D.test_minmaxs\ IIaL aLQUUWa QUUWa qbXXZ$qbXXZ#rc:tjd}||}dD]n}tt|j ||j |dtt|j ||j |dpdD]n}t jtd5|j |dddt jtd5|j |dddpy#1swYBxYw#1swYxYw) N2)raxisTr(keepdims)zaxis out of rangematch) rrrrrrpytestraises ValueError)r rr!r"r(s rtest_minmax_axisz#Test_MinMaxMixin1D.test_minmax_axis#s IIbM aL D 4()155dT5+J  4()155dT5+J    !Dz1DE !4  !z1DE !4  ! ! ! ! ! ! !s1D'DD D c tjgd}||}ttj|tj|ttj|tj|y)N)rr,)rarrayrrr)r rdatdatsps rtest_numpy_minmaxz$Test_MinMaxMixin1D.test_numpy_minmax4sMhhy!#266%="&&+6266%="&&+6rctjgd}tjgd}tjgd}tjgd}tjgd}|||||fD]}||}t|jtj|t|j tj|t|jdtj|dt|j dtj|dtj d} d D]v} || }t jtd 5|j | dddt jtd 5|j| dddxy#1swYBxYw#1swYxYw) N)r&rr4rrr&r+)r&r+r)r,r4r:r)r,r4rrrr')r)Nrzto an empty matrixr-) rr5rargmaxargminemptyr/r0r1) r rD1D2D3D4D5r!matD6r(s r test_argmaxzTest_MinMaxMixin1D.test_argmax;sn XXm $ XXn % XX& ' XXl # XXl #b"b"% CAA,C ryy| 4 ryy| 4 +RYYqq-A B +RYYqq-A B CXXd^ &DB-Cz1EF &  % &z1EF &  % & &  & & & & &sG7G!G !G* N)__name__ __module__ __qualname__r#r2r8rGrrrrs$!"7&rrceZdZdZy)Test_ShapeMinMax2DWithAxisc *tjgdgdgdg}||}|jtjf|jtjf|jtjf|j tj ffD]h\}}dD]^\}}t t|||||t ||j|||jdk(r^Jj|j|jfD],}dD]%}t||tjr%J.ttt t"d}||j|} | jtjf| jtjf| jtjf| j tj ffD]i\} } d D]_\}}t t| || ||d t | |j|| |jdk(r_Jk| j| jfD],}dD]%}t||tjr%J.y) N)r&rrr:r;)rrr,r4))r)r)r,)r:r'coo)rr,)rOcsrcscbsr))r)r,r)r,)r:r,Tr))rr5rrnanminnanmaxrrshapeformatr>r=rrr r r r ) r rr6r7spminmaxnpminmaxax result_shape spmat_formdatspmspmnpms rr#z&Test_ShapeMinMax2DWithAxis.test_minmaxWsLhh ~|DE#YY  YY  \\299 % \\299 % % 9 Xx %; 9 LWX2%67#B9OPX2.44lCR(//5888 9 9u||4 AH A!("3RZZ@@@ A A   *ELL)#.ZZ ZZ ]]BII & ]]BII &  4HC %? 4 LWSb\2C"t4TUSb\//>|**e333 4 4  6 AH A!("3RZZ@@@ A ArN)rHrIrJr#rKrrrMrMUs)ArrM)__doc__r/numpyr numpy.testingrr scipy.sparserrrrr r r r scipy.sparse._sputilsr rformats_for_minmax formats_for_minmax_supporting_1dmark parametrizerrMrKrrrhs9 :CCGG.  IyA$-y#9 &FG9&9&H9&x&89*A*A:*Ar