|
"""Tests for tools for manipulation of expressions using paths. """ |
|
|
|
from sympy.simplify.epathtools import epath, EPath |
|
from sympy.testing.pytest import raises |
|
|
|
from sympy.core.numbers import E |
|
from sympy.functions.elementary.trigonometric import (cos, sin) |
|
from sympy.abc import x, y, z, t |
|
|
|
|
|
def test_epath_select(): |
|
expr = [((x, 1, t), 2), ((3, y, 4), z)] |
|
|
|
assert epath("/*", expr) == [((x, 1, t), 2), ((3, y, 4), z)] |
|
assert epath("/*/*", expr) == [(x, 1, t), 2, (3, y, 4), z] |
|
assert epath("/*/*/*", expr) == [x, 1, t, 3, y, 4] |
|
assert epath("/*/*/*/*", expr) == [] |
|
|
|
assert epath("/[:]", expr) == [((x, 1, t), 2), ((3, y, 4), z)] |
|
assert epath("/[:]/[:]", expr) == [(x, 1, t), 2, (3, y, 4), z] |
|
assert epath("/[:]/[:]/[:]", expr) == [x, 1, t, 3, y, 4] |
|
assert epath("/[:]/[:]/[:]/[:]", expr) == [] |
|
|
|
assert epath("/*/[:]", expr) == [(x, 1, t), 2, (3, y, 4), z] |
|
|
|
assert epath("/*/[0]", expr) == [(x, 1, t), (3, y, 4)] |
|
assert epath("/*/[1]", expr) == [2, z] |
|
assert epath("/*/[2]", expr) == [] |
|
|
|
assert epath("/*/int", expr) == [2] |
|
assert epath("/*/Symbol", expr) == [z] |
|
assert epath("/*/tuple", expr) == [(x, 1, t), (3, y, 4)] |
|
assert epath("/*/__iter__?", expr) == [(x, 1, t), (3, y, 4)] |
|
|
|
assert epath("/*/int|tuple", expr) == [(x, 1, t), 2, (3, y, 4)] |
|
assert epath("/*/Symbol|tuple", expr) == [(x, 1, t), (3, y, 4), z] |
|
assert epath("/*/int|Symbol|tuple", expr) == [(x, 1, t), 2, (3, y, 4), z] |
|
|
|
assert epath("/*/int|__iter__?", expr) == [(x, 1, t), 2, (3, y, 4)] |
|
assert epath("/*/Symbol|__iter__?", expr) == [(x, 1, t), (3, y, 4), z] |
|
assert epath( |
|
"/*/int|Symbol|__iter__?", expr) == [(x, 1, t), 2, (3, y, 4), z] |
|
|
|
assert epath("/*/[0]/int", expr) == [1, 3, 4] |
|
assert epath("/*/[0]/Symbol", expr) == [x, t, y] |
|
|
|
assert epath("/*/[0]/int[1:]", expr) == [1, 4] |
|
assert epath("/*/[0]/Symbol[1:]", expr) == [t, y] |
|
|
|
assert epath("/Symbol", x + y + z + 1) == [x, y, z] |
|
assert epath("/*/*/Symbol", t + sin(x + 1) + cos(x + y + E)) == [x, x, y] |
|
|
|
|
|
def test_epath_apply(): |
|
expr = [((x, 1, t), 2), ((3, y, 4), z)] |
|
func = lambda expr: expr**2 |
|
|
|
assert epath("/*", expr, list) == [[(x, 1, t), 2], [(3, y, 4), z]] |
|
|
|
assert epath("/*/[0]", expr, list) == [([x, 1, t], 2), ([3, y, 4], z)] |
|
assert epath("/*/[1]", expr, func) == [((x, 1, t), 4), ((3, y, 4), z**2)] |
|
assert epath("/*/[2]", expr, list) == expr |
|
|
|
assert epath("/*/[0]/int", expr, func) == [((x, 1, t), 2), ((9, y, 16), z)] |
|
assert epath("/*/[0]/Symbol", expr, func) == [((x**2, 1, t**2), 2), |
|
((3, y**2, 4), z)] |
|
assert epath( |
|
"/*/[0]/int[1:]", expr, func) == [((x, 1, t), 2), ((3, y, 16), z)] |
|
assert epath("/*/[0]/Symbol[1:]", expr, func) == [((x, 1, t**2), |
|
2), ((3, y**2, 4), z)] |
|
|
|
assert epath("/Symbol", x + y + z + 1, func) == x**2 + y**2 + z**2 + 1 |
|
assert epath("/*/*/Symbol", t + sin(x + 1) + cos(x + y + E), func) == \ |
|
t + sin(x**2 + 1) + cos(x**2 + y**2 + E) |
|
|
|
|
|
def test_EPath(): |
|
assert EPath("/*/[0]")._path == "/*/[0]" |
|
assert EPath(EPath("/*/[0]"))._path == "/*/[0]" |
|
assert isinstance(epath("/*/[0]"), EPath) is True |
|
|
|
assert repr(EPath("/*/[0]")) == "EPath('/*/[0]')" |
|
|
|
raises(ValueError, lambda: EPath("")) |
|
raises(ValueError, lambda: EPath("/")) |
|
raises(ValueError, lambda: EPath("/|x")) |
|
raises(ValueError, lambda: EPath("/[")) |
|
raises(ValueError, lambda: EPath("/[0]%")) |
|
|
|
raises(NotImplementedError, lambda: EPath("Symbol")) |
|
|