debug

Edison-CBS · Oct 4, 2024 · da8d561 · da8d561
1 parent 4c145de
commit da8d561
Showing 1 changed file with 42 additions and 33 deletions.
diff --git a/selfdrive/selfdrived/tests/test_alertmanager.py b/selfdrive/selfdrived/tests/test_alertmanager.py
@@ -12,6 +12,7 @@ def test_duration(self):
     """
     for duration in range(1, 100):
       print('duration', duration)
+      duration = 5
       alert = None
       while not isinstance(alert, Alert):
         event = random.choice([e for e in EVENTS.values() if len(e)])
@@ -21,45 +22,53 @@ def test_duration(self):
 
       # check one case:
       # - if alert is re-added to AM before it ends the duration is extended
-      if duration > 1:
+      # if duration > 1:
+      #   AM = AlertManager()
+      #   show_duration = duration * 2
+      #   for frame in range(duration * 2 + 10):
+      #     print(frame)
+      #     if frame == 0:
+      #       AM.add_many(frame, [alert, ])
+      #       print('added!')
+      #
+      #     if frame == duration:
+      #       # add alert one frame before it ends
+      #       assert AM.current_alert == alert
+      #       AM.add_many(frame, [alert, ])
+      #       print('added!')
+      #     AM.process_alerts(frame, set())
+      #     print('alert', AM.current_alert)
+      #
+      #     shown = AM.current_alert != EmptyAlert
+      #     should_show = frame <= show_duration
+      #     print('shown', shown, 'should_show', should_show)
+      #     print()
+      #     assert shown == should_show, f"{frame=} {duration=}"
+      # break
+
+      # check two cases:
+      # - alert is added to AM for <= the alert's duration
+      # - alert is added to AM for > alert's duration
+      for greater in (True, False):
+        print('greater', greater)
+        if greater:
+          add_duration = duration + random.randint(1, 10)
+        else:
+          add_duration = random.randint(1, duration)
+        show_duration = max(duration, add_duration)
+        print('add_duration', add_duration, 'show_duration', show_duration)
+
         AM = AlertManager()
-        show_duration = duration * 2
-        for frame in range(duration * 2 + 10):
-          print(frame)
-          if frame == 0:
-            AM.add_many(frame, [alert, ])
+        for frame in range(duration+10):
+          print('frame', frame)
+          if frame < add_duration:
             print('added!')
-
-          if frame == duration:
-            # add alert one frame before it ends
-            assert AM.current_alert == alert
             AM.add_many(frame, [alert, ])
-            print('added!')
           AM.process_alerts(frame, set())
-          print('alert', AM.current_alert)
+          print(AM.current_alert)
 
           shown = AM.current_alert != EmptyAlert
           should_show = frame <= show_duration
           print('shown', shown, 'should_show', should_show)
+          assert shown == should_show, f"{frame=} {add_duration=} {duration=}"
           print()
-          assert shown == should_show, f"{frame=} {duration=}"
-
-      # # check two cases:
-      # # - alert is added to AM for <= the alert's duration
-      # # - alert is added to AM for > alert's duration
-      # for greater in (True, False):
-      #   if greater:
-      #     add_duration = duration + random.randint(1, 10)
-      #   else:
-      #     add_duration = random.randint(1, duration)
-      #   show_duration = max(duration, add_duration)
-      #
-      #   AM = AlertManager()
-      #   for frame in range(duration+10):
-      #     if frame < add_duration:
-      #       AM.add_many(frame, [alert, ])
-      #     AM.process_alerts(frame, set())
-      #
-      #     shown = AM.current_alert != EmptyAlert
-      #     should_show = frame <= show_duration
-      #     assert shown == should_show, f"{frame=} {add_duration=} {duration=}"