
    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                      e Zd Zed
d       Zed
d       Zed
d       Zed
d       Zed
d       Zed
d       Z	ed
d       Z
ed
d       Zy	)BitShiftc                 N   t         j                  j                  dddgdgd      } t        j                  g d      j                  t        j                        }t        j                  g d      j                  t        j                        }||z	  }t        | ||g|gd	
       y )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       a/var/www/html/answerous/venv/lib/python3.12/site-packages/onnx/backend/test/case/node/bitshift.pyexport_right_unit8zBitShift.export_right_unit8   s    {{$$SzC5G % 
 HHZ ''1HHY&&rxx0FtQFQC6QR    c                 N   t         j                  j                  dddgdgd      } t        j                  g d      j                  t        j                        }t        j                  g d      j                  t        j                        }||z	  }t        | ||g|gd	
       y )Nr   r	   r
   r   r   r   r   r   test_bitshift_right_uint16r   r   r   r   r   r    r!   uint16r   r#   s       r%   export_right_unit16zBitShift.export_right_unit16       {{$$SzC5G % 
 HHZ ''		2HHY&&ryy1FtQFQC6RSr'   c                 N   t         j                  j                  dddgdgd      } t        j                  g d      j                  t        j                        }t        j                  g d      j                  t        j                        }||z	  }t        | ||g|gd	
       y )Nr   r	   r
   r   r   r   r   r   test_bitshift_right_uint32r   r   r   r   r   r    r!   uint32r   r#   s       r%   export_right_unit32zBitShift.export_right_unit32$   r-   r'   c                 N   t         j                  j                  dddgdgd      } t        j                  g d      j                  t        j                        }t        j                  g d      j                  t        j                        }||z	  }t        | ||g|gd	
       y )Nr   r	   r
   r   r   r   r   r   test_bitshift_right_uint64r   r   r   r   r   r    r!   uint64r   r#   s       r%   export_right_unit64zBitShift.export_right_unit64/   r-   r'   c                 N   t         j                  j                  dddgdgd      } t        j                  g d      j                  t        j                        }t        j                  g d      j                  t        j                        }||z  }t        | ||g|gd	
       y )Nr   r	   r
   r   LEFTr   r   r   test_bitshift_left_uint8r   r   r#   s       r%   export_left_unit8zBitShift.export_left_unit8:   s    {{$$SzC5F % 
 HHZ ''1HHY&&rxx0FtQFQC6PQr'   c                 N   t         j                  j                  dddgdgd      } t        j                  g d      j                  t        j                        }t        j                  g d      j                  t        j                        }||z  }t        | ||g|gd	
       y )Nr   r	   r
   r   r9   r   r   r   test_bitshift_left_uint16r   r*   r#   s       r%   export_left_unit16zBitShift.export_left_unit16E       {{$$SzC5F % 
 HHZ ''		2HHY&&ryy1FtQFQC6QRr'   c                 N   t         j                  j                  dddgdgd      } t        j                  g d      j                  t        j                        }t        j                  g d      j                  t        j                        }||z  }t        | ||g|gd	
       y )Nr   r	   r
   r   r9   r   r   r   test_bitshift_left_uint32r   r0   r#   s       r%   export_left_unit32zBitShift.export_left_unit32P   r?   r'   c                 N   t         j                  j                  dddgdgd      } t        j                  g d      j                  t        j                        }t        j                  g d      j                  t        j                        }||z  }t        | ||g|gd	
       y )Nr   r	   r
   r   r9   r   r   r   test_bitshift_left_uint64r   r5   r#   s       r%   export_left_unit64zBitShift.export_left_unit64[   r?   r'   N)returnNone)__name__
__module____qualname__staticmethodr&   r,   r2   r7   r;   r>   rB   rE    r'   r%   r   r      s    S S T T T T T T R R S S S S S Sr'   r   )

__future__r   numpyr   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   rL   r'   r%   <module>rQ      s&    #   , .WSt WSr'   