1 files changed, 27 insertions, 0 deletions
diff --git a/challenge-325/roger-bell-west/python/ch-2.py b/challenge-325/roger-bell-west/python/ch-2.py
new file mode 100755
index 0000000000..9a6a8f82d2
--- /dev/null
+++ b/challenge-325/roger-bell-west/python/ch-2.py
@@ -0,0 +1,27 @@
+#! /usr/bin/python3
+
+def finalprice(a):
+  out = []
+  for i, n in enumerate(a):
+    discount = 0
+    for mi in range(i + 1, len(a)):
+      if a[mi] <= n:
+        discount = a[mi]
+        break
+    out.append(n - discount)
+  return out
+
+import unittest
+
+class TestFinalprice(unittest.TestCase):
+
+  def test_ex1(self):
+    self.assertEqual(finalprice([8, 4, 6, 2, 3]), [4, 2, 4, 2, 3], 'example 1')
+
+  def test_ex2(self):
+    self.assertEqual(finalprice([1, 2, 3, 4, 5]), [1, 2, 3, 4, 5], 'example 2')
+
+  def test_ex3(self):
+    self.assertEqual(finalprice([7, 1, 1, 5]), [6, 0, 1, 5], 'example 3')
+
+unittest.main()