
    hC                     B    S SK rS SKrS SKJr  S SKJr   " S S\5      rg)    N)Base)expectc                       \ rS rSr\SS j5       r\SS j5       r\SS j5       r\SS j5       r\SS j5       r	\SS j5       r
\SS	 j5       rS
rg)CumSum   Nc                  d   [         R                  R                  SSS/S/S9n [        R                  " / SQ5      R                  [        R                  5      n[        R                  " S5      n[        R                  " / SQ5      R                  [        R                  5      n[        XU/U/S	S
9  g )Nr   xaxisyinputsoutputs      ?       @      @      @      @r   )r   r         @      $@      .@test_cumsum_1dr   r   name	onnxhelper	make_nodenparrayastypefloat64int32r   noder	   r
   r   s       \/var/www/fran/franai/venv/lib/python3.13/site-packages/onnx/backend/test/case/node/cumsum.pyexport_cumsum_1dCumSum.export_cumsum_1d   s    {{$$XsFmcU$SHH./66rzzBxx{HH0188DtIs9IJ    c                  f   [         R                  R                  SSS/S/SS9n [        R                  " / SQ5      R                  [        R                  5      n[        R                  " S5      n[        R                  " / S	Q5      R                  [        R                  5      n[        XU/U/S
S9  g )Nr   r	   r
   r      )r   r   	exclusiver   r   )        r   r   r   r   test_cumsum_1d_exclusiver   r   r$   s       r&   export_cumsum_1d_exclusive!CumSum.export_cumsum_1d_exclusive   s    {{$$c6]SEQ % 
 HH./66rzzBxx{HH/077

CtIs9STr)   c                  f   [         R                  R                  SSS/S/SS9n [        R                  " / SQ5      R                  [        R                  5      n[        R                  " S5      n[        R                  " / S	Q5      R                  [        R                  5      n[        XU/U/S
S9  g )Nr   r	   r
   r   r+   )r   r   reverser   r   )r         ,@      (@      "@r   test_cumsum_1d_reverser   r   r$   s       r&   export_cumsum_1d_reverseCumSum.export_cumsum_1d_reverse   s    {{$$c6]SE1 % 
 HH./66rzzBxx{HH1299"**EtIs9QRr)   c                  h   [         R                  R                  SSS/S/SSS9n [        R                  " / SQ5      R                  [        R                  5      n[        R                  " S5      n[        R                  " / S	Q5      R                  [        R                  5      n[        XU/U/S
S9  g )Nr   r	   r
   r   r+   )r   r   r2   r,   r   r   )r3   r4   r5   r   r-    test_cumsum_1d_reverse_exclusiver   r   r$   s       r&   "export_cumsum_1d_reverse_exclusive)CumSum.export_cumsum_1d_reverse_exclusive)   s    {{$$c6]SE1PQ % 
 HH./66rzzBxx{HH0188DTQC6X	
r)   c                     [         R                  R                  SSS/S/S9n [        R                  " / SQ5      R                  [        R                  5      R                  S5      n[        R                  " S5      n[        R                  " / S	Q5      R                  [        R                  5      R                  S5      n[        XU/U/S
S9  g )Nr   r	   r
   r   r   r   r   r   r   r   r         r   )r   r   r   r   g      @r5   test_cumsum_2d_axis_0r   
r   r   r   r   r    r!   r"   reshaper#   r   r$   s       r&   export_cumsum_2d_axis_0CumSum.export_cumsum_2d_axis_05   s    {{$$=E % 

 HH34;;BJJGOOPVWxx{HH34;;BJJGOOPVWtIs9PQr)   c                     [         R                  R                  SSS/S/S9n [        R                  " / SQ5      R                  [        R                  5      R                  S5      n[        R                  " S5      n[        R                  " / S	Q5      R                  [        R                  5      R                  S5      n[        XU/U/S
S9  g )Nr   r	   r
   r   r   r>   r?   r+   r   r   r   r   r5   r   test_cumsum_2d_axis_1r   rC   r$   s       r&   export_cumsum_2d_axis_1CumSum.export_cumsum_2d_axis_1A   s    {{$$=E % 

 HH34;;BJJGOOPVWxx{HH45<<RZZHPPQWXtIs9PQr)   c                     [         R                  R                  SSS/S/S9n [        R                  " / SQ5      R                  [        R                  5      R                  S5      n[        R                  " S5      n[        R                  " / S	Q5      R                  [        R                  5      R                  S5      n[        XU/U/S
S9  g )Nr   r	   r
   r   r   r>   r?   rH   test_cumsum_2d_negative_axisr   rC   r$   s       r&   export_cumsum_2d_negative_axis%CumSum.export_cumsum_2d_negative_axisM   s    {{$$=E % 

 HH34;;BJJGOOPVWxx|HH45<<RZZHPPQWXtIs9WXr)    )returnN)__name__
__module____qualname____firstlineno__staticmethodr'   r/   r7   r;   rE   rJ   rO   __static_attributes__rQ   r)   r&   r   r      s    K K U U S S 	
 	
 	R 	R 	R 	R 	Y 	Yr)   r   )numpyr   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   rQ   r)   r&   <module>r\      s#   
   , .KYT KYr)   