aboutsummaryrefslogtreecommitdiff
path: root/challenge-252/roger-bell-west/python/ch-2.py
diff options
context:
space:
mode:
Diffstat (limited to 'challenge-252/roger-bell-west/python/ch-2.py')
-rwxr-xr-xchallenge-252/roger-bell-west/python/ch-2.py23
1 files changed, 23 insertions, 0 deletions
diff --git a/challenge-252/roger-bell-west/python/ch-2.py b/challenge-252/roger-bell-west/python/ch-2.py
new file mode 100755
index 0000000000..f8bf61a527
--- /dev/null
+++ b/challenge-252/roger-bell-west/python/ch-2.py
@@ -0,0 +1,23 @@
+#! /usr/bin/python3
+
+def uniquesumzero(n):
+ if n == 1:
+ return [0]
+ p = list(range(1, n))
+ p.append(-n * (n - 1) / 2)
+ return p
+
+import unittest
+
+class TestUniquesumzero(unittest.TestCase):
+
+ def test_ex1(self):
+ self.assertEqual(uniquesumzero(5), [1, 2, 3, 4, -10], 'example 1')
+
+ def test_ex2(self):
+ self.assertEqual(uniquesumzero(3), [1, 2, -3], 'example 2')
+
+ def test_ex3(self):
+ self.assertEqual(uniquesumzero(1), [0], 'example 3')
+
+unittest.main()