Merge pull request #9415 from Firedrake/rogerbw-challenge-252

RogerBW solutions for challenge no. 252

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()