# *************************************************************************** # * 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 * # * * # *************************************************************************** import unittest import FreeCAD import TestSketcherApp class TestPocket(unittest.TestCase): def setUp(self): self.Doc = FreeCAD.newDocument("PartDesignTestPocket") def testPocketDimensionCase(self): self.Body = self.Doc.addObject("PartDesign::Body", "Body") self.PadSketch = self.Doc.addObject("Sketcher::SketchObject", "PadSketch") self.Body.addObject(self.PadSketch) TestSketcherApp.CreateRectangleSketch(self.PadSketch, (0, 0), (10, 10)) self.Doc.recompute() self.Pad = self.Doc.addObject("PartDesign::Pad", "Pad") self.Body.addObject(self.Pad) self.Pad.Profile = self.PadSketch self.Pad.Length = 1 self.Pad.Reversed = 1 self.Doc.recompute() self.PocketSketch = self.Doc.addObject("Sketcher::SketchObject", "PocketSketch") self.Body.addObject(self.PocketSketch) TestSketcherApp.CreateRectangleSketch(self.PocketSketch, (2.5, 2.5), (5, 5)) self.Doc.recompute() self.Pocket = self.Doc.addObject("PartDesign::Pocket", "Pocket") self.Body.addObject(self.Pocket) self.Pocket.Profile = self.PocketSketch self.Pocket.Length = 1 self.Doc.recompute() self.assertAlmostEqual(self.Pocket.Shape.Volume, 75.0) def testPocketThroughAllCase(self): self.Body = self.Doc.addObject("PartDesign::Body", "Body") self.PadSketch = self.Doc.addObject("Sketcher::SketchObject", "PadSketch") self.Body.addObject(self.PadSketch) TestSketcherApp.CreateRectangleSketch(self.PadSketch, (0, 0), (10, 10)) self.Doc.recompute() self.Pad = self.Doc.addObject("PartDesign::Pad", "Pad") self.Body.addObject(self.Pad) self.Pad.Profile = self.PadSketch self.Pad.Length = 1 self.Pad.Reversed = 1 self.Doc.recompute() self.PocketSketch = self.Doc.addObject("Sketcher::SketchObject", "PocketSketch") self.Body.addObject(self.PocketSketch) TestSketcherApp.CreateRectangleSketch(self.PocketSketch, (2.5, 2.5), (5, 5)) self.Doc.recompute() self.Pocket = self.Doc.addObject("PartDesign::Pocket", "Pocket") self.Body.addObject(self.Pocket) self.Pocket.Profile = self.PocketSketch self.Pocket.Length = 1 self.Doc.recompute() self.PocketSketch1 = self.Doc.addObject("Sketcher::SketchObject", "PocketSketch") self.Body.addObject(self.PocketSketch1) self.PocketSketch1.MapMode = "FlatFace" self.PocketSketch1.AttachmentSupport = (self.Doc.XZ_Plane, [""]) self.Doc.recompute() TestSketcherApp.CreateRectangleSketch(self.PocketSketch1, (2.5, -0.75), (5, 0.50)) self.Doc.recompute() self.Pocket001 = self.Doc.addObject("PartDesign::Pocket", "Pocket001") self.Body.addObject(self.Pocket001) self.Pocket001.Profile = self.PocketSketch1 self.Pocket001.Type = 1 self.Doc.recompute() self.assertAlmostEqual(self.Pocket001.Shape.Volume, 62.5) def testPocketToFirstCase(self): self.Body = self.Doc.addObject("PartDesign::Body", "Body") self.PadSketch = self.Doc.addObject("Sketcher::SketchObject", "PadSketch") self.Body.addObject(self.PadSketch) TestSketcherApp.CreateRectangleSketch(self.PadSketch, (0, 0), (10, 10)) self.Doc.recompute() self.Pad = self.Doc.addObject("PartDesign::Pad", "Pad") self.Body.addObject(self.Pad) self.Pad.Profile = self.PadSketch self.Pad.Length = 1 self.Pad.Reversed = 1 self.Doc.recompute() self.PocketSketch = self.Doc.addObject("Sketcher::SketchObject", "PocketSketch") self.Body.addObject(self.PocketSketch) TestSketcherApp.CreateRectangleSketch(self.PocketSketch, (2.5, 2.5), (5, 5)) self.Doc.recompute() self.Pocket = self.Doc.addObject("PartDesign::Pocket", "Pocket") self.Body.addObject(self.Pocket) self.Pocket.Profile = self.PocketSketch self.Pocket.Length = 1 self.Doc.recompute() self.PocketSketch1 = self.Doc.addObject("Sketcher::SketchObject", "PocketSketch") self.Body.addObject(self.PocketSketch1) self.PocketSketch1.MapMode = "FlatFace" self.PocketSketch1.AttachmentSupport = (self.Doc.XZ_Plane, [""]) self.Doc.recompute() TestSketcherApp.CreateRectangleSketch(self.PocketSketch1, (2.5, -1), (5, 1)) self.Doc.recompute() self.Pocket001 = self.Doc.addObject("PartDesign::Pocket", "Pocket001") self.Body.addObject(self.Pocket001) self.Pocket001.Profile = self.PocketSketch1 self.Pocket001.Type = 2 self.Doc.recompute() self.assertAlmostEqual(self.Pocket001.Shape.Volume, 62.5) def testPocketToFaceCase(self): self.Body = self.Doc.addObject("PartDesign::Body", "Body") self.PadSketch = self.Doc.addObject("Sketcher::SketchObject", "PadSketch") self.Body.addObject(self.PadSketch) TestSketcherApp.CreateRectangleSketch(self.PadSketch, (0, 0), (10, 10)) self.Doc.recompute() self.Pad = self.Doc.addObject("PartDesign::Pad", "Pad") self.Body.addObject(self.Pad) self.Pad.Profile = self.PadSketch self.Pad.Length = 1 self.Pad.Reversed = 1 self.Doc.recompute() self.PocketSketch = self.Doc.addObject("Sketcher::SketchObject", "PocketSketch") self.Body.addObject(self.PocketSketch) TestSketcherApp.CreateRectangleSketch(self.PocketSketch, (2.5, 2.5), (5, 5)) self.Doc.recompute() self.Pocket = self.Doc.addObject("PartDesign::Pocket", "Pocket") self.Body.addObject(self.Pocket) self.Pocket.Profile = self.PocketSketch self.Pocket.Length = 1 self.Doc.recompute() self.PocketSketch1 = self.Doc.addObject("Sketcher::SketchObject", "PocketSketch") self.Body.addObject(self.PocketSketch1) self.PocketSketch1.MapMode = "FlatFace" self.PocketSketch1.AttachmentSupport = (self.Doc.XZ_Plane, [""]) self.Doc.recompute() TestSketcherApp.CreateRectangleSketch(self.PocketSketch1, (0, -1), (10, 1)) self.Doc.recompute() self.Pocket001 = self.Doc.addObject("PartDesign::Pocket", "Pocket001") self.Body.addObject(self.Pocket001) self.Pocket001.Profile = self.PocketSketch1 self.Pocket001.Type = 3 # Handle face-naming inconsistency in OCC < 7 self.FaceNumber = 7 self.Pocket001.UpToFace = (self.Pocket, ["Face" + str(self.FaceNumber)]) self.Doc.recompute() while ( "Invalid" in self.Pocket001.State or round(self.Pocket001.Shape.Volume, 7) != 50.0 ) and self.FaceNumber < 11: self.FaceNumber += 1 self.Pocket001.UpToFace = (self.Pocket, ["Face" + str(self.FaceNumber)]) self.Doc.recompute() self.assertAlmostEqual(self.Pocket001.Shape.Volume, 50.0) def tearDown(self): # closing doc FreeCAD.closeDocument("PartDesignTestPocket") # print ("omit closing document for debugging")