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
|
# SPDX-FileCopyrightText: Peter Pentchev <roam@ringlet.net>
# SPDX-License-Identifier: BSD-2-Clause
"""Test the evaluation of some simple expressions."""
from __future__ import annotations
import dataclasses
from typing import NamedTuple
import pytest
import parse_stages as pst
@dataclasses.dataclass(frozen=True)
class Environment(pst.TaggedFrozen):
"""Specify an environment to be matched by a spec (or not)."""
class Case(NamedTuple):
"""Specify a single test case: a string to parse, results."""
spec: str
matched: list[str]
_ALL = [
Environment(name="t-black", tags=["check"]),
Environment(name="t-black-reformat", tags=["do", "reformat"]),
Environment(name="t-pep8", tags=["check"]),
Environment(name="t-mypy", tags=["check"]),
Environment(name="t-pylint", tags=["check"]),
Environment(name="t-unit-tests", tags=["tests"]),
Environment(name="t-runner-pep8", tags=["check", "runner"]),
]
_TESTS = [
Case(spec="@check", matched=["t-black", "t-pep8", "t-mypy", "t-pylint", "t-runner-pep8"]),
Case(spec="@tests", matched=["t-unit-tests"]),
Case(spec="@check and not pep8", matched=["t-black", "t-mypy", "t-pylint"]),
Case(spec="not pep8 and @check", matched=["t-black", "t-mypy", "t-pylint"]),
Case(spec="@check and pep8 or @tests", matched=["t-pep8", "t-unit-tests", "t-runner-pep8"]),
Case(spec="black", matched=["t-black", "t-black-reformat"]),
Case(spec="black and not black-reformat", matched=["t-black"]),
Case(spec="black and not black-reformat", matched=["t-black"]),
Case(spec="(black or @runner) and @check", matched=["t-black", "t-runner-pep8"]),
Case(spec="not (@check or @tests)", matched=["t-black-reformat"]),
]
@pytest.mark.parametrize("case", _TESTS)
def test_basic(case: Case) -> None:
"""Make sure evaluation works more or less correctly."""
expr = pst.parse_spec(case.spec)
assert [env.name for env in _ALL if expr.evaluate(env)] == case.matched
|
