Fix | keon#58 | Correct the logic of Prime testing function in math/t…

…est_prime.py in an efficient way
guochangchun · May 14, 2017 · a4ca73b · a4ca73b
1 parent 76c50d6
commit a4ca73b
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/math/prime_test.py b/math/prime_test.py
@@ -10,9 +10,10 @@ def prime_test(n):
         return True
     if n%2==0 or n%3==0:
         return False
-    j = 6
-    while(j*j < n):
-        if n%(j-1)==0 or n%(j+1)==0:
+
+    j = 5
+    while(j*j <= n):
+        if n%(j)==0 or n%(j+2)==0:
             return False
         j += 6
     return True