def test_get_operators_by_level_complex(self):
exp = Expression(
"cos(x+Derivative(e**x,x)) + Derivative(x + x**(2*x+3), x)",
is_latex=False)
operators = exp.get_operators_by_level()
expected = {
0: [Add],
1: [cos, Derivative],
2: [Add, Add],
3: [Derivative, Pow],
4: [Pow, Add],
5: [Mul]
}
self.assertEquals(expected, operators)
def test_get_operators_by_level_one_level(self):
exp = Expression("x + 2", is_latex=False)
operators = exp.get_operators_by_level()
expected = {0: [Add]}
self.assertEquals(expected, operators)
def test_get_operators_by_level(self):
exp = Expression("x + Derivative(x + x**2, x)", is_latex=False)
operators = exp.get_operators_by_level()
expected = {0: [Add], 1: [Derivative], 2: [Add], 3: [Pow]}
self.assertEquals(expected, operators)
