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

Moved more problems to Python.

Browse Source
This commit is contained in:
Felix Martin
2019-07-14 23:58:22 -04:00
parent 86e68eeee2
commit 0ab214633e
14 changed files with 476 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
python/lib_prime_tests.py
View File

@@ -5,12 +5,14 @@ try:
from .lib_prime import is_prime
from .lib_prime import prime_nth
from .lib_prime import primes
from .lib_prime import get_divisors_count
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
from lib_prime import get_divisors_count
class TestPrimeMethods(unittest.TestCase):
@@ -23,6 +25,7 @@ class TestPrimeMethods(unittest.TestCase):
self.assertEqual(prime_factors(147), [3, 7, 7])
def test_prime_factors_count(self):
self.assertEqual(prime_factors_count(1), {})
self.assertEqual(prime_factors_count(2), {2: 1})
self.assertEqual(prime_factors_count(147), {3: 1, 7: 2})
@@ -59,6 +62,15 @@ class TestPrimeMethods(unittest.TestCase):
self.assertEqual(primes(25), [2, 3, 5, 7, 11, 13, 17, 19, 23])
self.assertEqual(primes(1), [])
def test_get_divisors_count(self):
self.assertEqual(get_divisors_count(1), 1)
self.assertEqual(get_divisors_count(3), 2)
self.assertEqual(get_divisors_count(6), 4)
self.assertEqual(get_divisors_count(10), 4)
self.assertEqual(get_divisors_count(15), 4)
self.assertEqual(get_divisors_count(21), 4)
self.assertEqual(get_divisors_count(28), 6)
if __name__ == '__main__':
unittest.main()