wip

tools/zookeeper/check_consumption.py

@@ -0,0 +1,27 @@
+#!/usr/bin/env python3
+
+import sys
+import time
+from openpilot.tools.zookeeper import Zookeeper
+
+# Usage: check_consumption.py <averaging_time_sec> <max_average_power_W>
+# Exit code: 0 -> passed
+#	           1 -> failed
+
+if __name__ == "__main__":
+  z = Zookeeper()
+
+  averaging_time_s = int(sys.argv[1])
+  max_average_power = float(sys.argv[2])
+
+  start_time = time.time()
+  measurements = []
+  while time.time() - start_time < averaging_time_s:
+    measurements.append(z.read_power())
+    time.sleep(0.1)
+
+  average_power = sum(measurements)/len(measurements)
+  print(f"Average power: {round(average_power, 4)}W")
+
+  if average_power > max_average_power:
+    exit(1)