Merge panda subtree

panda/tests/automated/5_wifi_functionality.py

@@ -0,0 +1,69 @@
+import time
+from panda import Panda
+from .helpers import start_heartbeat_thread, reset_pandas, time_many_sends, connect_wifi, test_white, panda_type_to_serial
+
+# Reset the pandas before running tests
+def aaaa_reset_before_tests():
+  reset_pandas()
+
+@test_white
+@panda_type_to_serial
+def test_get_serial_wifi(serials=None):
+  connect_wifi(serials[0])
+
+  p = Panda("WIFI")
+  print(p.get_serial())
+
+@test_white
+@panda_type_to_serial
+def test_throughput(serials=None):
+  connect_wifi(serials[0])
+  p = Panda(serials[0])
+
+  # Start heartbeat
+  start_heartbeat_thread(p)
+
+  # enable output mode
+  p.set_safety_mode(Panda.SAFETY_ALLOUTPUT)
+
+  # enable CAN loopback mode
+  p.set_can_loopback(True)
+
+  p = Panda("WIFI")
+
+  for speed in [100,250,500,750,1000]:
+    # set bus 0 speed to speed
+    p.set_can_speed_kbps(0, speed)
+    time.sleep(0.1)
+
+    comp_kbps = time_many_sends(p, 0)
+
+    # bit count from https://en.wikipedia.org/wiki/CAN_bus
+    saturation_pct = (comp_kbps/speed) * 100.0
+    #assert_greater(saturation_pct, 80)
+    #assert_less(saturation_pct, 100)
+
+    print("WIFI loopback 100 messages at speed %d, comp speed is %.2f, percent %.2f" % (speed, comp_kbps, saturation_pct))
+
+@test_white
+@panda_type_to_serial
+def test_recv_only(serials=None):
+  connect_wifi(serials[0])
+  p = Panda(serials[0])
+
+  # Start heartbeat
+  start_heartbeat_thread(p)
+
+  p.set_safety_mode(Panda.SAFETY_ALLOUTPUT)
+
+  p.set_can_loopback(True)
+  pwifi = Panda("WIFI")
+
+  # TODO: msg_count=1000 drops packets, is this fixable?
+  for msg_count in [10,100,200]:
+    speed = 500
+    p.set_can_speed_kbps(0, speed)
+    comp_kbps = time_many_sends(p, 0, pwifi, msg_count)
+    saturation_pct = (comp_kbps/speed) * 100.0
+
+    print("HT WIFI loopback %d messages at speed %d, comp speed is %.2f, percent %.2f" % (msg_count, speed, comp_kbps, saturation_pct))