diff options
Diffstat (limited to 'challenge-324/sgreen/python/test.py')
| -rwxr-xr-x | challenge-324/sgreen/python/test.py | 21 |
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() |
