U
    rhG                     @  sX   d dl mZ d dlmZ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 )    )annotations)AnySequenceN)Base)expectc                   @  s   e Zd ZeddddZdS )ConcatNone)returnc               
   C  s  ddgddgfddgddggddgddggfddgddggddgddgggd	d
gddggddgddgggfd} |   D ]\}}dd |D }tt|d jD ]f}dd tt|D }tjjdt|dg|d}t	||}t
|t||gd| d t| d qtt|d j dD ]l}dd tt|D }tjjdt|dg|d}t	||}t
|t||gd| d tt| d q qxd S )N                        	   
                     )Z1dZ2dZ3dc                 S  s   g | ]}t j|t jd qS ))Zdtype)npZasarrayZfloat32).0v r   F/tmp/pip-unpacked-wheel-xnis5xre/onnx/backend/test/case/node/concat.py
<listcomp>   s     z!Concat.export.<locals>.<listcomp>r   c                 S  s   g | ]}d t | qS valuestrr   kr   r   r   r      s     r   output)inputsoutputsZaxisZtest_concat_Z_axis_)r'   r(   namec                 S  s   g | ]}d t | qS r    r"   r$   r   r   r   r   +   s     Z_axis_negative_)itemsrangelenshapeonnxhelperZ	make_nodelistr   Zconcatenater   r#   abs)Z
test_casesZ	test_caseZvalues_valuesiZin_argsnoder&   r   r   r   export   sL    	      zConcat.exportN)__name__
__module____qualname__staticmethodr5   r   r   r   r   r      s   r   )
__future__r   typingr   r   Znumpyr   r.   Zonnx.backend.test.case.baser   Zonnx.backend.test.case.noder   r   r   r   r   r   <module>   s   