
    ni                         S SK rS SKrS SKJr  S SKJs  Js  Jr  \" SS jS9\l	        S r
S rS r\R                  R                  SS	S
/5      S 5       rS rS rS rS rg)    NSimpleNamespacec                     [        XS9$ Nlowerupperr   )lustatics      j/home/james-whalen/.local/lib/python3.13/site-packages/tests/unit/disable_jit/test_annotation_functions.py<lambda>r      s
    VW@a    )closedc                     [        XS9$ r   r   r   s     r   	_intervalr   	   s    44r   c                  Z    [        SS5      [        SS5      /[        SS5      //n SS/nX4$ )	z8Return sample annotations and weights used across tests.g?g?g333333?皙?g      ?333333?      ?g       @)r   )annotationsweightss     r   _example_annotationsr      sB     
3	iS12	3	K CjGr   c                     [        5       u  p[        R                  " XSS9u  p#[        R                  " XSS9u  pE[        R                  " XSS9u  pg[        R                  R                  U[        R                  " SS/5      5        [        R                  R                  U[        R                  " SS/5      5        [        R                  R                  U[        R                  " S	S	/5      5        X5s=:X  a  Us=:X  a  S
:X  d   e   eg )Nr   )moder	   invalidr   r   r   333333?g           )r   af_get_weighted_sumnptestingassert_allclosearray)r   r   	lower_sumcnt_l	upper_sumcnt_uinvalid_sumcnt_is           r   test_get_weighted_sum_modesr+      s    /1K++KwOI++KwOI--kSKJJy"((C:*>?JJy"((C:*>?JJ{BHHc3Z,@A'U'a'''''r   zlower, upper, expected)g?g?)r      )r   g      ?)r,   r,   c                 <    [         R                  " X5      U:X  d   eg )N)r   _check_bound)r   r	   expecteds      r   test_check_boundr0       s     ??5(H444r   c                      [        5       u  p[        R                  " X5      nUR                  [        R
                  " S5      :X  d   eUR                  [        R
                  " S5      :X  d   eg Ng?r   )r   r   averager   pytestapproxr	   r   r   results      r   test_averager8   *   sQ    /1KZZ-F<<6==1111<<6==----r   c                      [        5       u  p[        R                  " X5      nUR                  [        R
                  " S5      :X  d   eUR                  [        R
                  " S5      :X  d   eg r2   )r   r   average_lowerr   r4   r5   r	   r6   s      r   test_average_lowerr;   0   sS    /1Kk3F<<6==1111<<6==----r   c                      [        5       u  p[        R                  " X5      nUR                  [        R
                  " S5      :X  d   eUR                  [        R
                  " S5      :X  d   eg )Nr   )r   r   maximumr   r4   r5   r	   r6   s      r   test_maximumr>   6   Q    /1KZZ-F<<6==----<<6==----r   c                      [        5       u  p[        R                  " X5      nUR                  [        R
                  " S5      :X  d   eUR                  [        R
                  " S5      :X  d   eg )Nr   r   )r   r   minimumr   r4   r5   r	   r6   s      r   test_minimumrB   <   r?   r   )F)numpyr!   r4   typesr   :pyreason.scripts.annotation_functions.annotation_functionsscriptsannotation_functionsr   intervalr   r   r+   markparametrizer0   r8   r;   r>   rB    r   r   <module>rL      su      ! G G%ab5 	( 55....r   