
    ie\                        % S SK r S SKJrJrJrJr  S SKJr  S SKJ	r	J
r
JrJrJr  S SKJr  S SKJrJr  S SKJr  S SKJrJrJr  S S	KJr  \" S
5      r " S S\5      r " S S\5      r " S S\5      r " S S\5      r " S S\5      r \!\"-  \#-  \$-  \%-  r&\
\'S'   \\-  \-  \-  \ -  r(\
\'S'   \	S   r)\
\'S'   \!\"-  \%-  \*\	S   \#4   -  \*\	S   \!4   -  r+\
\'S'   \" SSS9 " S S 5      5       r,\" SSS9 " S! S"5      5       r-S#\!S$\!S%\!4S& jr.S'\!S$\!S%\!4S( jr/S)\\   S*\!S$\!S+\\/\!4   S%\!4
S, jr0S'\!S*\!S$\!S-\\!/\4   S%\1\   4
S. jr2S/\!S0\!S%\!4S1 jr3S'\!S0\!S%\!4S2 jr4S)\&S3\(S%\!4S4 jr5S'\!S5\)S3\(S%\&4S6 jr6S)\&S3\(S%\#4S7 jr7S8\\&   S%\*\+S94   4S: jr8S)\&S%\+4S; jr9S<\&S=\&S%\#4S> jr:S5\)S?\(S@\(S%\#4SA jr;S5\)S3\(S%\*\S94   4SB jr<S8\\&   S%\!4SC jr=g)D    N)CallableHashableIterableSequence)	dataclass)Literal	TypeAlias	TypedDictTypeVarcast)ChoiceTooLarge)float_to_lexlex_to_float)identity)float_to_intmake_float_clampersign_aware_lte)IntervalSetTc                   Z    \ rS rSr% \S-  \S'   \S-  \S'   \\\4   S-  \S'   \\S'   Srg)IntegerConstraints   N	min_value	max_valueweightsshrink_towards )	__name__
__module____qualname____firstlineno__int__annotations__dictfloat__static_attributes__r       d/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/hypothesis/internal/conjecture/choice.pyr   r      s0    TzTz#u*$$r'   r   c                   >    \ rS rSr% \\S'   \\S'   \\S'   \\S'   Srg)FloatConstraints&   r   r   	allow_nansmallest_nonzero_magnituder   N)r   r   r    r!   r%   r#   boolr&   r   r'   r(   r*   r*   &   s    O %%r'   r*   c                   4    \ rS rSr% \\S'   \\S'   \\S'   Srg)StringConstraints-   	intervalsmin_sizemax_sizer   N)r   r   r    r!   r   r#   r"   r&   r   r'   r(   r0   r0   -   s    MMr'   r0   c                   *    \ rS rSr% \\S'   \\S'   Srg)BytesConstraints3   r3   r4   r   N)r   r   r    r!   r"   r#   r&   r   r'   r(   r6   r6   3   s    MMr'   r6   c                        \ rS rSr% \\S'   Srg)BooleanConstraints8   pr   N)r   r   r    r!   r%   r#   r&   r   r'   r(   r9   r9   8   s    Hr'   r9   ChoiceTChoiceConstraintsT)integerstringbooleanr%   bytesChoiceTypeTr.   r%   
ChoiceKeyTTF)slotsfrozenc                   @    \ rS rSr% \S   \S'   \S-  \S'   SS jrSrg)	ChoiceTemplateJ   simplesttypeNcountc                 D    U R                   b  U R                   S:  d   eg g )Nr   )rK   selfs    r(   __post_init__ChoiceTemplate.__post_init__O   s!    ::!::>!> "r'   r   )returnN)	r   r   r    r!   r   r#   r"   rO   r&   r   r'   r(   rG   rG   J   s    
*
:"r'   rG   c                       \ rS rSr% \\S'   \\S'   \\S'   \\S'   Sr	\
S-  \S'   SSS.S	\S-  S
\S-  SS 4S jjr\S\4S j5       rS\S\4S jrS\
4S jrS\4S jrSrg)
ChoiceNodeT   rJ   valueconstraints
was_forcedNindex)
with_valuewith_constraintsrY   rZ   rQ   c                    U R                   (       a
  Ub   S5       e[        U R                  Uc  U R                  OUUc  U R                  OUU R                   S9$ )Nz*modifying a forced node doesn't make sense)rJ   rU   rV   rW   )rW   rS   rJ   rU   rV   )rN   rY   rZ   s      r(   copyChoiceNode.copy\   sY     ??%S'SS%  * 2$**
$4$<  BR
 	
r'   c                    U R                   (       a  gU R                  S:w  a7  [        SU R                  U R                  5      n[	        U R
                  U5      $ [        [        U R                  5      nUS   nUS   nSnU[        R                  * :X  a*  U[        R                  :X  a  [	        U R
                  U5      $ [        R                  " U5      (       d  [        R                  " U5      (       d  [        R                  " U5      [        R                  " U5      ::  a_  [        [        R                  " U5      U5      n[        [        R                  " U5      U5      n[	        U R
                  [        U5      5      $ g)a  
A node is trivial if it cannot be simplified any further. This does not
mean that modifying a trivial node can't produce simpler test cases when
viewing the tree as a whole. Just that when viewing this node in
isolation, this is the simplest the node can get.
Tr%   r   r   r   g        F)rW   rJ   choice_from_indexrV   choice_equalrU   r   r*   mathinfisinfceilfloormaxminr%   )rN   
zero_valuerV   r   r   r   s         r(   trivialChoiceNode.trivialr   s    ??99*1dii9I9IJJ

J77/1A1ABK#K0I#K0I NTXXI%)txx*?#DJJ?? JJy))

9--IIi(DJJy,AA "%TYYy%9>!J!$TZZ	%:N!K#DJJn0EFF r'   otherc                 \   [        U[        5      (       d  [        $ U R                  UR                  :H  =(       ar    [	        U R
                  UR
                  5      =(       aK    [        U R                  U R                  UR                  5      =(       a    U R                  UR                  :H  $ N)	
isinstancerS   NotImplementedrJ   r`   rU   choice_constraints_equalrV   rW   )rN   rk   s     r(   __eq__ChoiceNode.__eq__   s}    %,,!! II# 4TZZ54(D4D4DeFWFWX4 5#3#33		
r'   c                     [        U R                  [        U R                  5      [	        U R                  U R
                  5      U R                  45      $ rm   )hashrJ   
choice_keyrU   choice_constraints_keyrV   rW   rM   s    r(   __hash__ChoiceNode.__hash__   sB    		4::&&tyy$2B2BC	
 	
r'   c                     U R                   (       a  SOSnU R                   SU R                  < U SU R                  < 3$ )Nz	 [forced]  )rW   rJ   rU   rV   )rN   forced_markers     r(   __repr__ChoiceNode.__repr__   s9    '+B))Adjj^M?!D<L<L;OPPr'   r   )r   r   r    r!   rB   r#   r<   r=   r.   rX   r"   r\   propertyri   objectrq   rw   strr}   r&   r   r'   r(   rS   rS   T   s    
N##E3:
 &*6:	
 dN
 -t3	

 

, % % %N	
F 	
t 	

# 
Q# Qr'   rS   sizealphabet_sizerQ   c                j    US::  a	  U S:X  d   egUS:X  a  U $ X-  S-
  US-
  -  n[        [        U5      $ )Nr      )r   r"   )r   r   vs      r(   _size_to_indexr      sK     qyy		q	 ma&78AQ<r'   rX   c                    US:X  a  gUS:X  a  U $ XS-
  -  S-   n[         R                  " X!5      nS[         R                  " U5      U-
  s=:  a  S:  a  O  OSnX!:  a  X!-  nUS-  nX!:  a  M  U$ [         R                  " U5      $ )Nr   r   gHz>)ra   logrd   re   )rX   r   totalr   ss        r(   _index_to_sizer      s    	!	  Q&'!+E88E)D
 	499T?T!(D($#EFA $ ::dr'   choicer3   to_orderc                    [        [        U 5      US9[        XS9-
  nSn[        U 5       H  nXEU" U5      -  -  nXR-  nM     U$ )Nr   r   )r   lenreversed)r   r3   r   r   rX   running_expcs          r(   collection_indexr      sZ     3v;mD~H E Kfx{**$  Lr'   
from_orderc                   SSK Jn  U [        XS9-  n [        XS9nXT:  a  [        eU [        XRS9-  n / n[        [        U5      5       H3  nU S:X  a  SnOXU-  -  nXX'-  -  -  n UR                  U" U5      5        M5     U$ )Nr   )BUFFER_SIZEr   )%hypothesis.internal.conjecture.enginer   r   r   r   r   rangeappend)	rX   r3   r   r   r   r   valsins	            r(   collection_valuer      s     B
 
^HBBE%=D  
^D>>EDeDk" A:A*+A-*++EJqM" # Kr'   rU   r   c                :    S[        X-
  5      -  nX:  a  US-  nU$ )N   r   )abs)rU   r   rX   s      r(   zigzag_indexr      s*     N*++E
Lr'   c                F    U S:  d   eU S-   S-  nU S-  S:X  a  US-  nX-   $ )Nr   r   r   r   )rX   r   r   s      r(   zigzag_valuer   )  s:    A::	qA	a	Rr'   rV   c                    [        U [        5      (       Ga5  [        U [        5      (       Gd  [        [        U5      nUS   nUS   nUS   nUb  [        X25      nUb  [        XB5      nUc  Uc	  [        XS9$ Ub#  Uc   [        X-
  5      X#-
  ::  a	  [        XS9$ X-
  $ Ub#  Uc   [        X-
  5      XB-
  ::  a	  [        XS9$ X@-
  $ Uc   eUc   eUS   b/  [        S US   R                  5        5       5      (       d   S5       eX#-
  XB-
  :  a   [        X-
  5      X#-
  ::  a	  [        XS9$ X-
  $ [        X-
  5      XB-
  ::  a	  [        XS9$ X@-
  $ [        U [        5      (       a1  [        [        U5      nUS   nS	Us=:  a  S
:  d   g  g[        U 5      $ [        U [        5      (       a,  [        [        U5      n[        [        U 5      US   S[         S9$ [        U ["        5      (       a7  [        [$        U5      nUS   n[        U US   ['        U5      UR(                  S9$ [        U [*        5      (       a=  [        [,        R.                  " S
U 5      S:  5      nUS-  [1        [        U 5      5      -  $ [2        e)Nr   r   r   r   r   c              3   *   #    U  H	  oS :  v   M     g7fr   Nr   .0ws     r(   	<genexpr>"choice_to_index.<locals>.<genexpr>x        9>!A>   z@technically possible but really annoying to support zero weightsr;         ;      ?r   r3      )r3   r   r   r2   @   )rn   r"   r.   r   r   rf   rg   r   r   allvaluesr9   rA   r6   r   listr   r   r0   r   index_from_char_in_shrink_orderr%   ra   copysignr   NotImplementedError)r   rV   r   r   r   r;   r2   signs           r(   choice_to_indexr   3  s    &#z&$'?'? -{;$%56,	,	  ;N  ;N!2FF"y'8 6*+0JK#FJJ%%"y'86*+	0JK#FJJ%% ((((((y)1S 9*95<<>9 6 6 RQR 
 *y/IJ v./N4NO'NN)) v./I4NO'NN ))	FD	!	!-{;Q1.1 26{	FE	"	"+[9L ,	
 	
 
FC	 	 ,k:,	 ,i.>>	
 	
 
FE	"	"4==f-12
l3v;777!!r'   choice_typec                    U S:  d   eUS:X  Ga  [        [        U5      nUS   nUS   nUS   nUb  [        XC5      nUb  [        XS5      nUc  Uc	  [	        XS9$ Ub  Uc  U [        XCS9::  a	  [	        XS9$ X-   $ Ub  Uc  U [        XSS9::  a	  [	        XS9$ XP-
  $ Uc   eUc   eUS   b/  [        S US   R                  5        5       5      (       d   S	5       eX4-
  XS-
  :  a  U [        XCS9::  a	  [	        XS9$ X-   $ U [        XSS9::  a	  [	        XS9$ XP-
  $ US
:X  aH  [        [        U5      nUS   nS nUS::  a  SnOUS:  a  SnU S;   d   eUb
  U S:X  d   eU$ [        U 5      $ US:X  a.  [        [        U5      n[        U US   S[        S9n[        U5      $ US:X  aH  [        [        U5      nUS   n	[        U US   [        U	5      U	R                   S9n
SR#                  U
5      $ US:X  aO  [        [$        U5      nU S-	  (       a  SOSnU['        U S-  5      -  n[)        US   US   US   US   S9nU" U5      $ [*        e) Nr   r>   r   r   r   r   r   c              3   *   #    U  H	  oS :  v   M     g7fr   r   r   s     r(   r   $choice_from_index.<locals>.<genexpr>  r   r   z4possible but really annoying to support zero weightsr@   r;   r   Fr   T>   r   r   rA   r3   r   )r3   r   r   r?   r2   rz   r%   r   r   r   l    r-   r,   )r   r   r-   r,   )r   r   rf   rg   r   r   r   r   r9   r.   r6   r   r   rA   r0   r   char_in_shrink_orderjoinr*   r   r   r   )rX   r   rV   r   r   r   r;   onlyvalue_br2   value_sr   resultclampers                 r(   r_   r_     s    A::i-{;$%56,	,	  ;N  ;N!2EE"y'8YNN#EII$$"y'8YNN#EII$$ ((((((y)1S 9*95<<>9 6 6 FEF  *y/IJLRR'MM(( LRR'MM ((			!-{;
?D>"DA::KE{		+[9" ,	
 W~		 ,k:,	 # ,i. 55	
 www		+[9b[raUm%<==$!+.!+.'23O'P!+.	
 v!!r'   c                    ^ [        U [        5      (       aL  [        U [        5      (       d7  [        [        T5      mTS   nTS   nUb  X:  a  gUS L=(       a    X:  (       + $ [        U [
        5      (       aq  [        [        T5      m[        R                  " U 5      (       a  TS   $ S[        U 5      s=:  a
  TS   :  a   g  [        TS   U 5      =(       a    [        U TS   5      $ [        U [        5      (       aQ  [        [        T5      m[        U 5      TS   :  a  gTS   b  [        U 5      TS   :  a  g[        U4S	 jU  5       5      $ [        U [        5      (       aB  [        [         T5      m[        U 5      TS   :  a  gTS   S L =(       d    [        U 5      TS   :*  $ [        U [        5      (       a+  [        ["        T5      mTS
   S::  a  U SL $ TS
   S:  a  U SL $ g[%        S['        U 5       SU  35      e)Nr   r   Fr,   r   r-   r3   r4   c              3   F   >#    U  H  n[        U5      TS    ;   v   M     g7f)r2   N)ord)r   r   rV   s     r(   r   #choice_permitted.<locals>.<genexpr>&  s     Fv!3q6[55vs   !r;   r   Tzunhandled type z with value )rn   r"   r.   r   r   r%   r*   ra   isnanr   r   r   r0   r   r   rA   r6   r9   r   rJ   )r   rV   r   r   s    `  r(   choice_permittedr     s   &#z&$'?'?-{;,	,	 V%7T)@f.@AA	FE	"	"+[9::f{++s6{F[)EFF Gk+6? 
NK,E
 	
 
FC	 	 ,k:v;Z00
#/Fk*55FvFFF	FE	"	"+[9v;Z00:&$.X#f+ZAX2XX	FD	!	!-{;sq U?"sq T>!!ODL>fX"VWWr'   choices.c                 &    [        S U  5       5      $ )Nc              3   8   #    U  H  n[        U5      v   M     g 7frm   )ru   )r   r   s     r(   r   choices_key.<locals>.<genexpr>8  s     :'F##'s   )tuple)r   s    r(   choices_keyr   7  s    :':::r'   c                 |    [        U [        5      (       a  S[        U 5      4$ [        U [        5      (       a  SU 4$ U $ )Nr%   r.   )rn   r%   r   r.   )r   s    r(   ru   ru   ;  s>    &%   f-..&$Mr'   choice1choice2c                 n    [        U 5      [        U5      L d   X45       e[        U 5      [        U5      :H  $ rm   )rJ   ru   )r   r   s     r(   r`   r`   F  s5    =DM)=G+==)g*W"555r'   constraints1constraints2c                 0    [        X5      [        X5      :H  $ rm   )rv   )r   r   r   s      r(   rp   rp   K  s"    
 "+<@VA  r'   c                 &  ^ U S:X  a4  [        [        T5      m[        TS   5      [        TS   5      TS   TS   4$ U S:X  a3  [        [        T5      mTS   TS   TS   c  S O[	        TS   5      TS   4$ [	        U4S	 j[        T5       5       5      $ )
Nr%   r   r   r,   r-   r>   r   r   c              3   .   >#    U  H
  nTU   v   M     g 7frm   r   )r   keyrV   s     r(   r   )choice_constraints_key.<locals>.<genexpr>h  s     A-@cS!-@s   )r   r*   r   r   r   sorted)r   rV   s    `r(   rv   rv   U  s     g+[9[12[12$45	
 	
 i-{;$$	*2Dk)>T8U()	
 	
 AVK-@AAAr'   c                 0    SSK Jn  [        U" U 5      5      $ )Nr   )choices_to_bytes)hypothesis.databaser   r   )r   r   s     r(   choices_sizer   k  s    4())r'   )>ra   collections.abcr   r   r   r   dataclassesr   typingr   r	   r
   r   r   hypothesis.errorsr   %hypothesis.internal.conjecture.floatsr   r   $hypothesis.internal.conjecture.utilsr   hypothesis.internal.floatsr   r   r    hypothesis.internal.intervalsetsr   r   r   r*   r0   r6   r9   r"   r   r.   r%   rA   r<   r#   r=   rB   r   rC   rG   rS   r   r   r   r   r   r   r   r   r_   r   r   ru   r`   rp   rv   r   r   r'   r(   <module>r      su    B B !  - L 9 W W 8CL &y &	 y 
  3Y%-5 5  	 I  !!QRY R#Igfot344uWW=Ms=R7SS I 
 e$" " %" e$\Q \Q %\Q~   # c c :QK  	
 sCx  	:!! ! 	!
 #"! 
!W!H      s"G s"2D s" s"l`"`"(`"7I`"`"F(XW (X3E (X$ (XV;'* ;uZ_/E ;w : 6' 6G 6 6
$ % 
	BB+=B
8S=B,*(7+ * *r'   