Challenge 1 Week 41 Python

1 files changed, 24 insertions, 0 deletions

diff --git a/challenge-041/lubos-kolouch/python/ch-1.py b/challenge-041/lubos-kolouch/python/ch-1.py
new file mode 100644
index 0000000000..e32db9a798
--- /dev/null
+++ b/challenge-041/lubos-kolouch/python/ch-1.py
@@ -0,0 +1,24 @@
+#!python3
+
+import sys
+from sympy.ntheory import factorint, isprime
+import unittest
+
+def is_attractive(what):
+    factors = factorint(int(what))
+    return (isprime(sum(factors.values())))
+
+
+# TESTS
+
+class PrimeTest(unittest.TestCase):
+    def test_20(self):
+        self.assertEqual(is_attractive(20),True)
+    def test_128(self):
+        self.assertEqual(is_attractive(128),True)
+    def test_256(self):
+        self.assertEqual(is_attractive(256),False)
+
+if __name__ == "__main__":
+    unittest.main()
+