From 17830050f0d1dd7ae25e891ccd32ee78c4cb648c Mon Sep 17 00:00:00 2001 From: Vivien Nicolas Date: Thu, 15 Jun 2023 09:41:38 +0200 Subject: [PATCH] [matter_yamltests] Add SpecDefinitions.get_cluster_names to get all the cluster names (#27252) --- scripts/py_matter_yamltests/matter_yamltests/definitions.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/scripts/py_matter_yamltests/matter_yamltests/definitions.py b/scripts/py_matter_yamltests/matter_yamltests/definitions.py index 8597262d1a06d9..03a791b62d1006 100644 --- a/scripts/py_matter_yamltests/matter_yamltests/definitions.py +++ b/scripts/py_matter_yamltests/matter_yamltests/definitions.py @@ -87,6 +87,9 @@ def __init__(self, sources: List[ParseSource]): self.__responses_by_id[code][struct.code] = struct self.__responses_by_name[name][struct.name] = struct.code + def get_cluster_names(self) -> List[str]: + return [name for name, _ in self.__clusters_by_name.items()] + def get_cluster_name(self, cluster_id: int) -> str: cluster = self.__clusters_by_id.get(cluster_id) return cluster.name if cluster else None