o
    soi                     @  sH   d dl mZ d dlZd dlZd dlmZ d dlmZ G dd deZ	dS )    )annotationsN)Base)expectc                   @  s   e Zd ZedddZedddZedddZedd	d
ZedddZedddZ	edddZ
edddZedddZdS )CumSumreturnNonec                  C  sh   t jjdddgdgd} tg dtj}td}tg dtj}t| ||g|gd	d
 d S )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
    r%   V/home/ubuntu/.local/lib/python3.10/site-packages/onnx/backend/test/case/node/cumsum.pyexport_cumsum_1d   s
   
zCumSum.export_cumsum_1dc                  C  j   t jjdddgdgdd} tg dtj}td}tg d	tj}t| ||g|gd
d d S )Nr   r   r	   r
      r   r   	exclusiver   r   )        r   r   r   r   test_cumsum_1d_exclusiver   r   r#   r%   r%   r&   export_cumsum_1d_exclusive      
z!CumSum.export_cumsum_1d_exclusivec                  C  r(   )Nr   r   r	   r
   r)   )r   r   reverser   r   )r         ,@      (@      "@r   test_cumsum_1d_reverser   r   r#   r%   r%   r&   export_cumsum_1d_reverse    r/   zCumSum.export_cumsum_1d_reversec                  C  sl   t jjdddgdgddd} tg dtj}td}tg d	tj}t| ||g|gd
d d S )Nr   r   r	   r
   r)   )r   r   r0   r+   r   r   )r1   r2   r3   r   r,    test_cumsum_1d_reverse_exclusiver   r   r#   r%   r%   r&   "export_cumsum_1d_reverse_exclusive*   s   

z)CumSum.export_cumsum_1d_reverse_exclusivec                  C  t   t jjdddgdgd} tg dtjd}td}tg d	tjd}t	| ||g|gd
d d S )Nr   r   r	   r
   r   r   r   r   r   r   r         r   )r   r   r   r   g      @r3   test_cumsum_2d_axis_0r   
r   r   r   r   r   r    r!   reshaper"   r   r#   r%   r%   r&   export_cumsum_2d_axis_06      
zCumSum.export_cumsum_2d_axis_0c                  C  r8   )Nr   r   r	   r
   r   r9   r:   r)   r   r   r   r   r3   r   test_cumsum_2d_axis_1r   r>   r#   r%   r%   r&   export_cumsum_2d_axis_1B   rA   zCumSum.export_cumsum_2d_axis_1c                  C  r8   )Nr   r   r	   r
   r   r9   r:   rB   test_cumsum_2d_negative_axisr   r>   r#   r%   r%   r&   export_cumsum_2d_negative_axisN   rA   z%CumSum.export_cumsum_2d_negative_axisc                  C  st   t jjdddgdgd} tg dtjd}td}tg d	tjd}t| ||g|gd
d d S )Nr   r   r	   r
   r   )r)   r;   r<            r:   r   )r)   r;   r<   rI      	   test_cumsum_2d_int32r   )	r   r   r   r   r   r    r"   r?   r   r#   r%   r%   r&   export_cumsum_2d_int32Z   rA   zCumSum.export_cumsum_2d_int32c                  C  sj   t jjdddgdgdd} tg dtj}td}tg d	tj}t| ||g|gd
d d S )Nr   r   r	   r
   r)   r*   )r)   r;   r<   rH   rI   r   )r   r)   r<   rJ   
   test_cumsum_1d_int32_exclusiver   )r   r   r   r   r   r    r"   r   r#   r%   r%   r&    export_cumsum_1d_int32_exclusivef   s   

z'CumSum.export_cumsum_1d_int32_exclusiveN)r   r   )__name__
__module____qualname__staticmethodr'   r.   r5   r7   r@   rD   rG   rN   rQ   r%   r%   r%   r&   r      s&    		r   )

__future__r   numpyr   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r%   r%   r%   r&   <module>   s   