Task 1 Python

1 files changed, 60 insertions, 0 deletions

diff --git a/challenge-053/lubos-kolouch/python/ch-1.py b/challenge-053/lubos-kolouch/python/ch-1.py
new file mode 100644
index 0000000000..843c17a699
--- /dev/null
+++ b/challenge-053/lubos-kolouch/python/ch-1.py
@@ -0,0 +1,60 @@
+#!/usr/bin/env python
+
+import numpy as np
+import pytest
+
+
+class ArrRotator:
+
+    def __init__(self):
+        self.arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
+
+    def check_degree(self, degree: int):
+        if degree not in (90, 180, 270):
+            raise ValueError('degree must be 90, 180 or 270')
+
+    def rotate_arr(self, degree: int):
+        self.check_degree(degree)
+
+        result_arr = self.arr
+
+        for i in range(degree // 90):
+            result_arr = np.rot90(result_arr, -1)
+
+        return result_arr
+
+
+class RotatorTest:
+    # TESTS
+
+    def __init__(self):
+        self.rotator = ArrRotator()
+
+    def do_tests(self):
+
+        assert np.alltrue(self.rotator.rotate_arr(degree=90)
+                          == [[7, 4, 1], [8, 5, 2], [9, 6, 3]])
+        assert np.alltrue(self.rotator.rotate_arr(degree=180)
+                          == [[9, 8, 7], [6, 5, 4], [3, 2, 1]])
+        assert np.alltrue(self.rotator.rotate_arr(degree=270)
+                          == [[3, 6, 9], [2, 5, 8], [1, 4, 7]])
+        with pytest.raises(ValueError):
+            rotator.rotate_arr(degree=222)
+
+
+if __name__ == "__main__":
+    rotator = ArrRotator()
+
+    print('Original array:')
+    print(rotator.arr)
+    print('Rotate by 90 : ')
+    print(rotator.rotate_arr(degree=90))
+
+    print('Rotate by 180 : ')
+    print(rotator.rotate_arr(degree=180))
+
+    print('Rotate by 270 : ')
+    print(rotator.rotate_arr(degree=270))
+
+    tester = RotatorTest()
+    tester.do_tests()