
    g*                     h    d dl mZ d dlmZ d dlmZ d dlmZmZ d dl	m
Z
mZ d Zd Zd Zd	 Zd
 ZdS )    )Basic)Integer)S)top_downsall)do_oneidentityc              #   F   K   t          | t                    r	| dz   V  d S d S )N   )
isinstancer   xs    g/var/www/html/ai-engine/env/lib/python3.11/site-packages/sympy/strategies/branch/tests/test_traverse.pyincr      s3      !W !e     c                     t          t          d          t          d                    } t          t          d          t          d                    }t          t                    }t	           ||                     |hk    sJ d S )Nr         r   r   r   r   setexprexpectedbrls      r   test_top_down_easyr      sj    1qttDQqTT1Q44  H
3--Css4yy>>hZ''''''r   c                  X   t          t          d          t          t          d                    t          t          d          t          t          d                    t          d                              } t          t          d          t          t          d                    t          t          d          t          t          d                    t          d                              }t          t                    }t	           ||                     |hk    sJ d S )Nr   r   r            r   r   s      r   test_top_down_big_treer       s    1uQqTT{{E!A$$addQqTT$B$BCCDQqTT51;;addE!A$$KK1(F(FGGH
3--Css4yy>>hZ''''''r   c            	         d } t          t          t          d          t          d                    t          d                    }t          t          t          d          t          d                    t          d                    t          t          t          d          t          d                    t          d                    h}t          |           }t           ||                    |k    sJ d S )Nc              3   6   K   | dk    r| dz
  V  | dz   V  d S d S )Nr   r    r   s    r   split5z-test_top_down_harder_function.<locals>.split5   s4      66a%KKKa%KKKKK 6r   r   r   r   r   )r   r   r   r   )r$   r   r   r   s       r   test_top_down_harder_functionr%      s      
 qttQqTT""AaDD))DeAaDD!A$$''1..eAaDD!A$$6G6G10N0NOH
6

Css4yy>>X%%%%%%r   c            
         t          t          d          t          d                    } t          t          d          t          d                    }t          t                    }t	           ||                     |gk    sJ t          t          d          t          d          t          t          d          t          d                              } t          t          d          t          d          t          t          d          t          d                              }t          t          t          t                              }t	           ||                     |gk    sJ d S )Nr   r   r   r   )r   r   r   r   listr   r	   r   s      r   	test_sallr(   *   s   1qttDQqTT1Q44  H
s))CD		??xj((((1qttU1Q441..//DQqTT1Q44qttQqTT!2!233H
vc8$$
%
%CD		??xj((((((r   N)sympy.core.basicr   sympy.core.numbersr   sympy.core.singletonr    sympy.strategies.branch.traverser   r   sympy.strategies.branch.corer   r	   r   r   r    r%   r(   r#   r   r   <module>r.      s    " " " " " " & & & & & & " " " " " " ; ; ; ; ; ; ; ; 9 9 9 9 9 9 9 9  
( ( (( ( (
& 
& 
&) ) ) ) )r   