freecad-cam/Mod/PartDesign/PartDesignTests/TestSketch.py
2026-02-01 01:59:24 +01:00

102 lines
4.0 KiB
Python

#***************************************************************************
#* Copyright (c) 2024 Werner Mayer <wmayer[at]users.sourceforge.net> *
#* *
#* This file is part of FreeCAD. *
#* *
#* FreeCAD is free software: you can redistribute it and/or modify it *
#* under the terms of the GNU Lesser General Public License as *
#* published by the Free Software Foundation, either version 2.1 of the *
#* License, or (at your option) any later version. *
#* *
#* 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 *
#* Lesser General Public License for more details. *
#* *
#* You should have received a copy of the GNU Lesser General Public *
#* License along with FreeCAD. If not, see *
#* <https://www.gnu.org/licenses/>. *
#* *
#***************************************************************************
import unittest
import FreeCAD
""" Test transaction interaction """
class TestSketch(unittest.TestCase):
def setUp(self):
self.doc = FreeCAD.newDocument("PartDesignTestSketch")
self.doc.UndoMode = True
def testIssue17553(self):
self.doc.openTransaction("Create box")
box = self.doc.addObject("Part::Box", "Box")
self.doc.commitTransaction()
self.doc.recompute()
self.doc.openTransaction("Create sketch")
body = self.doc.addObject('PartDesign::Body', 'Body')
plane = self.doc.getObject('XY_Plane')
self.doc.commitTransaction()
self.doc.openTransaction("Rename object")
box.Label = "Object"
self.doc.commitTransaction()
sketch = body.newObject('Sketcher::SketchObject', 'Sketch')
sketch.AttachmentSupport = (plane, [''])
sketch.MapMode = 'FlatFace'
self.doc.recompute()
self.assertEqual(sketch.InList, [body])
self.assertEqual(sketch.OutList, [plane])
sketch.AttachmentSupport == [(plane, ("",))]
self.doc.undo() # undo renaming
self.doc.undo() # undo body creation
self.doc.undo() # undo box creation
self.doc.openTransaction("Remove sketch")
self.doc.removeObject(sketch.Name)
self.doc.commitTransaction()
self.doc.undo() # undo removal
self.assertEqual(sketch.InList, [])
self.assertEqual(sketch.OutList, [])
self.assertEqual(sketch.AttachmentSupport, [])
def testDependency(self):
self.doc.openTransaction("Create box")
box = self.doc.addObject("Part::Box", "Box")
self.doc.commitTransaction()
self.doc.recompute()
self.doc.openTransaction("Create sketch")
body = self.doc.addObject('PartDesign::Body', 'Body')
plane = self.doc.getObject('XY_Plane')
self.doc.commitTransaction()
self.doc.openTransaction("Rename object")
box.Label = "Object"
self.doc.commitTransaction()
sketch = body.newObject('Sketcher::SketchObject', 'Sketch')
sketch.AttachmentSupport = (plane, [''])
sketch.MapMode = 'FlatFace'
self.doc.recompute()
sketch.OutList
sketch.AttachmentSupport
self.doc.undo() # undo renaming
self.doc.undo() # undo body creation
self.doc.undo() # undo box creation
self.doc.DependencyGraph
def tearDown(self):
FreeCAD.closeDocument("PartDesignTestSketch")