diff options
Diffstat (limited to 'math_utils.py')
| -rw-r--r-- | math_utils.py | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/math_utils.py b/math_utils.py index e0e3f6c..3216d4a 100644 --- a/math_utils.py +++ b/math_utils.py @@ -39,7 +39,7 @@ class RunningMedian(object): def get_median(self): if len(self.lowers) == len(self.highers): - return (-self.lowers[0] + self.highers[0])/2 + return (-self.lowers[0] + self.highers[0]) / 2 elif len(self.lowers) > len(self.highers): return -self.lowers[0] else: @@ -143,12 +143,12 @@ def is_prime(n: int) -> bool: # This is checked so that we can skip middle five numbers in below # loop - if (n % 2 == 0 or n % 3 == 0): + if n % 2 == 0 or n % 3 == 0: return False i = 5 while i * i <= n: - if (n % i == 0 or n % (i + 2) == 0): + if n % i == 0 or n % (i + 2) == 0: return False i = i + 6 return True @@ -156,4 +156,5 @@ def is_prime(n: int) -> bool: if __name__ == '__main__': import doctest + doctest.testmod() |