소스 검색

Check for zeros in calculate_average_lifetime()

Related to #102
Vadim Markovtsev 6 년 전
부모
커밋
b25b0cba04
1개의 변경된 파일5개의 추가작업 그리고 2개의 파일을 삭제
  1. 5 2
      labours.py

+ 5 - 2
labours.py

@@ -362,8 +362,11 @@ def calculate_average_lifetime(matrix):
                 continue
             lifetimes[i - start] = band[i - 1] - line
         lifetimes[i - start] = band[i - 1]
-    return (lifetimes.dot(numpy.arange(1, matrix.shape[1], 1))
-            / (lifetimes.sum() * matrix.shape[1]))
+    lsum = lifetimes.sum()
+    if lsum != 0:
+        return (lifetimes.dot(numpy.arange(1, matrix.shape[1], 1))
+                / (lsum * matrix.shape[1]))
+    return numpy.nan
 
 
 def interpolate_burndown_matrix(matrix, granularity, sampling):