00001 import unittest
00002 from PyKDL import *
00003 from math import *
00004
00005 class FrameVelTestFunctions(unittest.TestCase):
00006
00007 def testVectorVel(self):
00008 v=VectorVel(Vector(3,-4,5),Vector(6,3,-5))
00009 vt = Vector(-4,-6,-8);
00010 self.assert_(Equal( 2*v-v,v))
00011 self.assert_(Equal( v*2-v,v))
00012 self.assert_(Equal( v+v+v-2*v,v))
00013 v2=VectorVel(v)
00014 self.assert_(Equal( v,v2))
00015 v2+=v
00016 self.assert_(Equal( 2*v,v2))
00017 v2-=v
00018 self.assert_(Equal( v,v2))
00019 v2.ReverseSign()
00020 self.assert_(Equal( v,-v2))
00021 self.assert_(Equal( v*vt,-vt*v))
00022 v2 = VectorVel(Vector(-5,-6,-3),Vector(3,4,5))
00023 self.assert_(Equal( v*v2,-v2*v))
00024
00025
00026 def testRotationVel(self):
00027 v=VectorVel(Vector(9,4,-2),Vector(-5,6,-2))
00028 vt=Vector(2,3,4)
00029 a= radians(-15)
00030 b= radians(20)
00031 c= radians(-80)
00032 R = RotationVel(Rotation.RPY(a,b,c),Vector(2,4,1))
00033 R2=RotationVel(R)
00034 self.assert_(Equal(R,R2))
00035 self.assert_(Equal((R*v).Norm(),(v.Norm())))
00036 self.assert_(Equal(R.Inverse(R*v),v))
00037 self.assert_(Equal(R*R.Inverse(v),v))
00038 self.assert_(Equal(R*Rotation.Identity(),R))
00039 self.assert_(Equal(Rotation.Identity()*R,R))
00040 self.assert_(Equal(R*(R*(R*v)),(R*R*R)*v))
00041 self.assert_(Equal(R*(R*(R*vt)),(R*R*R)*vt))
00042 self.assert_(Equal(R*R.Inverse(),RotationVel.Identity()))
00043 self.assert_(Equal(R.Inverse()*R,RotationVel.Identity()))
00044 self.assert_(Equal(R.Inverse()*v,R.Inverse(v)))
00045
00046
00047
00048
00049 def testFrameVel(self):
00050 v=VectorVel(Vector(3,4,5),Vector(-2,-4,-1))
00051 vt=Vector(-1,0,-10)
00052 F = FrameVel(Frame(Rotation.EulerZYX(radians(10),radians(20),radians(-10)),Vector(4,-2,1)),
00053 Twist(Vector(2,-2,-2),Vector(-5,-3,-2)))
00054 F2=FrameVel(F)
00055 self.assert_(Equal( F,F2))
00056 self.assert_(Equal( F.Inverse(F*v),v))
00057 self.assert_(Equal( F.Inverse(F*vt), vt))
00058 self.assert_(Equal( F*F.Inverse(v),v))
00059 self.assert_(Equal( F*F.Inverse(vt),vt))
00060 self.assert_(Equal( F*Frame.Identity(),F))
00061 self.assert_(Equal( Frame.Identity()*F,F))
00062 self.assert_(Equal( F*(F*(F*v)),(F*F*F)*v))
00063 self.assert_(Equal( F*(F*(F*vt)),(F*F*F)*vt))
00064 self.assert_(Equal( F*F.Inverse(),FrameVel.Identity()))
00065 self.assert_(Equal( F.Inverse()*F,Frame.Identity()))
00066 self.assert_(Equal( F.Inverse()*vt,F.Inverse(vt)))
00067
00068
00069
00070
00071
00072
00073
00074
00075
00076
00077
00078
00079
00080
00081
00082
00083
00084
00085
00086
00087
00088
00089
00090
00091
00092
00093
00094
00095
00096
00097
00098
00099
00100
00101
00102
00103
00104
00105
00106
00107
00108
00109
00110
00111
00112
00113
00114
00115
00116
00117
00118
00119
00120
00121
00122
00123
00124
00125
00126
00127
00128
00129
00130
00131
00132
00133
00134
00135
00136
00137
00138
00139
00140
00141
00142
00143
00144
00145
00146
00147
00148
00149
00150
00151
00152
00153
00154
00155
00156
00157
00158
00159
00160 def suite():
00161 suite=unittest.TestSuite()
00162 suite.addTest(FrameVelTestFunctions('testVectorVel'))
00163 suite.addTest(FrameVelTestFunctions('testRotationVel'))
00164 suite.addTest(FrameVelTestFunctions('testFrameVel'))
00165 return suite
00166
00167
00168
00169
00170
00171