aboutsummaryrefslogtreecommitdiff
path: root/challenge-324/sgreen/python/test.py
diff options
context:
space:
mode:
Diffstat (limited to 'challenge-324/sgreen/python/test.py')
-rwxr-xr-xchallenge-324/sgreen/python/test.py21
1 files changed, 21 insertions, 0 deletions
diff --git a/challenge-324/sgreen/python/test.py b/challenge-324/sgreen/python/test.py
new file mode 100755
index 0000000000..9203697c3b
--- /dev/null
+++ b/challenge-324/sgreen/python/test.py
@@ -0,0 +1,21 @@
+#!/usr/bin/env python3
+
+import unittest
+ch_1 = __import__('ch-1')
+ch_2 = __import__('ch-2')
+
+
+class TestClass(unittest.TestCase):
+ def test_ch_1(self):
+ self.assertEqual(ch_1.twod_array([1, 2, 3, 4], 2, 2), [[1, 2], [3, 4]])
+ self.assertEqual(ch_1.twod_array([1, 2, 3], 1, 3), [[1, 2, 3]])
+ self.assertEqual(ch_1.twod_array([1, 2, 3, 4], 4, 1), [[1], [2], [3], [4]])
+
+ def test_ch_2(self):
+ self.assertEqual(ch_2.total_xor([1, 3]), 6)
+ self.assertEqual(ch_2.total_xor([5, 1, 6]), 28)
+ self.assertEqual(ch_2.total_xor([3, 4, 5, 6, 7, 8]), 480)
+
+
+if __name__ == '__main__':
+ unittest.main()