Moved 1 to 5 to Python and learned a ton.

This commit is contained in:
2019-07-13 23:58:38 -04:00
parent 414cf4b074
commit b246d56acd
13 changed files with 402 additions and 2 deletions

25
python/lib_prime_tests.py Normal file
View File

@@ -0,0 +1,25 @@
import unittest
try:
from .lib_prime import prime_factors
from .lib_prime import prime_factors_count
except ModuleNotFoundError:
from lib_prime import prime_factors
from lib_prime import prime_factors_count
class TestPrimeMethods(unittest.TestCase):
def test_prime_factors(self):
self.assertEqual(prime_factors(2), [2])
self.assertEqual(prime_factors(5), [5])
self.assertEqual(prime_factors(10), [2, 5])
self.assertEqual(prime_factors(13), [13])
self.assertEqual(prime_factors(147), [3, 7, 7])
def test_prime_factors_count(self):
self.assertEqual(prime_factors_count(2), {2: 1})
self.assertEqual(prime_factors_count(147), {3: 1, 7: 2})
if __name__ == '__main__':
unittest.main()