diff options
Diffstat (limited to 'challenge-078/roger-bell-west/python/ch-2.py')
| -rwxr-xr-x | challenge-078/roger-bell-west/python/ch-2.py | 30 |
1 files changed, 30 insertions, 0 deletions
diff --git a/challenge-078/roger-bell-west/python/ch-2.py b/challenge-078/roger-bell-west/python/ch-2.py new file mode 100755 index 0000000000..860ec393d6 --- /dev/null +++ b/challenge-078/roger-bell-west/python/ch-2.py @@ -0,0 +1,30 @@ +#! /usr/bin/python + +import unittest + +def leftrot(a,b): + l=len(a) + t=list() + map(t.append,a) + map(t.append,a) + o=list() + for c in (b): + o.append(list((t[c:c+l]))) + return o + +class TestLeftrot(unittest.TestCase): + + def test_ex1(self): + self.assertEqual(leftrot((10, 20, 30, 40, 50),(3, 4)), + [[40, 50, 10, 20, 30], + [50, 10, 20, 30, 40]], + 'example 1') + + def test_ex2(self): + self.assertEqual(leftrot((7, 4, 2, 6, 3),(1, 3, 4)), + [[4, 2, 6, 3, 7], + [6, 3, 7, 4, 2], + [3, 7, 4, 2, 6]], + 'example 2') + +unittest.main() |
