# *************************************************************************** # * Copyright (c) 2011 Juergen Riegel * # * * # * This program is free software; you can redistribute it and/or modify * # * it under the terms of the GNU Lesser General Public License (LGPL) * # * as published by the Free Software Foundation; either version 2 of * # * the License, or (at your option) any later version. * # * for detail see the LICENCE text file. * # * * # * This program is distributed in the hope that it will be useful, * # * but WITHOUT ANY WARRANTY; without even the implied warranty of * # * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * # * GNU Library General Public License for more details. * # * * # * You should have received a copy of the GNU Library General Public * # * License along with this program; if not, write to the Free Software * # * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 * # * USA * # * * # *************************************************************************** from math import pi, sqrt import unittest import FreeCAD class TestPrimitive(unittest.TestCase): def setUp(self): self.Doc = FreeCAD.newDocument("PartDesignTestPrimitive") def testPrimitiveBox(self): self.Body = self.Doc.addObject("PartDesign::Body", "Body") self.Box = self.Doc.addObject("PartDesign::AdditiveBox", "Box") self.Box.Length = 11 self.Box.Width = 11 self.Box.Height = 11 self.Body.addObject(self.Box) self.Doc.recompute() self.Box001 = self.Doc.addObject("PartDesign::SubtractiveBox", "Box001") self.Box001.Length = 10 self.Box001.Width = 10 self.Box001.Height = 10 self.Body.addObject(self.Box001) self.Doc.recompute() self.assertAlmostEqual(self.Box001.Shape.Volume, 11**3 - 10**3) def testPrimitiveCylinder(self): self.Body = self.Doc.addObject("PartDesign::Body", "Body") self.Cylinder = self.Doc.addObject("PartDesign::AdditiveCylinder", "Cylinder") self.Cylinder.Radius = 11 self.Cylinder.Height = 10 self.Cylinder.Angle = 360 self.Body.addObject(self.Cylinder) self.Doc.recompute() self.Cylinder001 = self.Doc.addObject("PartDesign::SubtractiveCylinder", "Cylinder001") self.Cylinder001.Radius = 10 self.Cylinder001.Height = 10 self.Cylinder001.Angle = 360 self.Body.addObject(self.Cylinder001) self.Doc.recompute() self.assertAlmostEqual(self.Cylinder001.Shape.Volume, pi * 10 * (11**2 - 10**2)) def testPrimitiveSphere(self): self.Body = self.Doc.addObject("PartDesign::Body", "Body") self.Sphere = self.Doc.addObject("PartDesign::AdditiveSphere", "Sphere") self.Sphere.Radius = 6 self.Body.addObject(self.Sphere) self.Doc.recompute() self.Sphere001 = self.Doc.addObject("PartDesign::SubtractiveSphere", "Sphere001") self.Sphere001.Radius = 5 self.Body.addObject(self.Sphere001) self.Doc.recompute() self.assertAlmostEqual(self.Sphere001.Shape.Volume, 4 / 3.0 * pi * (6**3 - 5**3)) def testPrimitiveCone(self): self.Body = self.Doc.addObject("PartDesign::Body", "Body") self.Cone = self.Doc.addObject("PartDesign::AdditiveCone", "Cone") self.Cone.Radius1 = 0 self.Cone.Radius2 = 4 self.Cone.Height = 10 self.Body.addObject(self.Cone) self.Doc.recompute() self.Cone001 = self.Doc.addObject("PartDesign::SubtractiveCone", "Cone") self.Cone001.Radius1 = 0 self.Cone001.Radius2 = 3 self.Cone001.Height = 10 self.Body.addObject(self.Cone001) self.Doc.recompute() self.assertAlmostEqual(self.Cone001.Shape.Volume, 1 / 3.0 * pi * 10 * (4**2 - 3**2)) def testPrimitiveEllipsoid(self): self.Body = self.Doc.addObject("PartDesign::Body", "Body") self.Ellipsoid = self.Doc.addObject("PartDesign::AdditiveEllipsoid", "Ellipsoid") self.Ellipsoid.Radius1 = 2 self.Ellipsoid.Radius2 = 4 self.Body.addObject(self.Ellipsoid) self.Doc.recompute() self.Ellipsoid001 = self.Doc.addObject("PartDesign::SubtractiveEllipsoid", "Ellipsoid001") self.Ellipsoid001.Radius1 = 1.5 self.Ellipsoid001.Radius2 = 3 self.Body.addObject(self.Ellipsoid001) self.Doc.recompute() self.assertAlmostEqual( self.Ellipsoid001.Shape.Volume, 4 / 3.0 * pi * (2 * 4**2 - 1.5 * 3**2), places=1 ) def testPrimitiveTorus(self): self.Body = self.Doc.addObject("PartDesign::Body", "Body") self.Torus = self.Doc.addObject("PartDesign::AdditiveTorus", "Torus") self.Torus.Radius1 = 10 self.Torus.Radius2 = 4 self.Body.addObject(self.Torus) self.Doc.recompute() self.Torus001 = self.Doc.addObject("PartDesign::SubtractiveTorus", "Torus001") self.Torus001.Radius1 = 10 self.Torus001.Radius2 = 3 self.Body.addObject(self.Torus001) self.Doc.recompute() self.assertAlmostEqual(self.Torus001.Shape.Volume, 2 * pi**2 * 10 * (4**2 - 3**2)) def testPrimitivePrism(self): self.Body = self.Doc.addObject("PartDesign::Body", "Body") self.Prism = self.Doc.addObject("PartDesign::AdditivePrism", "Prism") self.Prism.Polygon = 6 self.Prism.Circumradius = 4 self.Prism.Height = 10 self.Body.addObject(self.Prism) self.Doc.recompute() self.Prism001 = self.Doc.addObject("PartDesign::SubtractivePrism", "Prism001") self.Prism001.Polygon = 6 self.Prism001.Circumradius = 3 self.Prism001.Height = 10 self.Body.addObject(self.Prism001) self.Doc.recompute() self.assertAlmostEqual(self.Prism001.Shape.Volume, 3 * sqrt(3) / 2.0 * 10 * (4**2 - 3**2)) def testPrimitiveWedge(self): self.Body = self.Doc.addObject("PartDesign::Body", "Body") self.Wedge = self.Doc.addObject("PartDesign::AdditiveWedge", "Wedge") self.Wedge.X2min = 5 self.Wedge.X2max = 5 self.Wedge.Z2min = 0 self.Wedge.Z2max = 10 self.Body.addObject(self.Wedge) self.Doc.recompute() self.Wedge001 = self.Doc.addObject("PartDesign::SubtractiveWedge", "Wedge001") self.Wedge001.Xmin = 1 self.Wedge001.Xmax = 9 self.Wedge001.Ymax = 9 self.Wedge001.X2min = 5 self.Wedge001.X2max = 5 self.Wedge001.Z2min = 0 self.Wedge001.Z2max = 10 self.Body.addObject(self.Wedge001) self.Doc.recompute() self.assertAlmostEqual(self.Wedge001.Shape.Volume, 1 / 2.0 * (10 * 10 - 9 * 8) * 10) def tearDown(self): # closing doc FreeCAD.closeDocument("PartDesignTestPrimitive") # print ("omit closing document for debugging")