<s> """<STR_LIT>""" <EOL> from sympy . simplify . traversaltools import use <EOL> from sympy import expand , factor , I <EOL> from sympy . abc import x , y <EOL> def test_use ( ) : <EOL> assert use ( <NUM_LIT:0> , expand ) == <NUM_LIT:0> <EOL> f = ( x + y ) ** <NUM_LIT:2> * x + <NUM_LIT:1> <EOL> assert use ( f , expand , level = <NUM_LIT:0> ) == x ** <NUM_LIT:3> + <NUM_LIT:2> * x ** <NUM_LIT:2> * y + x * y ** <NUM_LIT:2> + + <NUM_LIT:1> <EOL> assert use ( f , expand , level = <NUM_LIT:1> ) == x ** <NUM_LIT:3> + <NUM_LIT:2> * x ** <NUM_LIT:2> * y + x * y ** <NUM_LIT:2> + + <NUM_LIT:1> <EOL> assert use ( f , expand , level = <NUM_LIT:2> ) == <NUM_LIT:1> + x * ( <NUM_LIT:2> * x * y + x ** <NUM_LIT:2> + y ** <NUM_LIT:2> ) <EOL> assert use ( f , expand , level = <NUM_LIT:3> ) == ( x + y ) ** <NUM_LIT:2> * x + <NUM_LIT:1> <EOL> f = ( x ** <NUM_LIT:2> + <NUM_LIT:1> ) ** <NUM_LIT:2> - <NUM_LIT:1> <EOL> kwargs = { '<STR_LIT>' : True } <EOL> assert use ( f , factor , level = <NUM_LIT:0> , kwargs = kwargs ) == x ** <NUM_LIT:2> * ( x ** <NUM_LIT:2> + <NUM_LIT:2> ) <EOL> assert use ( f , factor , level = <NUM_LIT:1> , kwargs = kwargs ) == ( x + I ) ** <NUM_LIT:2> * ( x - I ) ** <NUM_LIT:2> - <NUM_LIT:1> <EOL> assert use ( f , factor , level = <NUM_LIT:2> , kwargs = kwargs ) == ( x + I ) ** <NUM_LIT:2> * ( x - I ) ** <NUM_LIT:2> - <NUM_LIT:1> <EOL> assert use ( f , factor , level = <NUM_LIT:3> , kwargs = kwargs ) == ( x ** <NUM_LIT:2> + <NUM_LIT:1> ) ** <NUM_LIT:2> - <NUM_LIT:1> </s>
