1 files changed, 36 insertions, 0 deletions
diff --git a/challenge-337/roger-bell-west/python/ch-2.py b/challenge-337/roger-bell-west/python/ch-2.py
new file mode 100755
index 0000000000..1137f8491b
--- /dev/null
+++ b/challenge-337/roger-bell-west/python/ch-2.py
@@ -0,0 +1,36 @@
+#! /usr/bin/python3
+
+def oddmatrix(rows, cols, points):
+  rm = set()
+  cm = set()
+  for p in points:
+    if p[0] in rm:
+      rm.discard(p[0])
+    else:
+      rm.add(p[0])
+    if p[1] in cm:
+      cm.discard(p[1])
+    else:
+      cm.add(p[1])
+  return len(rm) * (cols - len(cm)) + len(cm) * (rows - len(rm))
+
+import unittest
+
+class TestOddmatrix(unittest.TestCase):
+
+  def test_ex1(self):
+    self.assertEqual(oddmatrix(2, 3, [[0, 1], [1, 1]]), 6, 'example 1')
+
+  def test_ex2(self):
+    self.assertEqual(oddmatrix(2, 2, [[1, 1], [0, 0]]), 0, 'example 2')
+
+  def test_ex3(self):
+    self.assertEqual(oddmatrix(3, 3, [[0, 0], [1, 2], [2, 1]]), 0, 'example 3')
+
+  def test_ex4(self):
+    self.assertEqual(oddmatrix(1, 5, [[0, 2], [0, 4]]), 2, 'example 4')
+
+  def test_ex5(self):
+    self.assertEqual(oddmatrix(4, 2, [[1, 0], [3, 1], [2, 0], [0, 1]]), 8, 'example 5')
+
+unittest.main()