blob: da7c53186e51cb8b7d227fafc4c78dfd5e482e9e (
plain)
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
|
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import unittest
from math import sqrt
def is_prime(num):
if num < 2:
return False
for i in range(2, int(sqrt(num)) + 1):
if num % i == 0:
return False
return True
def prime_count(n):
return sum(1 for i in range(1, n) if is_prime(i))
class TestPrimeCount(unittest.TestCase):
def test_cases(self):
self.assertEqual(prime_count(10), 4, 'Test Case 1')
self.assertEqual(prime_count(15), 6, 'Test Case 2')
self.assertEqual(prime_count(1), 0, 'Test Case 3')
self.assertEqual(prime_count(25), 9, 'Test Case 4')
if __name__ == '__main__':
unittest.main()
|
