pwc261 solution in python

1 files changed, 71 insertions, 0 deletions

diff --git a/challenge-261/pokgopun/python/ch-1.py b/challenge-261/pokgopun/python/ch-1.py
new file mode 100644
index 0000000000..073ab918fa
--- /dev/null
+++ b/challenge-261/pokgopun/python/ch-1.py
@@ -0,0 +1,71 @@
+### https://theweeklychallenge.org/blog/perl-weekly-challenge-261/
+"""
+
+Task 1: Element Digit Sum
+
+Submitted by: [45]Mohammad Sajid Anwar
+     __________________________________________________________________
+
+   You are given an array of integers, @ints.
+
+   Write a script to evaluate the absolute difference between element and
+   digit sum of the given array.
+
+Example 1
+
+Input: @ints = (1,2,3,45)
+Output: 36
+
+Element Sum: 1 + 2 + 3 + 45 = 51
+Digit Sum: 1 + 2 + 3 + 4 + 5 = 15
+Absolute Difference: | 51 - 15 | = 36
+
+Example 2
+
+Input: @ints = (1,12,3)
+Output: 9
+
+Element Sum: 1 + 12 + 3 = 16
+Digit Sum: 1 + 1 + 2 + 3 = 7
+Absolute Difference: | 16 - 7 | = 9
+
+Example 3
+
+Input: @ints = (1,2,3,4)
+Output: 0
+
+Element Sum: 1 + 2 + 3 + 4 = 10
+Digit Sum: 1 + 2 + 3 + 4 = 10
+Absolute Difference: | 10 - 10 | = 0
+
+Example 4
+
+Input: @ints = (236, 416, 336, 350)
+Output: 1296
+
+Task 2: Multiply by Two
+"""
+### solution by pokgopun@gmail.com
+
+from itertools import chain
+
+def EDS(ints: tuple):
+    return sum(ints) - sum(
+            int(e) for e in chain.from_iterable(
+                str(i) for i in ints
+                )
+            )
+
+import unittest
+
+class TestEDS(unittest.TestCase):
+    def test(self):
+        for inpt, otpt in {
+                (1,2,3,45): 36,
+                (1,12,3): 9,
+                (1,2,3,4): 0,
+                (236, 416, 336, 350): 1296,
+                }.items():
+            self.assertEqual(EDS(inpt),otpt)
+
+unittest.main()