diff --git a/exp/tests/test_geometry.py b/exp/tests/test_geometry.py --- a/exp/tests/test_geometry.py +++ b/exp/tests/test_geometry.py @@ -2,6 +2,8 @@ import math import random from unittest import TestCase +import numpy as np + from geometry import Line,EPoint random.seed(361) @@ -89,3 +91,15 @@ class TestLine(TestCase): q_=q.shiftBasis(newBasis) self.assertAlmostEqual(q_._alpha,math.pi*7/4) self.assertAlmostEqual(q_._d,3/2*diag) + + def testTransform(self): + rot90=np.array([[0,-1,0],[1,0,0],[0,0,1]]) + p=Line(math.pi/4,10) + p_=p.transform(rot90) + self.assertAlmostEqual(p_.alpha,math.pi*3/4) + self.assertAlmostEqual(p_.d,p.d) + + shiftXY10=np.array([[1,0,10],[0,1,10],[0,0,1]]) + p_=p.transform(shiftXY10) + self.assertAlmostEqual(p_.alpha,p.alpha) + self.assertAlmostEqual(p_.d,10+10*math.sqrt(2))