Restyled by autopep8

src/python_testing/TC_IDM_2_2.py

@@ -21,6 +21,7 @@
 from matter_testing_support import MatterBaseTest, async_test_body, default_matter_test_main
 from mobly import asserts
 
+
 class TC_IDM_2_2(MatterBaseTest):
 
     ROOT_NODE_ENDPOINT_ID = 0
@@ -37,9 +38,9 @@ def get_typed_attribute_path(attribute, ep=ROOT_NODE_ENDPOINT_ID):
     @async_test_body
     async def test_TC_IDM_2_2(self):
         # Test Setup
-        cluster_rev_attr = Clusters.BasicInformation.Attributes.ClusterRevision # Global attribute
+        cluster_rev_attr = Clusters.BasicInformation.Attributes.ClusterRevision  # Global attribute
         cluster_rev_attr_path = [(cluster_rev_attr)]
-        node_label_attr = Clusters.BasicInformation.Attributes.NodeLabel # Borrowed from TC_IDM_4_2
+        node_label_attr = Clusters.BasicInformation.Attributes.NodeLabel  # Borrowed from TC_IDM_4_2
         node_label_attr_path = [(0, node_label_attr)]
         node_label_attr_all = Clusters.BasicInformation
         node_label_attr_all_path = [(0, node_label_attr_all)]
@@ -57,82 +58,89 @@ async def test_TC_IDM_2_2(self):
         # AttributePath = [[Endpoint = Specific Endpoint, Cluster = Specific ClusterID, Attribute = Specific Attribute]]
         # On receipt of this message, DUT should send a report data action with the attribute value to the DUT
 
-        read_request_1 = await self.default_controller.ReadAttribute(self.dut_node_id, node_label_attr_path) # [(0, Clusters.BasicInformation.Attributes.NodeLabel)]
+        # [(0, Clusters.BasicInformation.Attributes.NodeLabel)]
+        read_request_1 = await self.default_controller.ReadAttribute(self.dut_node_id, node_label_attr_path)
         attributes = read_request_1[0]
         basic_information = attributes[basic_info_attr]
 
         asserts.assert_in(node_label_attr, basic_information, "NodeLabel not in BasicInformation")
         asserts.assert_in(data_version_attr, basic_information, "DataVersion not in BasicInformation")
 
-        ### TH sends the Read Request Message to the DUT to read all attributes on a given cluster and Endpoint
-        ### AttributePath = [[Endpoint = Specific Endpoint, Cluster = Specific ClusterID]]
-        ### On receipt of this message, DUT should send a report data action with the attribute value to the DUT.
+        # TH sends the Read Request Message to the DUT to read all attributes on a given cluster and Endpoint
+        # AttributePath = [[Endpoint = Specific Endpoint, Cluster = Specific ClusterID]]
+        # On receipt of this message, DUT should send a report data action with the attribute value to the DUT.
 
         self.print_step(2, "Send Request Message to read all attributes on a given cluster and endpoint")
         asserts.assert_in(node_label_attr, basic_information, "AttributeList not in attributes")
-        read_request_2 = await self.default_controller.ReadAttribute(self.dut_node_id, node_label_attr_all_path) # [(0, Clusters.BasicInformation)]
+        # [(0, Clusters.BasicInformation)]
+        read_request_2 = await self.default_controller.ReadAttribute(self.dut_node_id, node_label_attr_all_path)
         attributes = read_request_2[0]
         basic_information = attributes[basic_info_attr]
         asserts.assert_in(node_label_attr, basic_information, "NodeLabel not in BasicInformation")
         asserts.assert_in(data_version_attr, basic_information, "DataVersion not in BasicInformation")
         # The output from this command gets many more values from basic_info_attr compared to the first test -- get attribute list, expand test if needed
         asserts.assert_in(attribute_list_attr, basic_information, "AttributeList not in BasicInformation")
 
-        ### TH sends the Read Request Message to the DUT to read an attribute from a cluster at all Endpoints
-        ### AttributePath = [[Cluster = Specific ClusterID, Attribute = Specific Attribute]]
-        ### On receipt of this message, DUT should send a report data action with the attribute value from all the Endpoints to the DUT.
+        # TH sends the Read Request Message to the DUT to read an attribute from a cluster at all Endpoints
+        # AttributePath = [[Cluster = Specific ClusterID, Attribute = Specific Attribute]]
+        # On receipt of this message, DUT should send a report data action with the attribute value from all the Endpoints to the DUT.
         self.print_step(3, "Send Request Message to read one attribute on a given cluster at all endpoints")
-        read_request_3 = await self.default_controller.ReadAttribute(self.dut_node_id, [node_label_attr]) # [Clusters.BasicInformation.Attributes.NodeLabel]
+        # [Clusters.BasicInformation.Attributes.NodeLabel]
+        read_request_3 = await self.default_controller.ReadAttribute(self.dut_node_id, [node_label_attr])
         attributes = read_request_3[0]
         asserts.assert_in(node_label_attr, basic_information, "NodeLabel not in BasicInformation")
 
-        ### TH sends the Read Request Message to the DUT to read a global attribute from all clusters at that Endpoint
-        ### AttributePath = [[Endpoint = Specific Endpoint, Attribute = Specific Global Attribute]]
-        ### On receipt of this message, DUT should send a report data action with the attribute value from all the clusters to the DUT.
+        # TH sends the Read Request Message to the DUT to read a global attribute from all clusters at that Endpoint
+        # AttributePath = [[Endpoint = Specific Endpoint, Attribute = Specific Global Attribute]]
+        # On receipt of this message, DUT should send a report data action with the attribute value from all the clusters to the DUT.
         self.print_step(4, "Send Request Message to read one global attribute from all clusters at that endpoint")
         # The output from this command gets many more values from unit_testing_attr compared to the some other tests
