#*************************************************************************** #* 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")