def getRotatedRectanglePoints(angle, base_point, height, width):
# base_point is the upper left (ul)
ur = point.Point(width * math.cos(angle), -width * math.sin(angle))
lr = point.Point(ur.x + height * math.sin(angle),
ur.y + height * math.cos(angle))
ll = point.Point(height * math.cos(math.pi / 2 - angle),
height * math.sin(math.pi / 2 - angle))
return tuple(base_point + pt for pt in (point.Point(0, 0), ur, lr, ll))
def testGetBounds(self):
points = [
point.Point(0, 0),
point.Point(1, 0),
point.Point(0, 1),
point.Point(1, 1)
]
correct = util.Bound(0, 0, 1, 1)
self.assertNamedTupleAlmostEqual(correct, util.getBounds(points))
def testGetRotatedRectanglePoints(self):
"""Rotate a unit square by 45 degrees"""
m = math.sqrt(2) / 2
correct = (point.Point(0, 0), point.Point(m,
-m), point.Point(2 * m, 0),
point.Point(m, m))
results = util.getRotatedRectanglePoints(math.pi / 4,
point.Point(0, 0), 1, 1)
for points in zip(correct, results):
self.assertPointEqual(*points)
def getCenter(im):
return point.Point(*(d / 2 for d in im.size))
def getBoundsCenter(bounds):
return point.Point((bounds.right - bounds.left) / 2 + bounds.left,
(bounds.lower - bounds.upper) / 2 + bounds.upper)
def testGetBoundsCenter(self):
self.assertPointEqual(point.Point(.5, 1),
util.getBoundsCenter(util.Bound(0, 0, 1, 2)))
def testGetCenter(self):
im = mock.Mock()
im.size = (23, 82)
self.assertPointEqual(point.Point(11.5, 41), util.getCenter(im))
def testAdd(self):
self.assertPointEqual(point.Point(2, 3),
point.Point(1, 1) + point.Point(1, 2))
def testRotate90Degrees(self):
p = point.Point(1, 0)
center = point.Point(0, 0)
result = point.Point(0, 1)
self.assertPointEqual(result, p.rotate(center, math.pi / 2))
def testRecenter(self):
p = point.Point(1, 1)
old_center = point.Point(0, 0)
new_center = point.Point(1, 1)
result = point.Point(2, 2)
self.assertPointEqual(result, p.recenter(old_center, new_center))
def testSubtract(self):
self.assertPointEqual(point.Point(1, 0),
point.Point(2, 1) - point.Point(1, 1))