# (c) Juergen Riegel (FreeCAD@juergen-riegel.net) 2011 LGPL * # * # This file is part of the FreeCAD CAx development system. * # * # 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. * # * # FreeCAD 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 FreeCAD; 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.Support = (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.Support = (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.Support = (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")