Fixes by format action

welcome_aboard/battery_testing/main.py → docs_tofupilot/api_v1/assembly_test_with_voltage_sampling.py

@@ -136,6 +136,43 @@ def voltage_value():
     return passed, value_measured, "V", 10.0, 12.0
 
 
+def voltage_sampling_test():
+    """
+    Multi-dimensional voltage sampling test.
+    Samples 100 voltage measurements over time.
+    Returns: (passed, measurements, units, limit_low, limit_high)
+    where measurements is a list of (timestamp, voltage) tuples
+    """
+    passed_prob = 0.98
+    num_samples = 100
+    limit_low = 10.0
+    limit_high = 12.0
+
+    measurements = []
+    all_passed = True
+
+    # Simulate voltage measurements over time
+    for i in range(num_samples):
+        timestamp = i / num_samples  # Time in seconds (0 to 0.99)
+
+        # Determine if this sample should pass
+        sample_passed = simulate_test_result(passed_prob)
+
+        if sample_passed:
+            voltage = round(random.uniform(limit_low, limit_high), 2)
+        else:
+            voltage = round(random.uniform(8.0, 9.5), 2)
+            all_passed = False
+
+        measurements.append((timestamp, voltage))
+
+    # Return multi-dimensional measurement
+    # Format: (passed, value_measured, unit, limit_low, limit_high)
+    # For multi-dimensional, value_measured is the list of tuples
+    # and unit should be a list of units for each dimension
+    return all_passed, measurements, ["s", "V"], limit_low, limit_high
+
+
 def internal_resistance():
     passed = simulate_test_result(0.98)
     value_measured = (
@@ -349,10 +386,13 @@ def execute_procedures(end):
         if not passed_cell:
             continue
 
-        # Execute Assembly Tests
+        # Execute Assembly Tests with voltage sampling
         tests_assembly = [
             (battery_connection, timedelta(seconds=0.1)),
-            (voltage_value, timedelta(seconds=1)),
+            (
+                voltage_sampling_test,
+                timedelta(seconds=10),
+            ),  # New multi-dimensional test
             (internal_resistance, timedelta(seconds=1)),
             (thermal_runaway_detection, timedelta(seconds=2)),
             (state_of_health, timedelta(seconds=2)),