1 files changed, 134 insertions, 0 deletions
diff --git a/qml/python/tests/test_transformations.py b/qml/python/tests/test_transformations.py
new file mode 100644
index 0000000..b802b9f
--- /dev/null
+++ b/qml/python/tests/test_transformations.py
@@ -0,0 +1,134 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+import unittest
+
+from python.transformations import Rotation, Translation, Symmetry
+
+class FakeBoard():
+
+    def __init__(self, min_x, min_y, max_x, max_y):
+        self.min_x = min_x
+        self.min_y = min_y
+        self.max_x = max_x
+        self.max_y = max_y
+
+    def get_size(self):
+        x_size = self.max_x - self.min_x
+        y_size = self.max_y - self.min_y
+        return min(19, x_size), min(19, y_size)
+
+class TestRotation(unittest.TestCase):
+    """ Test the rotation transformation
+    """
+
+    def test_apply_points(self):
+        """ Test the points rotation.
+        """
+        rotation = Rotation(FakeBoard(2, 1, 8, 5))
+        self.assertEqual((4, 2), rotation.apply_points((2, 1)))
+
+        rotation = Rotation(FakeBoard(0, 0, 6, 4))
+        self.assertEqual((4, 0), rotation.apply_points((0, 0)))
+        self.assertEqual((4, 3), rotation.apply_points((3, 0)))
+        self.assertEqual((1, 0), rotation.apply_points((0, 3)))
+        self.assertEqual((1, 3), rotation.apply_points((3, 3)))
+
+        rotation = Rotation(FakeBoard(1, 1, 6, 4))
+        self.assertEqual((4, 1), rotation.apply_points((1, 0)))
+
+
+
+    def test_get_new_size(self):
+        """ Test the goban size rotation.
+        """
+        rotation = Rotation(FakeBoard(2, 1, 8, 5))
+        self.assertEqual((0, 2, 4, 8), rotation.get_new_size())
+
+        rotation = Rotation(FakeBoard(0, 0, 6, 4))
+        self.assertEqual((0, 0, 4, 6), rotation.get_new_size())
+
+    def test_is_valid(self):
+        """ Test the is_valid method.
+        """
+
+        # Do not rotate if height > width
+        rotation = Rotation(FakeBoard(0, 0, 6, 4))
+        self.assertTrue(rotation.is_valid())
+
+        # Always rotate if width > height
+        rotation = Rotation(FakeBoard(0, 0, 4, 6))
+        self.assertFalse(rotation.is_valid())
+
+        # May rotate if width = height (not tested here…)
+
+class TestTranslation(unittest.TestCase):
+    """ Test the translation transformation.
+    """
+
+    def test_apply_points(self):
+        """ Test the points translation.
+        """
+        translation = Translation(FakeBoard(2, 1, 8, 5))
+        self.assertEqual((0, 0), translation.apply_points((2, 1)))
+        self.assertEqual((6, 4), translation.apply_points((8, 5)))
+
+    def test_get_new_size(self):
+        """ Test the goban size translation.
+        """
+        translation = Translation(FakeBoard(2, 1, 8, 5))
+
+        self.assertEqual((0, 0, 6, 4), translation.get_new_size())
+
+    def test_is_valid(self):
+        """ Test the is_valid method.
+        """
+
+        translation = Translation(FakeBoard(1, 0, 6, 4))
+        self.assertTrue(translation.is_valid())
+
+        translation = Translation(FakeBoard(0, 1, 6, 4))
+        self.assertTrue(translation.is_valid())
+
+        translation = Translation(FakeBoard(0, 0, 4, 6))
+        self.assertFalse(translation.is_valid())
+
+class TestRotation(unittest.TestCase):
+    """ Test the simetry transformation.
+    """
+
+    def test_apply_points(self):
+        """ Test the points Symmetry.
+        """
+        symmetry = Symmetry(FakeBoard(2, 1, 8, 5))
+        symmetry.x_flip = True
+        symmetry.y_flip = False
+        self.assertEqual((8, 1), symmetry.apply_points((2, 1)))
+        self.assertEqual((2, 5), symmetry.apply_points((8, 5)))
+
+        symmetry.x_flip = False
+        symmetry.y_flip = True
+        self.assertEqual((2, 5), symmetry.apply_points((2, 1)))
+        self.assertEqual((8, 1), symmetry.apply_points((8, 5)))
+
+    def test_get_new_size(self):
+        """ Test the goban size Symmetry.
+        """
+        symmetry = Symmetry(FakeBoard(2, 1, 8, 5))
+
+        self.assertEqual((2, 1, 8, 5), symmetry.get_new_size())
+
+    def test_is_valid(self):
+        """ Test the is_valid method.
+        """
+
+        symmetry = Symmetry(FakeBoard(1, 0, 6, 4))
+        symmetry.x_flip = True
+        self.assertTrue(symmetry.is_valid())
+
+
+        symmetry = Symmetry(FakeBoard(0, 0, 4, 6))
+        symmetry.x_flip = False
+        symmetry.y_flip = False
+        self.assertFalse(symmetry.is_valid())
+