
    h                     B    S SK rS SKrS SKJr  S SKJr   " S S\5      rg)    N)Base)expectc                       \ 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S	 j5       r\SS
 j5       rSrg)BitShift   Nc                  B   [         R                  R                  SSS/S/SS9n [        R                  " / SQ5      R                  [        R                  5      n[        R                  " / SQ5      R                  [        R                  5      nX-	  n[        XU/U/S	S
9  g )Nr   xyzRIGHTinputsoutputs	direction         r         test_bitshift_right_uint8r   r   nameonnxhelper	make_nodenparrayastypeuint8r   noder	   r
   r   s       ^/var/www/fran/franai/venv/lib/python3.13/site-packages/onnx/backend/test/case/node/bitshift.pyexport_right_unit8BitShift.export_right_unit8   s    {{$$SzC5G % 
 HHZ ''1HHY&&rxx0FtFQC6QR    c                  B   [         R                  R                  SSS/S/SS9n [        R                  " / SQ5      R                  [        R                  5      n[        R                  " / SQ5      R                  [        R                  5      nX-	  n[        XU/U/S	S
9  g )Nr   r	   r
   r   r   r   r   r   test_bitshift_right_uint16r   r   r   r   r   r    r!   uint16r   r#   s       r%   export_right_unit16BitShift.export_right_unit16       {{$$SzC5G % 
 HHZ ''		2HHY&&ryy1FtFQC6RSr(   c                  B   [         R                  R                  SSS/S/SS9n [        R                  " / SQ5      R                  [        R                  5      n[        R                  " / SQ5      R                  [        R                  5      nX-	  n[        XU/U/S	S
9  g )Nr   r	   r
   r   r   r   r   r   test_bitshift_right_uint32r   r   r   r   r   r    r!   uint32r   r#   s       r%   export_right_unit32BitShift.export_right_unit32#   r/   r(   c                  B   [         R                  R                  SSS/S/SS9n [        R                  " / SQ5      R                  [        R                  5      n[        R                  " / SQ5      R                  [        R                  5      nX-	  n[        XU/U/S	S
9  g )Nr   r	   r
   r   r   r   r   r   test_bitshift_right_uint64r   r   r   r   r   r    r!   uint64r   r#   s       r%   export_right_unit64BitShift.export_right_unit64.   r/   r(   c                  B   [         R                  R                  SSS/S/SS9n [        R                  " / SQ5      R                  [        R                  5      n[        R                  " / SQ5      R                  [        R                  5      nX-  n[        XU/U/S	S
9  g )Nr   r	   r
   r   LEFTr   r   r   test_bitshift_left_uint8r   r   r#   s       r%   export_left_unit8BitShift.export_left_unit89   s    {{$$SzC5F % 
 HHZ ''1HHY&&rxx0FtFQC6PQr(   c                  B   [         R                  R                  SSS/S/SS9n [        R                  " / SQ5      R                  [        R                  5      n[        R                  " / SQ5      R                  [        R                  5      nX-  n[        XU/U/S	S
9  g )Nr   r	   r
   r   r=   r   r   r   test_bitshift_left_uint16r   r+   r#   s       r%   export_left_unit16BitShift.export_left_unit16D       {{$$SzC5F % 
 HHZ ''		2HHY&&ryy1FtFQC6QRr(   c                  B   [         R                  R                  SSS/S/SS9n [        R                  " / SQ5      R                  [        R                  5      n[        R                  " / SQ5      R                  [        R                  5      nX-  n[        XU/U/S	S
9  g )Nr   r	   r
   r   r=   r   r   r   test_bitshift_left_uint32r   r2   r#   s       r%   export_left_unit32BitShift.export_left_unit32O   rE   r(   c                  B   [         R                  R                  SSS/S/SS9n [        R                  " / SQ5      R                  [        R                  5      n[        R                  " / SQ5      R                  [        R                  5      nX-  n[        XU/U/S	S
9  g )Nr   r	   r
   r   r=   r   r   r   test_bitshift_left_uint64r   r8   r#   s       r%   export_left_unit64BitShift.export_left_unit64Z   rE   r(    )returnN)__name__
__module____qualname____firstlineno__staticmethodr&   r-   r4   r:   r?   rC   rH   rL   __static_attributes__rN   r(   r%   r   r      s    S S T T T T T T R R S S S S S Sr(   r   )numpyr   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   rN   r(   r%   <module>rY      s#   
   , .WSt WSr(   