Bugfix: possible divide by zero in stats.py

1 files changed, 4 insertions, 1 deletions

diff --git a/stats.py b/stats.py
index 6216f44..d61be6f 100644
--- a/stats.py
+++ b/stats.py
@@ -24,7 +24,10 @@ def stats(stream):
                     min_inversion = length
                 sum_inversions += length
         yield record
-    avg_inversion = int(sum_inversions / num_inversions)
+    if num_inversions > 0:
+        avg_inversion = int(sum_inversions / num_inversions)
+    else:
+        avg_inversion = 0
     class Obj(): pass
     rec = Obj()
     rec.record_type = "meta"