Fix ZeroDivisionError in throughput()

boxine · Feb 27, 2024 · 9e702ab · 9e702ab
1 parent eb1901a
commit 9e702ab
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/huey_monitor/humanize.py b/huey_monitor/humanize.py
@@ -70,11 +70,15 @@ def throughput(num, elapsed_sec, suffix='', divisor=1000) -> str:
     '2.00kBytes/s'
     >>> throughput(4, 250, suffix='subtask')
     '1.0\xa0minutes/subtask'
+    >>> throughput(123, 0)
+    '∞'
     """
     if num == 0:
         if suffix:
             return f'0/{suffix}'
         return '0'
+    if elapsed_sec == 0:
+        return '∞'
     rate = num / elapsed_sec
 
     if rate > 1: