1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
|
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import unittest
from python.transformations import Rotation, Translation, Symmetry, ToIndex
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())
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)))
|
