Onuralp SEZER
bb4d741aff
This commit adds the WHITE, BLACK, RED, GREEN, and BLUE color constants to the Color class in color.py. These constants represent the corresponding colors in the RGB color space. Signed-off-by: Onuralp SEZER <thunderbirdtr@gmail.com>
52 lines
1.6 KiB
Python
52 lines
1.6 KiB
Python
from contextlib import ExitStack as DoesNotRaise
|
|
from typing import Optional
|
|
|
|
import pytest
|
|
|
|
from supervision.draw.color import Color
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"color_hex, expected_result, exception",
|
|
[
|
|
("fff", Color.WHITE, DoesNotRaise()),
|
|
("#fff", Color.WHITE, DoesNotRaise()),
|
|
("ffffff", Color.WHITE, DoesNotRaise()),
|
|
("#ffffff", Color.WHITE, DoesNotRaise()),
|
|
("f00", Color.RED, DoesNotRaise()),
|
|
("0f0", Color.GREEN, DoesNotRaise()),
|
|
("00f", Color.BLUE, DoesNotRaise()),
|
|
("#808000", Color(r=128, g=128, b=0), DoesNotRaise()),
|
|
("", None, pytest.raises(ValueError)),
|
|
("00", None, pytest.raises(ValueError)),
|
|
("0000", None, pytest.raises(ValueError)),
|
|
("0000000", None, pytest.raises(ValueError)),
|
|
("ffg", None, pytest.raises(ValueError)),
|
|
],
|
|
)
|
|
def test_color_from_hex(
|
|
color_hex, expected_result: Optional[Color], exception: Exception
|
|
) -> None:
|
|
with exception:
|
|
result = Color.from_hex(color_hex=color_hex)
|
|
assert result == expected_result
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"color, expected_result, exception",
|
|
[
|
|
(Color.WHITE, "#ffffff", DoesNotRaise()),
|
|
(Color.BLACK, "#000000", DoesNotRaise()),
|
|
(Color.RED, "#ff0000", DoesNotRaise()),
|
|
(Color.GREEN, "#00ff00", DoesNotRaise()),
|
|
(Color.BLUE, "#0000ff", DoesNotRaise()),
|
|
(Color(r=128, g=128, b=0), "#808000", DoesNotRaise()),
|
|
],
|
|
)
|
|
def test_color_as_hex(
|
|
color: Color, expected_result: Optional[str], exception: Exception
|
|
) -> None:
|
|
with exception:
|
|
result = color.as_hex()
|
|
assert result == expected_result
|