21 lines
512 B
Python
21 lines
512 B
Python
|
import pytest
|
||
|
|
|
||
|
|
from styles import STYLES, get_style
|
||
|
|
from styles.blink import BlinkStyle
|
||
|
|
from styles.simple import SimpleStyle
|
||
|
|
|
||
|
|
|
||
|
|
def test_registry_has_blink_and_simple():
|
||
|
|
assert STYLES["blink"] is BlinkStyle
|
||
|
|
assert STYLES["simple"] is SimpleStyle
|
||
|
|
|
||
|
|
|
||
|
|
def test_get_style_returns_class():
|
||
|
|
assert get_style("blink") is BlinkStyle
|
||
|
|
assert get_style("simple") is SimpleStyle
|
||
|
|
|
||
|
|
|
||
|
|
def test_get_style_unknown_raises():
|
||
|
|
with pytest.raises(ValueError, match="Unknown alarm style"):
|
||
|
|
get_style("nope")
|