
    Qi                     \    d dl mZmZ d dlmZmZ d dlmZ d dlm	Z	 d dl
mZ d Zd Zd Zd	S )
    )SRational)schur_partitionSchurNumber)_randint)raises)symbolsc                 0    | D ]}| D ]}||z   | v du sJ dS )z`
    Checks if subset is sum-free(There are no x,y,z in the subset such that
    x + y = z)
    FN )subsetijs      /var/www/development/aibuddy-work/election-extract/venv/lib/python3.11/site-packages/sympy/combinatorics/tests/test_schur_number.py_sum_free_testr      sM    
  . . 	. 	.AEVO-----	.. .    c                  ,   t          t          d            t          t          d            t          t          d            t          d          ddggk    sJ t          d          } t	          d          D ]r} | dd          }t          |          }d}g }|D ]F}t          |           	 |t          |          z  }|D ]}||v d	u sJ |                    |            G||k    sJ st          d
          t          t          fd           d S )Nc                  4    t          t          j                  S N)r   r   Infinityr   r   r   <lambda>z&test_schur_partition.<locals>.<lambda>   s    qz:: r   c                       t          d          S )Nr   r   r   r   r   z&test_schur_partition.<locals>.<lambda>   s    r22 r   c                       t          d          S )Nr   r   r   r   r   r   z&test_schur_partition.<locals>.<lambda>   s    q11 r         i     r   Fxc                  "    t                     S r   r   )r   s   r   r   z&test_schur_partition.<locals>.<lambda>*   s    q11 r   )	r   
ValueErrorr   r   ranger   lenappendr	   )	random_number_generator_nresulttnumbersitemlr   s	           @r   test_schur_partitionr,      sU   
:::;;;
:22333
:1122211a&))))&tnn1XX  ##At,, ## 	" 	"D4    TNA " "W....q!!!!" AvvvvvA
:111122222r   c                     dddddd} | D ]}t          |          | |         k    sJ t          t          j                  t          j        k    sJ t          d          dk    sJ t          t          d            t          d	          }t          |                                          d
|z  dz  t          dd          z
  k    sJ t          d                                          dk    sJ d S )Nr         ,      )r   r      r.   r   r   c                       t          d          S )Ng      ?)r   r   r   r   r   z#test_schur_number.<locals>.<lambda>3   s    {3// r   r&   r2   r      i  )r   r   r   r   r    r	   lower_boundr   )first_known_schur_numberskr&   s      r   test_schur_numberr8   ,   s    $%!rc B B& > >1~~!:1!======qz""aj0000q>>Q
://000Aq>>%%''1a46HQNN+BBBBBq>>%%''4//////r   N)
sympy.corer   r    sympy.combinatorics.schur_numberr   r   sympy.core.randomr   sympy.testing.pytestr   sympy.core.symbolr	   r   r,   r8   r   r   r   <module>r>      s    " " " " " " " " I I I I I I I I & & & & & & ' ' ' ' ' ' % % % % % %. . .3 3 340 0 0 0 0r   