ByoungSooPark/cadquery
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

More tests for the Location class

Browse Source
This commit is contained in:
adam-urbanczyk
2020-06-22 17:35:58 +02:00
parent c82e908e61
commit 86810448e1
1 changed files with 17 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
tests/test_cad_objects.py
View File

@ -1,20 +1,14 @@
# system modules # system modules
import math import math
import sys
import unittest import unittest
from tests import BaseTest from tests import BaseTest
from OCP.gp import gp_Vec, gp_Pnt, gp_Ax2, gp_Circ, gp_Elips, gp, gp_XYZ from OCP.gp import gp_Vec, gp_Pnt, gp_Ax2, gp_Circ, gp_Elips, gp, gp_XYZ
from OCP.BRepBuilderAPI import ( from OCP.BRepBuilderAPI import BRepBuilderAPI_MakeEdge
BRepBuilderAPI_MakeVertex,
BRepBuilderAPI_MakeEdge,
BRepBuilderAPI_MakeFace,
)
from OCP.GC import GC_MakeCircle
from cadquery import * from cadquery import *
DEG2RAD = 2 * math.pi / 360 DEG2RAD = math.pi / 180
RAD2DEG = 180 / math.pi
class TestCadObjects(BaseTest): class TestCadObjects(BaseTest):
@ -400,6 +394,20 @@ class TestCadObjects(BaseTest):
Plane(origin=(0, 0, 0), xDir=(1, 0, 0), normal=(0, 1, 1)), Plane(origin=(0, 0, 0), xDir=(1, 0, 0), normal=(0, 1, 1)),
) )
def testLocation(self):
# Vector
loc1 = Location(Vector(0, 0, 1))
T = loc1.wrapped.Transformation().TranslationPart()
self.assertTupleAlmostEquals((T.X(), T.Y(), T.Z()), (0, 0, 1), 6)
# rotation + translation
loc2 = Location(Vector(0, 0, 1), Vector(0, 0, 1), 45)
angle = loc2.wrapped.Transformation().GetRotation().GetRotationAngle() * RAD2DEG
self.assertAlmostEqual(45, angle)
if __name__ == "__main__": if __name__ == "__main__":
unittest.main() unittest.main()