Merge pull request keon#66 from mathagician/issue-56

keon#58 | Correct the logic of Prime testing function in math/test_primt.py
RoWisdom · May 14, 2017 · c8c2f99 · c8c2f99
2 parents 238e68b + 5055f2d
commit c8c2f99
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/math/prime_test.py b/math/prime_test.py
@@ -10,9 +10,9 @@ def prime_test(n):
         return True
     if n%2==0 or n%3==0:
         return False
-    j = 6
-    while((j-1)*(j-1) <= 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