
    h 
              
           S SK JrJr  S SKrS SKrS SKJr  S SKJ	r	   SS\\   S\R                  S\R                  S\\   4S	 jjr " S
 S\5      rg)    )AnyListN)Base)expectsequencetensorpositionreturnc                 t    [        U 5      nUb  US   nUR                  XA5        U$ UR                  U5        U$ )Nr   )listinsertappend)r   r   r	   seqinsert_positions        d/var/www/fran/franai/venv/lib/python3.13/site-packages/onnx/backend/test/case/node/sequenceinsert.py(sequence_insert_reference_implementationr      sD     x.C #1+

?+ J 	

6J    c                   (    \ rS rSr\SS j5       rSrg)SequenceInsert   Nc                     [         R                  " / SQ5      R                  [         R                  5      /[         R                  " / SQ5      [         R                  " S/5      R                  [         R                  5      /S.n [         R                  " / SQ5      R                  [         R                  5      [         R                  " / SQ5      R                  [         R                  5      [         R                  " SS/5      R                  [         R                  5      /nU R	                  5        H  u  p#US   R                  [         R                  5      n[        U5      S	:  aG  [        R                  R                  S
/ SQS/S9nUS	   n[        XU5      n[        UXU/U/SU-   S9  M}  [        R                  R                  S
SS/S/S9n[        X5      n[        UX/U/SU-   S9  M     g )N)
         )r   r   )at_backat_front)            )            	   r   r   )r   r   r	   output_sequence)inputsoutputstest_sequence_insert_)r)   r*   namer   r   )nparrayastypeint64itemslenonnxhelper	make_noder   r   )
test_casesr   	test_nametest_inputsr   noder	   inserteds           r   exportSequenceInsert.export   s    .55bhh?@+.!0D0DRXX0NO


 HH\"))"((3HHY&&rxx0HHaV##BHH-
 '1&6&6&8"I ^**2884F;!#{{,,$=./ - 
 'q>Ch $h7%J09<	 {{,,$&1./ - 
 DHU$-%J09<	5 '9r    )r
   N)__name__
__module____qualname____firstlineno__staticmethodr;   __static_attributes__r=   r   r   r   r      s    * *r   r   )N)typingr   r   numpyr-   r3   onnx.backend.test.case.baser   onnx.backend.test.case.noder   ndarrayr   r   r=   r   r   <module>rI      s^   
    , . EI3i!#79zz	#Y ,T ,r   