Python epathの例

コード例 #1

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]')"

    pytest.raises(ValueError, lambda: EPath(""))
    pytest.raises(ValueError, lambda: EPath("/"))
    pytest.raises(ValueError, lambda: EPath("/|x"))
    pytest.raises(ValueError, lambda: EPath("/["))
    pytest.raises(ValueError, lambda: EPath("/[0]%"))

    pytest.raises(NotImplementedError, lambda: EPath("Symbol"))

コード例 #2

ファイル: test_epathtools.py プロジェクト: skirpichev/diofant

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]')"

    pytest.raises(ValueError, lambda: EPath(""))
    pytest.raises(ValueError, lambda: EPath("/"))
    pytest.raises(ValueError, lambda: EPath("/|x"))
    pytest.raises(ValueError, lambda: EPath("/["))
    pytest.raises(ValueError, lambda: EPath("/[0]%"))

    pytest.raises(NotImplementedError, lambda: EPath("Symbol"))

コード例 #3

def test_epath_apply():
    expr = [((x, 1, t), 2), ((3, y, 4), z)]

    def func(expr):
        return 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)

コード例 #4

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]')"

    pytest.raises(ValueError, lambda: EPath(''))
    pytest.raises(ValueError, lambda: EPath('/'))
    pytest.raises(ValueError, lambda: EPath('/|x'))
    pytest.raises(ValueError, lambda: EPath('/['))
    pytest.raises(ValueError, lambda: EPath('/[0]%'))

    pytest.raises(NotImplementedError, lambda: EPath('Symbol'))

コード例 #5

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]

コード例 #6

ファイル: test_epathtools.py プロジェクト: skirpichev/diofant

def test_epath_apply():
    expr = [((x, 1, t), 2), ((3, y, 4), z)]

    def func(expr):
        return 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)
    assert epath("/*/*/Symbol", t*sin(x + 1)*cos(x + y + E), func) == \
        t*sin(x**2 + 1)*cos(x**2 + y**2 + E)
    assert epath("/*/*/Symbol", [((x, 1), 2), ((3, y), z)], func) == \
        [((x**2, 1), 2), ((3, y**2), z)]

コード例 #7

ファイル: test_epathtools.py プロジェクト: skirpichev/diofant

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]

コード例 #8

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]