U
    rhf                     @  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dZeddddZeddddZeddd	d
ZeddddZeddddZ	eddddZ
dS )CumSumNone)returnc                  C  st   t jjdddgdgd} tdddd	d
gtj}td}tdddddgtj}t| ||g|gdd d S )Nr   xaxisyinputsoutputs      ?       @      @      @      @r         @      $@      .@Ztest_cumsum_1dr   r   name	onnxhelper	make_nodenparrayastypefloat64int32r   noder   r	   r
    r#   F/tmp/pip-unpacked-wheel-xnis5xre/onnx/backend/test/case/node/cumsum.pyexport_cumsum_1d   s
    
zCumSum.export_cumsum_1dc                  C  sv   t jjdddgdgdd} tddd	d
dgtj}td}tddd	ddgtj}t| ||g|gdd d S )Nr   r   r	   r
      )r   r   	exclusiver   r   r   r   r   r           r   r   Ztest_cumsum_1d_exclusiver   r   r!   r#   r#   r$   export_cumsum_1d_exclusive   s       
z!CumSum.export_cumsum_1d_exclusivec                  C  sv   t jjdddgdgdd} tddd	d
dgtj}td}tdddddgtj}t| ||g|gdd d S )Nr   r   r	   r
   r&   )r   r   reverser   r   r   r   r   r   r         ,@      (@      "@Ztest_cumsum_1d_reverser   r   r!   r#   r#   r$   export_cumsum_1d_reverse    s       
zCumSum.export_cumsum_1d_reversec                  C  sx   t jjdddgdgddd} tddd	d
dgtj}td}tdddddgtj}t| ||g|gdd d S )Nr   r   r	   r
   r&   )r   r   r*   r'   r   r   r   r   r   r   r+   r,   r-   r(   Z test_cumsum_1d_reverse_exclusiver   r   r!   r#   r#   r$   "export_cumsum_1d_reverse_exclusive*   s         
   z)CumSum.export_cumsum_1d_reverse_exclusivec                  C  s   t jjdddgdgd} tdddd	d
dgtjd}td}tdddd
ddgtjd}t	| ||g|gdd d S )Nr   r   r	   r
   r   r   r   r   r   r   r         r   g      @r-   Ztest_cumsum_2d_axis_0r   
r   r   r   r   r   r   r   Zreshaper    r   r!   r#   r#   r$   export_cumsum_2d_axis_06   s    $
$zCumSum.export_cumsum_2d_axis_0c                  C  s   t jjdddgdgd} tdddd	d
dgtjd}td}tdddd	ddgtjd}t	| ||g|gdd d S )Nr   r   r	   r
   r   r   r   r   r   r   r   r0   r&   r-   r   Ztest_cumsum_2d_axis_1r   r3   r!   r#   r#   r$   export_cumsum_2d_axis_1B   s    $
$zCumSum.export_cumsum_2d_axis_1c                  C  s   t jjdddgdgd} tdddd	d
dgtjd}td}tdddd	ddgtjd}t	| ||g|gdd d S )Nr   r   r	   r
   r   r   r   r   r   r   r   r0   r-   r   Ztest_cumsum_2d_negative_axisr   r3   r!   r#   r#   r$   export_cumsum_2d_negative_axisN   s    $
$z%CumSum.export_cumsum_2d_negative_axisN)__name__
__module____qualname__staticmethodr%   r)   r.   r/   r4   r5   r7   r#   r#   r#   r$   r      s   		r   )

__future__r   Znumpyr   r   Zonnx.backend.test.case.baser   Zonnx.backend.test.case.noder   r   r#   r#   r#   r$   <module>   s
   