angle aok tests

Author: nicolaskruchten

Diff for: packages/python/plotly/_plotly_utils/tests/validators/test_angle_validator.py

@@ -7,11 +7,16 @@
 
 # Fixtures
 # --------
-@pytest.fixture()
-def validator():
+@pytest.fixture
+def validator(request):
     return AngleValidator("prop", "parent")
 
 
+@pytest.fixture
+def validator_aok(request):
+    return AngleValidator("prop", "parent", array_ok=True)
+
+
 # Tests
 # -----
 # ### Test acceptance ###
@@ -36,3 +41,32 @@ def test_rejection(val, validator):
         validator.validate_coerce(val)
 
     assert "Invalid value" in str(validation_failure.value)
+
+
+# ### Test acceptance ###
+@pytest.mark.parametrize("val", [[0, 179, -179]])
+def test_aok_acceptance(val, validator_aok):
+    assert validator_aok.validate_coerce(val) == val
+    assert validator_aok.validate_coerce(tuple(val)) == val
+    assert np.array_equal(validator_aok.validate_coerce(np.array(val)), np.array(val))
+
+
+# ### Test coercion above 180 ###
+@pytest.mark.parametrize(
+    "val,expected",
+    [(180, -180), (181, -179), (-180.25, 179.75), (540, -180), (-541, 179)],
+)
+def test_aok_coercion(val, expected, validator_aok):
+    assert validator_aok.validate_coerce([val]) == [expected]
+    assert np.array_equal(
+        validator_aok.validate_coerce(np.array([val])), np.array([expected])
+    )
+
+
+# ### Test rejection ###
+@pytest.mark.parametrize("val", [["hello"], [()], [[]], [set()], ["34"]])
+def test_aok_rejection(val, validator_aok):
+    with pytest.raises(ValueError) as validation_failure:
+        validator_aok.validate_coerce(val)
+
+    assert "Invalid element(s)" in str(validation_failure.value)