FilesExpand file tree

 Write-for-current-Python

/

prime_check.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

58 lines (47 loc) · 1.69 KB

 Write-for-current-Python

/

prime_check.py

Copy path

Top

File metadata and controls

• Code
• Blame

58 lines (47 loc) · 1.69 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

"""Prime Check."""

import math

import unittest

def prime_check(number: int) -> bool:

"""Checks to see if a number is a prime.

A number is prime if it has exactly two factors: 1 and itself.

"""

if 1 < number < 4:

# 2 and 3 are primes

return True

elif number < 2 or not number % 2:

# Negatives, 0, 1 and all even numbers are not primes

return False

odd_numbers = range(3, int(math.sqrt(number) + 1), 2)

return not any(not number % i for i in odd_numbers)

class Test(unittest.TestCase):

def test_primes(self):

self.assertTrue(prime_check(2))

self.assertTrue(prime_check(3))

self.assertTrue(prime_check(5))

self.assertTrue(prime_check(7))

self.assertTrue(prime_check(11))

self.assertTrue(prime_check(13))

self.assertTrue(prime_check(17))

self.assertTrue(prime_check(19))

self.assertTrue(prime_check(23))

self.assertTrue(prime_check(29))

def test_not_primes(self):

self.assertFalse(

prime_check(-19),

"Negative numbers are excluded by definition of prime numbers.",

)

self.assertFalse(

prime_check(0),

"Zero doesn't have any positive factors, primes must have exactly two.",

)

self.assertFalse(

prime_check(1),

"One only has 1 positive factor, primes must have exactly two.",

)

self.assertFalse(prime_check(2 * 2))

self.assertFalse(prime_check(2 * 3))

self.assertFalse(prime_check(3 * 3))

self.assertFalse(prime_check(3 * 5))

self.assertFalse(prime_check(3 * 5 * 7))

if __name__ == "__main__":

unittest.main()