-        read_request_4 = await self.default_controller.ReadAttribute(self.dut_node_id, cluster_rev_attr_path) # [(Clusters.BasicInformation.Attributes.ClusterRevision)]
+        # [(Clusters.BasicInformation.Attributes.ClusterRevision)]
+        read_request_4 = await self.default_controller.ReadAttribute(self.dut_node_id, cluster_rev_attr_path)
         attributes = read_request_4[0]
         basic_information = attributes[basic_info_attr]
         asserts.assert_true(basic_information[cluster_revision_attr], 3)
 
-        ### TH sends the Read Request Message to the DUT to read all attributes from all clusters on all Endpoints 
+        # TH sends the Read Request Message to the DUT to read all attributes from all clusters on all Endpoints
         ### AttributePath = [[]]
-        ### On receipt of this message, DUT should send a report data action with the attribute value from all the clusters to the DUT.
+        # On receipt of this message, DUT should send a report data action with the attribute value from all the clusters to the DUT.
         self.print_step(5, "Send Request Message to read all attributes from all clusters on all endpoints")
-        read_request_5 = await self.default_controller.ReadAttribute(self.dut_node_id, [()]) # This returns a dataclass key of 1, unlike 0 for the earlier ones
+        # This returns a dataclass key of 1, unlike 0 for the earlier ones
+        read_request_5 = await self.default_controller.ReadAttribute(self.dut_node_id, [()])
         attributes = read_request_5[1]
         unit_testing = attributes[unit_testing_attr]
 
         # asserts.assert_in(node_label_attr, unit_testing, "NodeLabel not in UnitTesting")
         asserts.assert_in(data_version_attr, unit_testing, "DataVersion not in UnitTesting")
-        ### TH sends the Read Request Message to the DUT to read a global attribute from all clusters at all Endpoints
-        ### AttributePath = [[Attribute = Specific Global Attribute]]
-        ### On receipt of this message, DUT should send a report data action with the attribute value from all the clusters to the DUT.
+        # TH sends the Read Request Message to the DUT to read a global attribute from all clusters at all Endpoints
+        # AttributePath = [[Attribute = Specific Global Attribute]]
+        # On receipt of this message, DUT should send a report data action with the attribute value from all the clusters to the DUT.
         self.print_step(6, "Send Request Message to read one global attribute from all clusters on all endpoints")
-        read_request_6 = await self.default_controller.ReadAttribute(self.dut_node_id, [cluster_rev_attr]) # Clusters.BasicInformation.Attributes.ClusterRevision
+        # Clusters.BasicInformation.Attributes.ClusterRevision
+        read_request_6 = await self.default_controller.ReadAttribute(self.dut_node_id, [cluster_rev_attr])
 
         attributes = read_request_6[0]
         basic_information = attributes[basic_info_attr]
         # asserts.assert_in(node_label_attr, basic_information, "NodeLabel not in BasicInformation")
         asserts.assert_in(data_version_attr, basic_information, "DataVersion not in BasicInformation")
         asserts.assert_true(basic_information[cluster_revision_attr], 3)
 
-        ### TH sends the Read Request Message to the DUT to read all attributes from a cluster at all Endpoints
-        ### AttributePath = [[Cluster = Specific ClusterID]]
-        ### On receipt of this message, DUT should send a report data action with the attribute value from all the Endpoints to the DUT.
+        # TH sends the Read Request Message to the DUT to read all attributes from a cluster at all Endpoints
+        # AttributePath = [[Cluster = Specific ClusterID]]
+        # On receipt of this message, DUT should send a report data action with the attribute value from all the Endpoints to the DUT.
         self.print_step(7, "Send Request Message to read all attributes from one cluster at all endpoints")
-        read_request_7 = await self.default_controller.ReadAttribute(self.dut_node_id, [node_label_attr_all]) # [Clusters.BasicInformation]
+        # [Clusters.BasicInformation]
+        read_request_7 = await self.default_controller.ReadAttribute(self.dut_node_id, [node_label_attr_all])
         attributes = read_request_7[0]
         basic_information = attributes[basic_info_attr]
         asserts.assert_in(node_label_attr, basic_information, "NodeLabel not in BasicInformation")
         asserts.assert_in(data_version_attr, basic_information, "DataVersion not in BasicInformation")
         # The output from this command gets many more values from basic_info_attr compared to the first test -- get attribute list, expand test if needed
         asserts.assert_in(attribute_list_attr, basic_information, "AttributeList not in BasicInformation")
 
-        ### TH sends the Read Request Message to the DUT to read all attributes from all clusters at one Endpoint
-        ### AttributePath = [[Endpoint = Specific Endpoint]]
-        ### On receipt of this message, DUT should send a report data action with the attribute value from all the Endpoints to the DUT.
+        # TH sends the Read Request Message to the DUT to read all attributes from all clusters at one Endpoint
+        # AttributePath = [[Endpoint = Specific Endpoint]]
+        # On receipt of this message, DUT should send a report data action with the attribute value from all the Endpoints to the DUT.
         self.print_step(8, "Send Request Message to read all attributes from all clusters at one endpoint")
         read_request_8 = await self.default_controller.ReadAttribute(self.dut_node_id, [0])
         attributes = read_request_8[0]
@@ -142,5 +150,6 @@ async def test_TC_IDM_2_2(self):
         asserts.assert_in(data_version_attr, network_diagnostics, "DataVersion not in ThreadNetworkDiagnostics")
         asserts.assert_in(attribute_list_attr, basic_information, "AttributeList not in BasicInformation")
 
+
 if __name__ == "__main__":
-    default_matter_test_main()
+    default_matter_test_main()