1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
|
import pytest
from deepdiff.path import _path_to_elements, GET, GETATTR, extract, parse_path, stringify_path, _add_to_elements
@pytest.mark.parametrize('test_num, path, expected', [
(1, "root[4]['b'][3]", [(4, GET), ('b', GET), (3, GET)]),
(2, "root[4].b[3]", [(4, GET), ('b', GETATTR), (3, GET)]),
(3, "root[4].b['a3']", [(4, GET), ('b', GETATTR), ('a3', GET)]),
(4, "root[4.3].b['a3']", [(4.3, GET), ('b', GETATTR), ('a3', GET)]),
(5, "root.a.b", [('a', GETATTR), ('b', GETATTR)]),
(6, "root.hello", [('hello', GETATTR)]),
(7, "root['h']", [('h', GET)]),
(8, "root['a\rb']", [('a\rb', GET)]),
(9, "root['a\\rb']", [('a\\rb', GET)]),
(10, "root", []),
(11, ((4, GET), ('b', GET)), ((4, GET), ('b', GET))),
])
def test_path_to_elements(test_num, path, expected):
result = _path_to_elements(path, root_element=None)
assert tuple(expected) == result, f"test_path_to_elements #{test_num} failed"
if isinstance(path, str):
path_again = stringify_path(path=result)
assert path == path_again, f"test_path_to_elements #{test_num} failed"
@pytest.mark.parametrize('obj, path, expected', [
({1: [2, 3], 2: [4, 5]},
"root[2][1]",
5),
({1: [{'2': 'b'}, 3], 2: {4, 5}},
"root[1][0]['2']",
'b'
),
({'test [a]': 'b'},
"root['test [a]']",
'b'
),
({"a']['b']['c": 1},
"""root["a']['b']['c"]""",
1
),
])
def test_get_item(obj, path, expected):
result = extract(obj, path)
assert expected == result
def test_parse_path():
result = parse_path("root[1][2]['age']")
assert [1, 2, 'age'] == result
result2 = parse_path("root[1][2]['age']", include_actions=True)
assert [{'element': 1, 'action': 'GET'}, {'element': 2, 'action': 'GET'}, {'element': 'age', 'action': 'GET'}] == result2
result3 = parse_path("root['joe'].age")
assert ['joe', 'age'] == result3
result4 = parse_path("root['joe'].age", include_actions=True)
assert [{'element': 'joe', 'action': 'GET'}, {'element': 'age', 'action': 'GETATTR'}] == result4
@pytest.mark.parametrize('test_num, elem, inside, expected', [
(
1,
"'hello'",
None,
[('hello', GET)],
),
(
2,
"'a\rb'",
None,
[('a\rb', GET)],
),
])
def test__add_to_elements(test_num, elem, inside, expected):
elements = []
_add_to_elements(elements, elem, inside)
assert expected == elements
|
