test_angle_validator.py

import pytest
from _plotly_utils.basevalidators import AngleValidator
import numpy as np


# Fixtures
# --------
@pytest.fixture()
def validator():
    return AngleValidator('prop', 'parent')


# Tests
# -----
# ### Test acceptance ###
@pytest.mark.parametrize('val', [0] + list(np.linspace(-180, 179.99)))
def test_acceptance(val, validator):
    assert validator.validate_coerce(val) == val


# ### Test coercion above 180 ###
@pytest.mark.parametrize('val,expected', [
    (180, -180),
    (181, -179),
    (-180.25, 179.75),
    (540, -180),
    (-541, 179)
])
def test_coercion(val, expected, validator):
    assert validator.validate_coerce(val) == expected


# ### Test rejection ###
@pytest.mark.parametrize('val',
                         ['hello', (), [], [1, 2, 3], set(), '34'])
def test_rejection(val, validator: AngleValidator):
    with pytest.raises(ValueError) as validation_failure:
        validator.validate_coerce(val)

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