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
|
from __future__ import annotations
from typing import TYPE_CHECKING
import pytest
from contourpy import FillType, LineType, ZInterp
from contourpy.enum_util import as_fill_type, as_line_type, as_z_interp
from . import util_test
if TYPE_CHECKING:
from collections.abc import Callable
@pytest.mark.parametrize("name, value", util_test.all_fill_types_str_value())
def test_fill_type(name: str, value: int) -> None:
t = FillType.__members__[name]
assert t.name == name
assert t.value == value
@pytest.mark.parametrize("name, value", util_test.all_line_types_str_value())
def test_line_type(name: str, value: int) -> None:
t = LineType.__members__[name]
assert t.name == name
assert t.value == value
def test_all_fill_types() -> None:
# Check that all_fill_types() matches FillType.__members__
fill_types = dict(util_test.all_fill_types_str_value())
for name, enum in dict(FillType.__members__).items():
assert name in fill_types
assert fill_types[name] == enum.value
def test_all_line_types() -> None:
# Check that all_line_types() matches LineType.__members__
line_types = dict(util_test.all_line_types_str_value())
for name, enum in dict(LineType.__members__).items():
assert name in line_types
assert line_types[name] == enum.value
def test_all_z_interps() -> None:
# Check that all_z_interps() matches ZInterp.__members__
z_interps = dict(util_test.all_z_interps_str_value())
for name, enum in dict(ZInterp.__members__).items():
assert name in z_interps
assert z_interps[name] == enum.value
@pytest.mark.parametrize(
["enum_type", "from_string_function"],
[(FillType, as_fill_type), (LineType, as_line_type), (ZInterp, as_z_interp)])
def test_string_to_enum(
enum_type: FillType | LineType | ZInterp,
from_string_function: Callable[[FillType | LineType | ZInterp | str],
FillType | LineType | ZInterp],
) -> None:
for name, enum in enum_type.__members__.items():
line_type = from_string_function(name)
assert line_type == enum
msg = "'unknown' is not a valid"
with pytest.raises(ValueError, match=msg):
_ = from_string_function("unknown")
|
