aboutsummaryrefslogtreecommitdiff
path: root/.venv/lib/python3.12/site-packages/orgparse/tests/data/00_simple.py
blob: c0b23d1d9375e5c271b64e7f2bfe938180bc825d (about) (plain)
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
from typing import Any, Dict, Set


def nodedict(i, level, todo=None, shallow_tags=set([]), tags=set([])) -> Dict[str, Any]:
    return dict(
        heading="Heading {0}".format(i),
        level=level,
        todo=todo,
        shallow_tags=shallow_tags,
        tags=tags,
    )


def tags(nums) -> Set[str]:
    return set(map('TAG{0}'.format, nums))


data = [
    nodedict(i, *vals) for (i, vals) in enumerate([  # type: ignore[misc]
        [1, 'TODO1', tags([1])   , tags(range(1, 2))],
        [2, 'TODO2', tags([2])   , tags(range(1, 3))],
        [3, 'TODO3', tags([3])   , tags(range(1, 4))],
        [4, 'TODO4', tags([4])   , tags(range(1, 5))],
        [2, None   , tags([])    , tags([1])        ],
        [2, None   , tags([])    , tags([1])        ],
        [1, None   , tags([2])   , tags([2])        ],
        [2, None   , tags([2])   , tags([2])        ],
        [3, None   , tags([])    , tags([2])        ],
        [5, None   , tags([3, 4]), tags([2, 3, 4])  ],
        [4, None   , tags([1])   , tags([1, 2])     ],
        [2, None   , tags([])    , tags([2])        ],
        [1],
    ])]