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
|
import pytest
from _plotly_utils.basevalidators import AnyValidator
import numpy as np
# Fixtures
# --------
@pytest.fixture()
def validator():
return AnyValidator("prop", "parent")
@pytest.fixture()
def validator_aok():
return AnyValidator("prop", "parent", array_ok=True)
# Tests
# -----
# ### Acceptance ###
@pytest.mark.parametrize("val", [set(), "Hello", 123, np.inf, np.nan, {}])
def test_acceptance(val, validator):
assert validator.validate_coerce(val) is val
# ### Acceptance of arrays ###
@pytest.mark.parametrize(
"val",
[
23,
"Hello!",
[],
(),
np.array([]),
("Hello", "World"),
["Hello", "World"],
[np.pi, np.e, {}],
],
)
def test_acceptance_array(val, validator_aok):
coerce_val = validator_aok.validate_coerce(val)
if isinstance(val, np.ndarray):
assert isinstance(coerce_val, np.ndarray)
assert coerce_val.dtype == "object"
assert np.array_equal(coerce_val, val)
elif isinstance(val, (list, tuple)):
assert coerce_val == list(val)
assert validator_aok.present(coerce_val) == tuple(val)
else:
assert coerce_val == val
assert validator_aok.present(coerce_val) == val
|
