
    h;                     H    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)    N)Base)expectc                 \    [         R                  R                  SU S9R                  U5      $ )N   )p)nprandombinomialastype)xdtypes     _/var/www/fran/franai/venv/lib/python3.13/site-packages/onnx/backend/test/case/node/bernoulli.py"bernoulli_reference_implementationr      s)    
 99a1%,,U33    c                   P    \ rS rSr\SS j5       r\SS j5       r\SS j5       rSrg)	Bernoulli   Nc                      [         R                  R                  SS/S/S9n [        R                  R                  SSS5      R                  [        5      n[        U[        5      n[        X/U/SS	9  g )
Nr   r   y)inputsoutputs              ?
   test_bernoullir   r   name)
onnxhelper	make_noder   r	   uniformr   floatr   r   noder   r   s      r   export_bernoulli_without_dtype(Bernoulli.export_bernoulli_without_dtype   sj    {{$$5E % 
 IIc3+2259.q%8tC!3CDr   c                  .   [         R                  R                  SS/S/[         R                  R                  S9n [
        R                  R                  SSS5      R                  [
        R                  5      n[        U[        5      n[        X/U/SS	9  g )
Nr   r   r   )r   r   r   r   r   r   test_bernoulli_doubler   )r   r   r    TensorProtoDOUBLEr   r	   r!   r   float32r   r"   r   r#   s      r   export_bernoulli_with_dtype%Bernoulli.export_bernoulli_with_dtype!   s}    {{$$5E""))	 % 
 IIc3+222::>.q%8tC!3JKr   c                  (   [        S5      n [        R                  R                  SS/S/U S9n[        R
                  R                  SSS5      R                  [        R                  5      n[        U[        R                  5      n[        X/U/S	S
9  g )Nr   r   r   r   )r   r   seedr   r   r   test_bernoulli_seedr   )r"   r   r   r    r   r	   r!   r   r+   r   r   )r/   r$   r   r   s       r   export_bernoulli_with_seed$Bernoulli.export_bernoulli_with_seed.   s~    Qx{{$$5E	 % 
 IIc3+222::>.q"**=tC!3HIr    )returnN)	__name__
__module____qualname____firstlineno__staticmethodr%   r,   r1   __static_attributes__r3   r   r   r   r      sB    	E 	E 
L 
L J Jr   r   )	numpyr   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r   r3   r   r   <module>r>      s&   
   , .4&J &Jr   