
    Siq                    T    S SK Jr  S SKrS SKrS SKJr  S SKJr  S r	 " S S\5      r
g)    )annotationsN)Base)expectc                    [         R                  " U [         R                  S9[         R                  " U[         R                  S9[         R                  " U[         R                  S94$ )Ndtype)nparrayint64)indicesinverse_indicescountss      a/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/onnx/backend/test/case/node/unique.pyspecify_int64r      sB    
)
1
rxx(     c                      \ 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rg	)Unique   c                    [         R                  R                  SS// SQS9n [        R                  " / SQ[        R
                  S9n[        R                  " USSS5      u  p#pE[        X4U5      u  p4n[        U U/X#XE/SS	9  g )
Nr   XYr   r   r   )inputsoutputs       @      ?r         @g      @r   r   Ttest_unique_sorted_without_axisr   r   name)	onnxhelper	make_noder	   r
   float32uniquer   r   node_sortedxyr   r   r   s         r   export_sorted_without_axis!Unique.export_sorted_without_axis   s    kk++5A , 
 HH32::F.0ii4t.L+O+8f,
(& 	392		
r   c            
     F   [         R                  R                  SS// SQSS9n [        R                  " / SQ[        R
                  S9n[        R                  " USSS5      u  p#pE[        R                  " U5      n[        [        U[        R                  " [        U5      5      SS	95      nX6   n[        R                  " XSS
9n[        R                  " U Vs/ s H  oU   PM	     sn[        R                  S9nXV   n[        X4U5      u  p4n[!        U U/X#XE/SS9  g s  snf )Nr   r   r   r   r   r   sortedr   r   T)strictaxis#test_unique_not_sorted_without_axisr    )r"   r#   r$   r	   r
   r%   r&   argsortdictziparangelentakeasarrayr   r   r   )	node_not_sortedr)   r*   r   r   r   argsorted_indicesinverse_indices_mapis	            r   export_not_sorted_without_axis%Unique.export_not_sorted_without_axis*   s   ++//5A	 0 
 HH32::F.0ii4t.L+O JJw/"!299S1B-C#DTR
 ,GGAQ'**-<=_#_=RXX
 *+8f,
(& 	396		
 >s   Dc                 8   [         R                  R                  SS// SQSSS9n [        R                  " / SQ/ SQ/ SQ/[        R
                  S	9n[        R                  " US
S
S
SS9u  p#pE[        X4U5      u  p4nUR                  S5      n[        U U/X#XE/SS9  g )Nr   r   r      r   r   r   r/   r2   rB   r   r   )         r   Tr1   test_unique_sorted_with_axisr    
r"   r#   r$   r	   r
   r%   r&   r   reshaper   r'   s         r   export_sorted_with_axisUnique.export_sorted_with_axisW   s    kk++5A , 
 HHiI6bjjI.0ii4tRS.T+O+8f,
(& *11"5 	39/		
r   c                 Z   [         R                  R                  SS// SQSSS9n [        R                  " SS/SS/SS/SS//SS/SS/SS/SS///[        R
                  S	9n[        R                  " US
S
S
SS9u  p#pE[        X4U5      u  p4nUR                  S5      n[        U U/X#XE/SS9  g )Nr   r   r   rB   rC   r   g        r   r   Tr1   rH   test_unique_sorted_with_axis_3dr    rJ   r'   s         r   export_sorted_with_axis_3d!Unique.export_sorted_with_axis_3dy   s    kk++5A , 
 HHsc3Z#sc3Z@sc3Z#sc3Z@ **
 /1ii4tRS.T+O+8f,
(& *11"5 	392		
r   c                 8   [         R                  R                  SS// SQSSS9n [        R                  " / SQ/ SQ/ SQ/[        R
                  S	9n[        R                  " US
S
S
SS9u  p#pE[        X4U5      u  p4nUR                  S5      n[        U U/X#XE/SS9  g )Nr   r   r   rB   rH   rC   rD   )rE   rF   rF   r   Tr1   %test_unique_sorted_with_negative_axisr    rJ   r'   s         r    export_sorted_with_negative_axis'Unique.export_sorted_with_negative_axis   s    kk++5A , 
 HHiI6bjjI.0ii4tRT.U+O+8f,
(& *11"5 	398		
r   c                 (   [         R                  R                  SS// SQSS9n [        R                  " S/[        R
                  S9n[        R                  " USSS5      u  p#pE[        X4U5      u  p4nUR                  S	5      n[        U U/X#XE/S
S9  g )Nr   r   r   rB   r.   r   r   TrH   test_unique_length_1r    )
r"   r#   r$   r	   r
   r   r&   r   rK   r   r'   s         r   export_length_1Unique.export_length_1   s    kk++5A	 , 
 HHaS).0ii4t.L+O+8f,
(& *11"5 	39'		
r    N)returnNone)__name__
__module____qualname____firstlineno__staticmethodr+   r?   rL   rP   rT   rX   __static_attributes__rZ   r   r   r   r      s|    
 
& *
 *
X 
 
B (
 (
T  
  
D 
 
r   r   )
__future__r   numpyr	   r"   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r   rZ   r   r   <module>rg      s)    #   , .P
T P
r   