U
    Þ@·f*  ã                   @   sp   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d„ Zdd	„ Zd
d„ Zdd„ Zdd„ ZdS )é    )ÚBasic)ÚInteger)ÚS)Útop_downÚsall)Údo_oneÚidentityc                 c   s   t | tƒr| d V  d S )Né   )Ú
isinstancer   ©Úx© r   úO/tmp/pip-unpacked-wheel-6uje5nh9/sympy/strategies/branch/tests/test_traverse.pyÚinc   s    
r   c                  C   sF   t tdƒtdƒƒ} t tdƒtdƒƒ}ttƒ}t|| ƒƒ|hksBt‚d S )Nr	   é   é   ©r   r   r   r   ÚsetÚAssertionError©ÚexprÚexpectedÚbrlr   r   r   Útest_top_down_easy   s    r   c                  C   s‚   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~t‚d S )Nr	   r   r   é   é   é   r   r   r   r   r   Útest_top_down_big_tree   s    00r   c                  C   s|   d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sxt‚d S )Nc                 s   s    | dkr| d V  | d V  d S )Nr   r	   r   r   r   r   r   Úsplit5   s    
z-test_top_down_harder_function.<locals>.split5r   r   r	   r   )r   r   r   r   r   )r   r   r   r   r   r   r   Útest_top_down_harder_function   s
    8r   c                  C   s®   t tdƒtdƒƒ} t tdƒtdƒƒ}ttƒ}t|| ƒƒ|gksBt‚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ªt‚d S )Nr	   r   r   r   )r   r   r   r   Úlistr   r   r   r   r   r   r   Ú	test_sall*   s    ""r!   N)Zsympy.core.basicr   Zsympy.core.numbersr   Zsympy.core.singletonr   Z sympy.strategies.branch.traverser   r   Zsympy.strategies.branch.corer   r   r   r   r   r   r!   r   r   r   r   Ú<module>   s   