Correctly check for squares of primes (#549)

As the for-loop was looping by two numbers, it would stop at 1 number less than the root of prime squares, resulting it in incorrectly classifying the squares as prime numbers. Incrementing the loop ensures the next number is also considered resulting in accurate classification.
This commit is contained in:
Kushal Naidu 2018-10-30 19:08:44 +05:30 committed by Harshil
parent fa86d3d954
commit 75d11f9034

View File

@ -1,6 +1,6 @@
def primeCheck(number):
prime = True
for i in range(2, int(number**(0.5)+1), 2):
for i in range(2, int(number**(0.5)+2), 2):
if i != 2:
i = i - 1
if number % i == 0: