summaryrefslogtreecommitdiff
path: root/qml/python/tests/test_transformations.py
diff options
context:
space:
mode:
Diffstat (limited to 'qml/python/tests/test_transformations.py')
-rw-r--r--qml/python/tests/test_transformations.py14
1 files changed, 13 insertions, 1 deletions
diff --git a/qml/python/tests/test_transformations.py b/qml/python/tests/test_transformations.py
index b802b9f..d244e48 100644
--- a/qml/python/tests/test_transformations.py
+++ b/qml/python/tests/test_transformations.py
@@ -3,7 +3,7 @@
import unittest
-from python.transformations import Rotation, Translation, Symmetry
+from python.transformations import Rotation, Translation, Symmetry, ToIndex
class FakeBoard():
@@ -132,3 +132,15 @@ class TestRotation(unittest.TestCase):
symmetry.y_flip = False
self.assertFalse(symmetry.is_valid())
+
+class TestToIndex(unittest.TestCase):
+ """ Test the toIndex transformation.
+ """
+
+ def test_apply_points(self):
+ """ Test the points index.
+ """
+ toIndex = ToIndex(FakeBoard(2, 1, 8, 5))
+ self.assertEqual(0, toIndex.apply_points((2, 1)))
+ self.assertEqual(7, toIndex.apply_points((2, 2)))
+ self.assertEqual(8, toIndex.apply_points((3, 2)))