felixm/euler
1
0
Fork 0
Code Pull Requests Releases Activity

Moved 6 to 11 from ipython to python.

Browse Source
This commit is contained in:
Felix Martin
2019-07-14 16:51:31 -04:00
parent b246d56acd
commit 86e68eeee2
8 changed files with 289 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

39
python/lib_prime_tests.py
View File

@@ -2,9 +2,15 @@ import unittest
try:
from .lib_prime import prime_factors
from .lib_prime import prime_factors_count
from .lib_prime import is_prime
from .lib_prime import prime_nth
from .lib_prime import primes
except ModuleNotFoundError:
from lib_prime import prime_factors
from lib_prime import prime_factors_count
from lib_prime import is_prime
from lib_prime import prime_nth
from lib_prime import primes
class TestPrimeMethods(unittest.TestCase):
@@ -20,6 +26,39 @@ class TestPrimeMethods(unittest.TestCase):
self.assertEqual(prime_factors_count(2), {2: 1})
self.assertEqual(prime_factors_count(147), {3: 1, 7: 2})
def test_is_prime(self):
self.assertTrue(is_prime(2))
self.assertTrue(is_prime(3))
self.assertTrue(is_prime(5))
self.assertTrue(is_prime(7))
self.assertTrue(is_prime(11))
self.assertTrue(is_prime(13))
self.assertTrue(is_prime(101))
self.assertFalse(is_prime(0))
self.assertFalse(is_prime(1))
self.assertFalse(is_prime(4))
self.assertFalse(is_prime(6))
self.assertFalse(is_prime(8))
self.assertFalse(is_prime(9))
self.assertFalse(is_prime(10))
self.assertFalse(is_prime(12))
self.assertFalse(is_prime(323))
def test_get_nth_prime(self):
self.assertEqual(prime_nth(1), 2)
self.assertEqual(prime_nth(2), 3)
self.assertEqual(prime_nth(3), 5)
self.assertEqual(prime_nth(4), 7)
self.assertEqual(prime_nth(5), 11)
self.assertEqual(prime_nth(6), 13)
def test_primes(self):
self.assertEqual(primes(19), [2, 3, 5, 7, 11, 13, 17, 19])
self.assertEqual(primes(20), [2, 3, 5, 7, 11, 13, 17, 19])
self.assertEqual(primes(25), [2, 3, 5, 7, 11, 13, 17, 19, 23])
self.assertEqual(primes(1), [])
if __name__ == '__main__':
unittest.main()