plotly.py/tests/test_plotly_utils/validators/test_subplotid_validator.py at cef18a9d95d22614e5e660ba6722645defaf7fe1 · plotly/plotly.py · GitHub

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
111
import pytest
from _plotly_utils.basevalidators import SubplotidValidator
from ...test_optional.test_utils.test_utils import np_nan, np_inf


# Fixtures


@pytest.fixture()
def validator():
    return SubplotidValidator("prop", "parent", dflt="geo")


@pytest.fixture()
def validator_aok():
    return SubplotidValidator("prop", "parent", dflt="legend", array_ok=True)


# Tests

# Array not ok (default)

# Acceptance


@pytest.mark.parametrize("val", ["geo"] + ["geo%d" % i for i in range(2, 10)])
def test_acceptance(val, validator):
    assert validator.validate_coerce(val) == val


# Coercion from {base}1 to {base}
def test_coerce(validator):
    v = validator.validate_coerce("geo1")
    assert ("geo") == v


# Rejection by type
@pytest.mark.parametrize("val", [23, [], {}, set(), np_inf(), np_nan()])
def test_rejection_type(val, validator):
    with pytest.raises(ValueError) as validation_failure:
        validator.validate_coerce(val)

    assert "Invalid value" in str(validation_failure.value)


# Rejection by value
@pytest.mark.parametrize(
    "val",
    [
        "",  # Cannot be empty
        "bogus",  # Must begin with 'geo'
        "geo0",  # If followed by a number the number must be > 1
    ],
)
def test_rejection_value(val, validator):
    with pytest.raises(ValueError) as validation_failure:
        validator.validate_coerce(val)

    assert "Invalid value" in str(validation_failure.value)


# Array ok

# Acceptance


@pytest.mark.parametrize(
    "val",
    ["legend2", ["legend", "legend2"], ["legend", "legend2"]],
)
def test_acceptance_aok(val, validator_aok):
    v = validator_aok.validate_coerce(val)
    if isinstance(val, tuple):
        assert val == tuple(v)
    else:
        assert val == v


# Coercion from {base}1 to {base}
def test_coerce_aok(validator_aok):
    v = validator_aok.validate_coerce(("legend1", "legend2"))
    assert ("legend", "legend2") == tuple(v)


# Rejection by type
@pytest.mark.parametrize("val", [23, [2, 3], {}, set(), np_inf(), np_nan()])
def test_rejection_type_aok(val, validator_aok):
    with pytest.raises(ValueError) as validation_failure:
        validator_aok.validate_coerce(val)

    failure_msg = str(validation_failure.value)
    assert "Invalid value" in failure_msg or "Invalid elements" in failure_msg


# Rejection by value
@pytest.mark.parametrize(
    "val",
    [
        "",  # Cannot be empty
        "bogus",  # Must begin with 'geo'
        "legend0",  # If followed by a number the number must be > 1,
        ["", "legend"],
        ("bogus", "legend2"),
    ],
)
def test_rejection_value_aok(val, validator_aok):
    with pytest.raises(ValueError) as validation_failure:
        validator_aok.validate_coerce(val)

    failure_msg = str(validation_failure.value)
    assert "Invalid value" in failure_msg or "Invalid elements" in failure_msg