formatting

Author: wilbersl

tedge_modbus/reader/mapper.py

@@ -220,7 +220,11 @@ def check_alarm(self, value, alarm_mapping, register_type, register_key):
             topic = topics["alarm"]
             topic = topic.replace("CHILD_ID", self.device.get("name"))
             topic = topic.replace("TYPE", alarm_type)
-            data = {"text": text, "severity": severity, "time": datetime.now(timezone.utc).isoformat()}
+            data = {
+                "text": text,
+                "severity": severity,
+                "time": datetime.now(timezone.utc).isoformat(),
+            }
             messages.append(MappedMessage(json.dumps(data), topic))
         return messages
 

tests/unit/test_mapper.py

@@ -68,9 +68,7 @@ def test_on_change_true_and_value_is_same(self):
         messages, _ = self.mapper.map_register(
             read_register=[123], register_def=register_def
         )
-        self.assertEqual(
-            len(messages), 0, "Should not publish when value is unchanged"
-        )
+        self.assertEqual(len(messages), 0, "Should not publish when value is unchanged")
 
     def test_on_change_false_and_value_is_same(self):
         register_def = {
@@ -198,7 +196,9 @@ def test_separate_measurements(self):
         self.mapper.map_register(read_register=[0], register_def=register_def)
         # Second poll: Check if Measurements get separated
         messages, measurement3 = self.mapper.map_register(
-            read_register=[1], register_def=register_def, device_combine_measurements=True
+            read_register=[1],
+            register_def=register_def,
+            device_combine_measurements=True,
         )
         topics = [message.topic for message in messages]
         self.assertTrue("te/device/test_device///m/" not in topics)
@@ -236,13 +236,19 @@ def test_combine_messages(self):
         }
 
         _, measurement1 = self.mapper.map_register(
-            read_register=[25], register_def=register_def1, device_combine_measurements=True
+            read_register=[25],
+            register_def=register_def1,
+            device_combine_measurements=True,
         )
         _, measurement2 = self.mapper.map_register(
-            read_register=[43], register_def=register_def2, device_combine_measurements=True
+            read_register=[43],
+            register_def=register_def2,
+            device_combine_measurements=True,
         )
         _, measurement3 = self.mapper.map_register(
-            read_register=[21], register_def=register_def3, device_combine_measurements=True
+            read_register=[21],
+            register_def=register_def3,
+            device_combine_measurements=True,
         )
 
         measurement1.extend_data(measurement2)
@@ -295,7 +301,7 @@ def test_alarm_mapping(self):
         alarm_data2 = json.loads(messages2[0].data)
         self.assertEqual(alarm_data2["severity"], "major")
         self.assertEqual(alarm_data2["text"], "This alarm tests the alarm mapping")
-    
+
     def test_event_mapping(self):
         register_def = {
             "number": 100,
@@ -333,4 +339,4 @@ def test_event_mapping(self):
         self.assertTrue("te/device/test_device///e/" in topics2)
         self.assertEqual(len(messages2), 1)
         event_data2 = json.loads(messages2[0].data)
-        self.assertEqual(event_data2["text"], "This event tests the event mapping")
+        self.assertEqual(event_data2["text"], "This event tests the event mapping")