From: Scott Gasch <scott@gasch.org>
Date: Thu, 10 Feb 2022 16:47:45 +0000 (-0800)
Subject: Fix a bug, add some testcases.
X-Git-Url: https://wannabe.guru.org/gitweb/?a=commitdiff_plain;h=d7d135be2f9dbb2d8398d7258f8bac24b4205a2b;p=python_utils.git

Fix a bug, add some testcases.
---

diff --git a/math_utils.py b/math_utils.py
index 28b8e6b..f77e0a1 100644
--- a/math_utils.py
+++ b/math_utils.py
@@ -21,6 +21,10 @@ class RunningMedian(object):
     >>> median.add_number(5)
     >>> median.get_median()
     5
+    >>> median.get_mean()
+    5.2
+    >>> round(median.get_stdev(), 2)
+    6.99
     """
 
     def __init__(self):
@@ -57,6 +61,7 @@ class RunningMedian(object):
         mean = self.get_mean()
         variance = 0.0
         for n in self.lowers:
+            n = -n
             variance += (n - mean) ** 2
         for n in self.highers:
             variance += (n - mean) ** 2