clean up

Author: dmulcahey

tests/common.py

@@ -548,12 +548,3 @@ def create_mock_zigpy_device(
                     cluster._attr_cache[attr_id] = value
 
     return device
-
-
-def async_find_group_entity_id(domain: str, group: Group) -> Optional[str]:
-    """Find the group entity id under test."""
-    entity_id = f"{domain}_zha_group_0x{group.group_id:04x}"
-
-    if entity_id in group.group_entities:
-        return entity_id
-    return None

tests/websocket/test_client_controller.py

@@ -41,7 +41,6 @@
     SIG_EP_OUTPUT,
     SIG_EP_PROFILE,
     SIG_EP_TYPE,
-    async_find_group_entity_id,
     create_mock_zigpy_device,
     find_entity,
     join_zigpy_device,
@@ -370,7 +369,7 @@ async def test_ws_client_gateway_groups(
         assert member.group == zha_group
         assert member.endpoint_id == 1
 
-    entity_id = async_find_group_entity_id(Platform.SWITCH, zha_group)
+    entity_id = f"{Platform.SWITCH}_zha_group_0x{zha_group.group_id:04x}"
     assert entity_id is not None
 
     group_proxy: Optional[WebSocketClientGroup] = ws_client_gateway.groups.get(