diff --git a/tests/test_paulmann.py b/tests/test_paulmann.py index 66523e7062..baac77935e 100644 --- a/tests/test_paulmann.py +++ b/tests/test_paulmann.py @@ -1,9 +1,12 @@ """Tests for Paulmann quirks.""" +import pytest + import zhaquirks.paulmann.fourbtnremote -def test_fourbtnremote_signature(assert_signature_matches_quirk): +@pytest.mark.parametrize("manufacturer", ("Paulmann LichtGmbH", "Paulmann Licht GmbH")) +def test_fourbtnremote_signature(assert_signature_matches_quirk, manufacturer): """Test PaulmannRemote4Btn signature is matched to its quirk.""" signature = { "node_descriptor": "NodeDescriptor(logical_type=, complex_descriptor_available=0, user_descriptor_available=0, reserved=0, aps_flags=0, frequency_band=, mac_capability_flags=, manufacturer_code=4644, maximum_buffer_size=82, maximum_incoming_transfer_size=82, server_mask=11264, maximum_outgoing_transfer_size=82, descriptor_capability_field=, *allocate_address=True, *is_alternate_pan_coordinator=False, *is_coordinator=False, *is_end_device=True, *is_full_function_device=False, *is_mains_powered=False, *is_receiver_on_when_idle=False, *is_router=False, *is_security_capable=False)", @@ -39,7 +42,7 @@ def test_fourbtnremote_signature(assert_signature_matches_quirk): ], }, }, - "manufacturer": "Paulmann LichtGmbH", + "manufacturer": manufacturer, "model": "501.34", "class": "paulmann.fourbtnremote.PaulmannRemote4Btn", } diff --git a/zhaquirks/paulmann/__init__.py b/zhaquirks/paulmann/__init__.py index 1d88c6f9de..de910e7c0e 100644 --- a/zhaquirks/paulmann/__init__.py +++ b/zhaquirks/paulmann/__init__.py @@ -1,3 +1,4 @@ """Paulmann module.""" PAULMANN = "Paulmann LichtGmbH" +PAULMANN_VARIANT = "Paulmann Licht GmbH" diff --git a/zhaquirks/paulmann/fourbtnremote.py b/zhaquirks/paulmann/fourbtnremote.py index 0c4c6f4c24..c5005515c4 100644 --- a/zhaquirks/paulmann/fourbtnremote.py +++ b/zhaquirks/paulmann/fourbtnremote.py @@ -38,7 +38,7 @@ PROFILE_ID, SHORT_PRESS, ) -from zhaquirks.paulmann import PAULMANN +from zhaquirks.paulmann import PAULMANN, PAULMANN_VARIANT class PaulmannRemote4Btn(CustomDevice): @@ -49,7 +49,7 @@ class PaulmannRemote4Btn(CustomDevice): # device_version=0 # input_clusters=[0, 1, 3, 2821, 4096] # output_clusters=[3, 4, 5, 6, 8, 25, 768, 4096]> - MODELS_INFO: [(PAULMANN, "501.34")], + MODELS_INFO: [(PAULMANN, "501.34"), (PAULMANN_VARIANT, "501.34")], ENDPOINTS: { 1: { PROFILE_ID: zha.PROFILE_ID,