
    bIg
                    N    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e      Z	y)    )annotationsN)Base)expectc                  f    e Zd Zedd       Zedd       Zedd       Zedd       Zedd       Zy)	Unsqueezec            
        t         j                  j                  ddd      j                  t         j                        } t        | j                        D ]  }t        j                  |g      j                  t         j                        }t        j                  j                  dddgdg      }t        j                  | |	      }t        || |g|gd
t        |      z           y )N         r   xaxesyinputsoutputsaxistest_unsqueeze_axis_r   r   name)nprandomrandnastypefloat32rangendimarrayint64onnxhelper	make_nodeexpand_dimsr   str)r   ir   noder   s        b/var/www/html/answerous/venv/lib/python3.12/site-packages/onnx/backend/test/case/node/unsqueeze.pyexport_unsqueeze_one_axisz#Unsqueeze.export_unsqueeze_one_axis   s    IIOOAq!$++BJJ7qvv 	A88QC=''1D;;((V} ) D
 qq)A4y+c!f4		    c                    t         j                  j                  ddd      j                  t         j                        } t        j
                  ddg      j                  t         j                        }t        j                  j                  dddgdg	      }t        j                  | d
      }t        j                  |d
      }t        || |g|gd       y )Nr	   r
   r      r   r   r   r   r   r   test_unsqueeze_two_axesr   r   r   r   r   r   r   r   r    r!   r"   r#   r   r   r   r&   r   s       r'   export_unsqueeze_two_axesz#Unsqueeze.export_unsqueeze_two_axes"   s    IIOOAq!$++BJJ7xxA&&rxx0{{$$=E % 

 NN11%NN11%tQIs9RSr)   c                    t         j                  j                  ddd      j                  t         j                        } t        j
                  g d      j                  t         j                        }t        j                  j                  dddgdg	      }t        j                  | d
      }t        j                  |d      }t        j                  |d      }t        || |g|gd       y )Nr	   r
   r   )   r
   r   r   r   r   r   r   r1   r   test_unsqueeze_three_axesr   r-   r.   s       r'   export_unsqueeze_three_axesz%Unsqueeze.export_unsqueeze_three_axes1   s    IIOOAq!$++BJJ7xx	"))"((3{{$$=E % 

 NN11%NN11%NN11%tQIs9TUr)   c                    t         j                  j                  ddd      j                  t         j                        } t        j
                  g d      j                  t         j                        }t        j                  j                  dddgdg	      }t        j                  | d
      }t        j                  |d      }t        j                  |d      }t        || |g|gd       y )Nr	   r
   r   )r   r
   r1   r   r   r   r   r   r1   r   test_unsqueeze_unsorted_axesr   r-   r.   s       r'   export_unsqueeze_unsorted_axesz(Unsqueeze.export_unsqueeze_unsorted_axesA   s    IIOOAq!$++BJJ7xx	"))"((3{{$$=E % 

 NN11%NN11%NN11%tQIs9WXr)   c                    t         j                  j                  dddgdg      } t        j                  j                  dddd      j                  t        j                        }t        j                  d	g      j                  t        j                        }t        j                  |d	
      }t        | ||g|gd       y )Nr   r   r   r   r   r+   r	   r   r   test_unsqueeze_negative_axesr   )r    r!   r"   r   r   r   r   r   r   r   r#   r   )r&   r   r   r   s       r'   export_unsqueeze_negative_axesz(Unsqueeze.export_unsqueeze_negative_axesQ   s    {{$$=E % 

 IIOOAq!Q'..rzz:xx~$$RXX.NN12&tQIs9WXr)   N)returnNone)	__name__
__module____qualname__staticmethodr(   r/   r3   r6   r:    r)   r'   r   r      sl     & T T V V Y Y 	Y 	Yr)   r   )

__future__r   numpyr   r    onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   rA   r)   r'   <module>rF      s&    #   , .NY NYr)   