Auto code format

charlottEv · web-flow · commit cb2c84abd817 · 2025-05-01T14:16:01.000Z
diff --git a/docs_tofupilot/integrations/openhtf/main.py b/docs_tofupilot/integrations/openhtf/main.py
@@ -7,7 +7,9 @@ def phase_one():
 
 
 def main():
-    test = htf.Test(phase_one, procedure_id="FVT1", part_number="PCB01") # Specify procedure and part_number
+    test = htf.Test(
+        phase_one, procedure_id="FVT1", part_number="PCB01"
+    )  # Specify procedure and part_number
 
     with TofuPilot(test):  # One-line integration
         test.execute(lambda: "PCB1A001")
diff --git a/docs_tofupilot/phases/advanced/openhtf/main.py b/docs_tofupilot/phases/advanced/openhtf/main.py
@@ -16,9 +16,11 @@ def phase_firmware_version_check(test):
 
 # Phase with multiple measurements
 @htf.measures(
-    htf.Measurement("input_voltage").in_range(3.1, 3.5).with_units(units.VOLT),
-    htf.Measurement("output_voltage").in_range(1.1, 1.3).with_units(units.VOLT),
-)
+    htf.Measurement("input_voltage").in_range(
+        3.1, 3.5).with_units(
+            units.VOLT), htf.Measurement("output_voltage").in_range(
+                1.1, 1.3).with_units(
+                    units.VOLT), )
 def phase_voltage_measurements(test):
     test.measurements.input_voltage = 3.3
     test.measurements.output_voltage = 1.2
diff --git a/docs_tofupilot/phases/dimensions/client/main.py b/docs_tofupilot/phases/dimensions/client/main.py
@@ -1,6 +1,7 @@
+import random
 from datetime import datetime
+
 from tofupilot import MeasurementOutcome, PhaseOutcome, TofuPilotClient
-import random
 
 client = TofuPilotClient()
 
@@ -14,7 +15,8 @@ def phase_voltage_measure():
         voltage = round(random.uniform(3.3, 3.5), 2)
         current = round(random.uniform(0.3, 0.8), 3)
         resistance = voltage / current
