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
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
|
# coding: utf-8
from typing import Any
import pytest # type: ignore # NOQA
from roundtrip import ( # type: ignore # NOQA
dedent,
round_trip,
round_trip_dump,
round_trip_load,
)
def load(s: str) -> Any:
return round_trip_load(dedent(s))
class TestLineCol:
def test_item_00(self) -> None:
data = load(
"""
- a
- e
- [b, d]
- c
"""
)
assert data[2].lc.line == 2
assert data[2].lc.col == 2
def test_item_01(self) -> None:
data = load(
"""
- a
- e
- {x: 3}
- c
"""
)
assert data[2].lc.line == 2
assert data[2].lc.col == 2
def test_item_02(self) -> None:
data = load(
"""
- a
- e
- !!set {x, y}
- c
"""
)
assert data[2].lc.line == 2
assert data[2].lc.col == 2
def test_item_03(self) -> None:
data = load(
"""
- a
- e
- !!omap
- x: 1
- y: 3
- c
"""
)
assert data[2].lc.line == 2
assert data[2].lc.col == 2
def test_item_04(self) -> None:
data = load(
"""
# testing line and column based on SO
# http://stackoverflow.com/questions/13319067/
- key1: item 1
key2: item 2
- key3: another item 1
key4: another item 2
"""
)
assert data[0].lc.line == 2
assert data[0].lc.col == 2
assert data[1].lc.line == 4
assert data[1].lc.col == 2
def test_pos_mapping(self) -> None:
data = load(
"""
a: 1
b: 2
c: 3
# comment
klm: 42
d: 4
"""
)
assert data.lc.key('klm') == (4, 0)
assert data.lc.value('klm') == (4, 5)
def test_pos_sequence(self) -> None:
data = load(
"""
- a
- b
- c
# next one!
- klm
- d
"""
)
assert data.lc.item(3) == (4, 2)
|
