kindred/create
1
1
Fork 0
Code Issues 30 Pull Requests Actions Packages Projects 9 Releases 2 Wiki Activity

Revise PartDesign Draft base test approach

Browse Source
This commit is contained in:
Kurt Kremitzki
2017-07-11 06:22:52 -05:00
parent 8f6b79e37c
commit 90bfae234d
1 changed files with 28 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
src/Mod/PartDesign/PartDesignTests/TestDraft.py
View File

@@ -22,11 +22,16 @@ import unittest
import FreeCAD
App = FreeCAD
class TestDraft(unittest.TestCase):
def setUp(self):
self.Doc = FreeCAD.newDocument("PartDesignTestDraft")
def testSimpleDraft(self):
# fix: create datum plane on YZ. create datum line on Z + 10i
# find top face by first making list comprehension of Z-normal faces
# and then find which has the higher center of mass Z-value
self.Body = self.Doc.addObject('PartDesign::Body','Body')
self.Box = self.Doc.addObject('PartDesign::AdditiveBox','Box')
self.Body.addObject(self.Box)
@@ -34,19 +39,34 @@ class TestDraft(unittest.TestCase):
self.Box.Width=10.00
self.Box.Height=10.00
self.Doc.recompute()
self.DatumPlane = self.Doc.addObject('PartDesign::Plane','DatumPlane')
self.DatumPlane.Support = [(self.Doc.YZ_Plane,'')]
self.DatumPlane.MapMode = 'FlatFace'
self.Body.addObject(self.DatumPlane)
self.Doc.recompute()
self.DatumLine = self.Doc.addObject('PartDesign::Line','DatumLine')
self.DatumLine.Support = [(self.Doc.X_Axis,'')]
self.DatumLine.MapMode = 'TwoPointLine'
self.Body.addObject(self.DatumLine)
self.Doc.recompute()
self.Draft = self.Doc.addObject("PartDesign::Draft","Draft")
self.Draft.Base = (self.Box, ["Face1"])
n1, n2 = self.Box.Shape.Faces[0].Surface.Axis, self.Box.Shape.Faces[1].Surface.Axis
if n1.dot(n2) == 0:
self.Draft.NeutralPlane = (self.Box, ["Face2"])
# Draft.Base needs to be top face
self.Faces = self.Box.Shape.Faces
# Grab the two faces with Z-normals and find the higher one
self.ZFaceIndexes = [i for i in range(len(self.Faces)) if self.Faces[i].Surface.Axis == App.Vector(0,0,1)]
if self.Faces[self.ZFaceIndexes[0]].CenterOfMass.z > self.Faces[self.ZFaceIndexes[1]].CenterOfMass.z:
self.TopFaceIndex = self.ZFaceIndexes[0]
else:
self.Draft.NeutralPlane = (self.Box, ["Face3"])
self.Draft.PullDirection = None
self.TopFaceIndex = self.ZFaceIndexes[1]
self.Draft.Base = (self.Box, ["Face"+str(self.TopFaceIndex+1)])
self.Draft.NeutralPlane = (self.DatumPlane, [''])
self.Draft.PullDirection = (self.DatumLine, [''])
self.Draft.Angle = 45.0
self.Draft.Reversed = 1
self.Body.addObject(self.Draft)
self.Doc.recompute()
if round(self.Draft.Shape.Volume, 7) - 500 == 0:
self.Draft.Reversed = 1
if 'Invalid' in self.Draft.State:
self.Draft.Reversed = 0
self.Doc.recompute()
self.assertAlmostEqual(self.Draft.Shape.Volume, 1500)