-        structured_measurements.append((timestamp, voltage, current, resistance))
+        structured_measurements.append(
+            (timestamp, voltage, current, resistance))
 
     phase = [
         {
@@ -43,7 +45,8 @@ def main():
         procedure_id="FVT1",  # Create the procedure first in the Application
         unit_under_test={"serial_number": "PCB1A001", "part_number": "PCB1"},
         phases=phases,
-        run_passed=all(phase["outcome"] == PhaseOutcome.PASS for phase in phases),
+        run_passed=all(
+            phase["outcome"] == PhaseOutcome.PASS for phase in phases),
     )
 
 
diff --git a/docs_tofupilot/phases/dimensions/openhtf/main.py b/docs_tofupilot/phases/dimensions/openhtf/main.py
@@ -1,7 +1,8 @@
+import random
+
 import openhtf as htf
 from openhtf.util import units
 from tofupilot.openhtf import TofuPilot
-import random
 
 
 @htf.measures(
diff --git a/docs_tofupilot/phases/multi-measurements/client/main.py b/docs_tofupilot/phases/multi-measurements/client/main.py
@@ -1,6 +1,7 @@
 import random
 from datetime import datetime
-from tofupilot import TofuPilotClient, PhaseOutcome, MeasurementOutcome
+
+from tofupilot import MeasurementOutcome, PhaseOutcome, TofuPilotClient
 
 client = TofuPilotClient()
 
diff --git a/docs_tofupilot/phases/multi-measurements/openhtf/main.py b/docs_tofupilot/phases/multi-measurements/openhtf/main.py
@@ -1,14 +1,16 @@
+import random
+
 import openhtf as htf
 from openhtf.util import units
 from tofupilot.openhtf import TofuPilot
-import random
 
 
 @htf.measures(
     htf.Measurement("is_connected").equals(True),  # Boolean measure
     htf.Measurement("firmware_version").equals("1.2.7"),  # String measure
     htf.Measurement("input_voltage").in_range(3.2, 3.4).with_units(units.VOLT),
-    htf.Measurement("input_current").in_range(maximum=1.5).with_units(units.AMPERE),
+    htf.Measurement("input_current").in_range(
+        maximum=1.5).with_units(units.AMPERE),
 )
 def phase_multi_measurements(test):
     test.measurements.is_connected = True
diff --git a/docs_tofupilot/phases/optional/client/main.py b/docs_tofupilot/phases/optional/client/main.py
@@ -36,9 +36,7 @@ def main():
 
     client.create_run(
         procedure_id="FVT1",  # Create the procedure first in the Application
-        unit_under_test={
-            "serial_number": "PCB1A001",
-            "part_number": "PCB1"},
+        unit_under_test={"serial_number": "PCB1A001", "part_number": "PCB1"},
         phases=phases,
         run_passed=all(
             phase["outcome"] == PhaseOutcome.PASS for phase in phases),
diff --git a/docs_tofupilot/phases/required/client/main.py b/docs_tofupilot/phases/required/client/main.py
@@ -22,9 +22,7 @@ def main():
 
     client.create_run(
         procedure_id="FVT1",  # Create the procedure first in the Application
-        unit_under_test={
-            "serial_number": "PCB1A001",
-            "part_number": "PCB1"},
+        unit_under_test={"serial_number": "PCB1A001", "part_number": "PCB1"},
         phases=phases,
         run_passed=all(
             phase["outcome"] == PhaseOutcome.PASS for phase in phases),
diff --git a/docs_tofupilot/procedures/client/main.py b/docs_tofupilot/procedures/client/main.py
@@ -1,4 +1,5 @@
 from datetime import timedelta
+
 from tofupilot import TofuPilotClient
 
 
diff --git a/docs_tofupilot/stations/client/main.py b/docs_tofupilot/stations/client/main.py
@@ -1,10 +1,12 @@
 # Before running the script, ensure you have created a station in the TofuPilot interface
 # and linked it to the specified procedure ID ("FVT1" in this example).
 # You also need to save your API key in an environment variable named "STATION_API_KEY"
-# or pass it directly as an argument like this: TofuPilotClient(api_key="STATION_API_KEY")
+# or pass it directly as an argument like this:
+# TofuPilotClient(api_key="STATION_API_KEY")
 
-from datetime import timedelta, datetime
-from tofupilot import TofuPilotClient, PhaseOutcome
+from datetime import datetime, timedelta
+
+from tofupilot import PhaseOutcome, TofuPilotClient
 
 
 def phase_one():
@@ -26,7 +28,8 @@ def main():
 
     client.create_run(
         procedure_id="FVT1",  # Create a station in TofuPilot linked to this procedure ID
-        run_passed=all(phase["outcome"] == PhaseOutcome.PASS for phase in phases),
+        run_passed=all(
+            phase["outcome"] == PhaseOutcome.PASS for phase in phases),
         unit_under_test={"serial_number": "PCB1A001", "part_number": "PCBA01"},
         phases=phases,
         duration=timedelta(minutes=1, seconds=45),
diff --git a/docs_tofupilot/stations/openhtf/main.py b/docs_tofupilot/stations/openhtf/main.py
@@ -1,7 +1,8 @@
 # Before running the script, ensure you have created a station in the TofuPilot interface
 # and linked it to the specified procedure ID ("FVT1" in this example).
 # You also need to save your API key in an environment variable named "STATION_API_KEY"
-# or pass it directly as an argument like this: TofuPilot(test, api_key="STATION_API_KEY")
+# or pass it directly as an argument like this: TofuPilot(test,
+# api_key="STATION_API_KEY")
 
 
 import openhtf as htf
diff --git a/docs_tofupilot/sub-units/client/main.py b/docs_tofupilot/sub-units/client/main.py
@@ -12,7 +12,8 @@ def main():
         procedure_id="FVT2",  # Create the procedure first in the Application
         unit_under_test={"serial_number": "CAM1A001", "part_number": "CAM1"},
         run_passed=True,
-        sub_units=[{"serial_number": "PCB1A001"}, {"serial_number": "LEN1A001"}],
+        sub_units=[{"serial_number": "PCB1A001"},
+                   {"serial_number": "LEN1A001"}],
     )
 
 
diff --git a/docs_tofupilot/sub-units/openhtf/main.py b/docs_tofupilot/sub-units/openhtf/main.py
@@ -10,7 +10,8 @@ def main():
     test = Test(
         procedure_id="FVT2",  # Create the procedure first in the Application
         part_number="CAM1",
-        sub_units=[{"serial_number": "PCB1A001"}, {"serial_number": "LEN1A001"}],
+        sub_units=[{"serial_number": "PCB1A001"},
+                   {"serial_number": "LEN1A001"}],
     )
     with TofuPilot(test):
         test.execute(lambda: "CAM1A001")
diff --git a/qa/client/create_run/basic/main.py b/qa/client/create_run/basic/main.py
@@ -8,9 +8,10 @@
 https://tofupilot.com/docs/user-management#api-key
 """
 
-from tofupilot import TofuPilotClient
 import random
 
+from tofupilot import TofuPilotClient
+
 
 def main():
     # Initialize the TofuPilot client.
@@ -21,7 +22,9 @@ def main():
     serial_number = f"00220D4K{random_digits}"
     client.create_run(
         procedure_id="FVT1",
-        unit_under_test={"serial_number": serial_number, "part_number": "PCB01"},
+        unit_under_test={
+            "serial_number": serial_number,
+            "part_number": "PCB01"},
         run_passed=True,
     )
 
diff --git a/qa/client/create_run/phases_string_outcome/main.py b/qa/client/create_run/phases_string_outcome/main.py
@@ -1,5 +1,6 @@
 import random
 import time
+
 from tofupilot import TofuPilotClient
 
 client = TofuPilotClient()
diff --git a/qa/client/create_run/procedure_version/main.py b/qa/client/create_run/procedure_version/main.py
@@ -1,6 +1,7 @@
 import random
 import time
 from datetime import datetime, timedelta
+
 from tofupilot import TofuPilotClient
 
 
diff --git a/qa/client/create_run/started_at/main.py b/qa/client/create_run/started_at/main.py
@@ -1,7 +1,8 @@
-from tofupilot import TofuPilotClient
 import random
 from datetime import datetime, timedelta
 
+from tofupilot import TofuPilotClient
+
 
 def main():
     # Initialize the TofuPilot client.
@@ -12,10 +13,15 @@ def main():
     serial_number = f"00220D4K{random_digits}"
     client.create_run(
         procedure_id="FVT1",
-        unit_under_test={"serial_number": serial_number, "part_number": "PCB01"},
+        unit_under_test={
+            "serial_number": serial_number,
+            "part_number": "PCB01"},
         run_passed=True,
-        started_at=datetime.now() - timedelta(days=1),
-        duration=timedelta(seconds=23),
+        started_at=datetime.now() -
+        timedelta(
+            days=1),
+        duration=timedelta(
+            seconds=23),
     )
 
 
diff --git a/qa/openhtf/generic/main.py b/qa/openhtf/generic/main.py
@@ -1,6 +1,7 @@
 import random
 import time
 from datetime import datetime, timedelta
+
 import openhtf as htf
 from openhtf.util import units
 from tofupilot.openhtf import TofuPilot
@@ -17,7 +18,8 @@ def check_button(test):
     time.sleep(1)
 
 
-@htf.measures(htf.Measurement("input_voltage").in_range(4.5, 5).with_units(units.VOLT))
+@htf.measures(htf.Measurement("input_voltage").in_range(4.5,
+              5).with_units(units.VOLT))
 def test_voltage_input(test):
     test.measurements.input_voltage = round(random.uniform(3.7, 4.9), 2)
 
@@ -35,7 +37,8 @@ def test_voltage_output(test):
     .with_units(units.AMPERE)
 )
 def test_overcurrent_protection(test):
-    test.measurements.current_protection_triggered = round(random.uniform(1.0, 1.7), 3)
+    test.measurements.current_protection_triggered = round(
+        random.uniform(1.0, 1.7), 3)
     time.sleep(1)
 
 
@@ -52,7 +55,8 @@ def test_battery_switch():
 def test_converter_efficiency(test):
     input_power = 500
     output_power = round(random.uniform(425, 480))
-    test.measurements.efficiency = round(((output_power / input_power) * 100), 1)
+    test.measurements.efficiency = round(
+        ((output_power / input_power) * 100), 1)
     time.sleep(1)
 
 
diff --git a/qa/openhtf/logger/main.py b/qa/openhtf/logger/main.py
@@ -1,7 +1,8 @@
+import random
+
 import openhtf as htf
-from tofupilot.openhtf import TofuPilot
 from openhtf.output.callbacks import json_factory
-import random
+from tofupilot.openhtf import TofuPilot
 
 
 @htf.measures(htf.Measurement("button_status").equals(True))
@@ -40,7 +41,9 @@ def main():
     random_digits = "".join([str(random.randint(0, 9)) for _ in range(5)])
     serial_number = f"00220D4K{random_digits}"
 
-    test.add_output_callbacks(json_factory.OutputToJSON("test_result.json", indent=2))
+    test.add_output_callbacks(
+        json_factory.OutputToJSON(
+            "test_result.json", indent=2))
 
     # Execute the test
     with TofuPilot(test):
diff --git a/qa/openhtf/multi_dimensions/main.py b/qa/openhtf/multi_dimensions/main.py
@@ -1,10 +1,11 @@
-import openhtf as htf
-from openhtf.util import units
-from tofupilot.openhtf import TofuPilot
-from openhtf.output.callbacks import json_factory
 import random
 import time
+
 import numpy as np
+import openhtf as htf
+from openhtf.output.callbacks import json_factory
+from openhtf.util import units
+from tofupilot.openhtf import TofuPilot
 
 
 @htf.measures(
@@ -65,7 +66,8 @@ def power_phase(test):
         voltage = round(random.uniform(3.3, 3.5), 2)
         current = round(random.uniform(0.3, 0.8), 3)
         resistance = voltage / current
-        test.measurements.current_and_voltage_over_time[timestamp, voltage] = current
+        test.measurements.current_and_voltage_over_time[timestamp,
+                                                        voltage] = current
         test.measurements.current_voltage_resistence_over_time[
             timestamp, voltage, current
         ] = resistance
@@ -84,7 +86,9 @@ def main():
     random_digits = "".join([str(random.randint(0, 9)) for _ in range(5)])
     serial_number = f"00220D4K{random_digits}"
 
-    test.add_output_callbacks(json_factory.OutputToJSON("test_result.json", indent=2))
+    test.add_output_callbacks(
+        json_factory.OutputToJSON(
+            "test_result.json", indent=2))
 
     start = time.time()
     # Execute the test
diff --git a/qa/openhtf/multi_measurements/main.py b/qa/openhtf/multi_measurements/main.py
diff --git a/qa/openhtf/procedure_version/main.py b/qa/openhtf/procedure_version/main.py
diff --git a/qa/openhtf/regex_serial_number/main.py b/qa/openhtf/regex_serial_number/main.py

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,5 @@`
`1`	`1`	`from datetime import timedelta`
	`2`	`+`
`2`	`3`	`from tofupilot import TofuPilotClient`
`3`	`4`
`4`	`5`