
    g                         d dl mZ d dlmZmZmZmZmZmZm	Z	m
Z
 d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d Zd	 Zd
 Zd Zd Zd Zd Zd Zd ZdS )    )S)rm_idglomflattenunpacksort
distributesubsrebuild)Basic)Add)Mul)symbolsxc                  4   t          d           }  | t          t          d          t          d                              t          t          d                    k    sJ  | t          t          d          t          d                              t          t          d                    k    sJ  | t          t          d          t          d                              t          t          d          t          d                    k    sJ d S )Nc                     | dk    S Nr    r   s    Z/var/www/html/ai-engine/env/lib/python3.11/site-packages/sympy/strategies/tests/test_rl.py<lambda>ztest_rm_id.<locals>.<lambda>   s
    a1f     r         )r   r   r   )rmzeross    r   
test_rm_idr      s    $$%%G751qtt$$%%qtt4444751qtt$$%%qtt4444751qtt$$%%qttQqTT):):::::::r   c            
      $   d } d }d }t          | ||          } |t          t          t           dt          z  ddd                    }t          dt          z  d          }t          |j                  t          |j                  k    sJ d S )	Nc                 ,     | j                     d         S )Nr   as_coeff_Mulr   s    r   keyztest_glom.<locals>.key       q~""r   c                 ,     | j                     d         S r   r   r   s    r   countztest_glom.<locals>.count   r"   r   c                     | |z  S )Nr   )cntargs     r   newargsztest_glom.<locals>.newargs   s    Syr      r   Fevaluate   )r   r   r   setargs)r!   r$   r(   rlresultexpecteds         r   	test_glomr2      s    # # ## # #   
c5'	"	"BRAr1q5!Q77788F1q5!}}Hv{s8=11111111r   c                  Z   t          t          t          d          t          d          t          t          d          t          d                                        t          t          d          t          d          t          d          t          d                    k    sJ d S )Nr   r   r)      )r   r   r   r   r   r   test_flattenr5   #   s    51qttU1Q441%6%67788addAaDD!A$$!%%& & & & & &r   c                  .   t          t          t          d                              dk    sJ t          t          t          d          t          d                              t          t          d          t          d                    k    sJ d S )Nr   r)   )r   r   r   r   r   r   test_unpackr7   (   sr    %!++!####%!add##$$addAaDD(9(9999999r   c            	      $    t          t                    t          t          d          t          d          t          d                              t          t          d          t          d          t          d                    k    sJ d S )Nr)   r   r   )r   strr   r   r   r   r   	test_sortr:   -   se    499U1Q441qtt,,--qttQqTT1Q441H1HHHHHHHr   c                  @    G d dt                     }  G d dt                     }t          | |          } | | t          d          t          d           |t          d          t          d                    t          d	                               | | t          d          t          d          t          d          t          d	                     | t          d          t          d          t          d          t          d	                              k    sJ  | | t          d          t          d          t          d                               | t          d          t          d          t          d                    k    sJ d S )
Nc                       e Zd ZdS )test_distribute.<locals>.T1N__name__
__module____qualname__r   r   r   T1r=   2           r   rB   c                       e Zd ZdS )test_distribute.<locals>.T2Nr>   r   r   r   T2rE   5   rC   r   rF   r   r   r)   r4   r,   )r   r	   r   )rB   rF   distribute_t12s      r   test_distributerH   1   s       U       U     B''N>""QqTT1Q44AaDD!A$$1>>??
22addAaDD!A$$!%%rr!A$$!addAaDD'A'ABBC C C C>""QqTT1Q441..//22addAaDD!A$$3G3GGGGGGGr   c                     t          d          \  } }t          dt          | |          d          }t          t          d|           t          d|                    }t          t          t                    } ||          |k    sJ d S )Nzx, yr   Fr*   )r   r   r   r	   )r   yexprr1   distribute_muls        r   test_distribute_add_mulrM   >   s|    6??DAqq#a))e,,,D3q!99c!Qii((HS))N>$8++++++r   c                  j    t          dd          }  | d          dk    sJ  | d          dk    sJ d S Nr   r   r)   )r
   )r/   s    r   	test_subsrP   F   s=    	aB2a55A::::2a55A::::::r   c                      t          j        t          t          d          t          d                    } t	          |           dk    sJ d S rO   )r   __new__r   r   r   )rK   s    r   test_rebuildrS   L   s>    =addAaDD))D4==Ar   N)sympy.core.singletonr   sympy.strategies.rlr   r   r   r   r   r	   r
   r   sympy.core.basicr   sympy.core.addr   sympy.core.mulr   sympy.core.symbolr   	sympy.abcr   r   r2   r5   r7   r:   rH   rM   rP   rS   r   r   r   <module>r[      s   " " " " " "C C C C C C C C C C C C C C C C C C C C " " " " " "             % % % % % %      ; ; ;2 2 2"& & &
: : :
I I I
H 
H 
H, , ,      r   