Moved 1 to 5 to Python and learned a ton.
This commit is contained in:
25
python/lib_prime_tests.py
Normal file
25
python/lib_prime_tests.py
Normal 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()
|
||||
Reference in New Issue
Block a user