1 files changed, 23 insertions, 0 deletions
diff --git a/challenge-199/roger-bell-west/python/ch-2.py b/challenge-199/roger-bell-west/python/ch-2.py
new file mode 100755
index 0000000000..b96dac98dc
--- /dev/null
+++ b/challenge-199/roger-bell-west/python/ch-2.py
@@ -0,0 +1,23 @@
+#! /usr/bin/python3
+
+import unittest
+
+def goodtriplets(a, x, y, z):
+  c = 0
+  for i in range(len(a) - 2):
+    for j in range(i + 1, len(a) - 1):
+      if abs(a[i] - a[j]) <= x:
+        for k in range(j + 1, len(a)):
+          if abs(a[j] - a[k]) <= y and abs(a[i] - a[k]) <= z:
+            c += 1
+  return c
+
+class TestGoodtriplets(unittest.TestCase):
+
+  def test_ex1(self):
+    self.assertEqual(goodtriplets([3, 0, 1, 1, 9, 7], 7, 2, 3), 4, 'example 1')
+
+  def test_ex2(self):
+    self.assertEqual(goodtriplets([1, 1, 2, 2, 3], 0, 0, 1), 0, 'example 2')
+
+unittest.main()