ó
Ąź\c           @   sm   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(
   i˙˙˙˙(   t   Basic(   t   top_downt   sall(   t   do_onet   identityc         c   s   t  |  t  r |  d Vn  d  S(   Ni   (   t
   isinstancet   int(   t   x(    (    sJ   lib/python2.7/site-packages/sympy/strategies/branch/tests/test_traverse.pyt   inc   s    c          C   sO   t  d d  }  t  d d  } t t  } t | |    | h k sK t  d  S(   Ni   i   i   (   R    R   R   t   sett   AssertionError(   t   exprt   expectedt   brl(    (    sJ   lib/python2.7/site-packages/sympy/strategies/branch/tests/test_traverse.pyt   test_top_down_easy	   s    c          C   s   t  d t  d  t  d t  d  d   }  t  d t  d  t  d t  d  d   } t t  } t | |    | h k s t  d  S(   Ni   i   i   i   i   i   (   R    R   R   R	   R
   (   R   R   R   (    (    sJ   lib/python2.7/site-packages/sympy/strategies/branch/tests/test_traverse.pyt   test_top_down_big_tree   s    **c          C   s   d   }  t  t  d d  d  } t  t  d d  d  t  t  d d  d  h } t |   } t | |   | k s{ t  d  S(   Nc         s   s%   |  d k r! |  d V|  d Vn  d  S(   Ni   i   (    (   R   (    (    sJ   lib/python2.7/site-packages/sympy/strategies/branch/tests/test_traverse.pyt   split5   s    	i   i   i   i   (   R    R   R	   R
   (   R   R   R   R   (    (    sJ   lib/python2.7/site-packages/sympy/strategies/branch/tests/test_traverse.pyt   test_top_down_harder_function   s
    	0c          C   sť   t  d d  }  t  d d  } t t  } t | |    | g k sK t  t  d d t  d d   }  t  d d t  d d   } t t t t   } t | |    | g k sˇ t  d  S(   Ni   i   i   i   (   R    R   R   t   listR
   R   R   (   R   R   R   (    (    sJ   lib/python2.7/site-packages/sympy/strategies/branch/tests/test_traverse.pyt	   test_sall#   s    !N(   t   sympyR    t    sympy.strategies.branch.traverseR   R   t   sympy.strategies.branch.coreR   R   R   R   R   R   R   (    (    (    sJ   lib/python2.7/site-packages/sympy/strategies/branch/tests/test_traverse.pyt   <module>   s   				