"""Unit test for the Draft module, object creation tests. """ # *************************************************************************** # * Copyright (c) 2013 Yorik van Havre * # * Copyright (c) 2019 Eliud Cabrera Castillo * # * * # * 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 as App import Draft from FreeCAD import Vector from .auxiliary import _msg from .auxiliary import _draw_header from .auxiliary import _no_gui from .auxiliary import _fake_function class DraftCreation(unittest.TestCase): """Test Draft creation functions.""" def setUp(self): """Set up a new document to hold the tests. This is executed before every test, so we create a document to hold the objects. """ _draw_header() self.doc_name = self.__class__.__name__ if App.ActiveDocument: if App.ActiveDocument.Name != self.doc_name: App.newDocument(self.doc_name) else: App.newDocument(self.doc_name) App.setActiveDocument(self.doc_name) self.doc = App.ActiveDocument _msg(" Temporary document '{}'".format(self.doc_name)) def test_line(self): """Create a line.""" operation = "Draft Line" _msg(" Test '{}'".format(operation)) a = Vector(0, 0, 0) b = Vector(2, 0, 0) _msg(" a={0}, b={1}".format(a, b)) obj = Draft.makeLine(a, b) self.assertTrue(obj, "'{}' failed".format(operation)) def test_polyline(self): """Create a polyline.""" operation = "Draft Wire" _msg(" Test '{}'".format(operation)) a = Vector(0, 0, 0) b = Vector(2, 0, 0) c = Vector(2, 2, 0) _msg(" a={0}, b={1}".format(a, b)) _msg(" c={}".format(c)) obj = Draft.makeWire([a, b, c]) self.assertTrue(obj, "'{}' failed".format(operation)) def test_fillet(self): """Create two lines, and a fillet between them.""" operation = "Draft Fillet" _msg(" Test '{}'".format(operation)) a = Vector(0, 0, 0) b = Vector(8, 0, 0) c = Vector(8, 8, 0) _msg(" Lines") _msg(" a={0}, b={1}".format(a, b)) _msg(" b={0}, c={1}".format(b, c)) L1 = Draft.makeLine(a, b) L2 = Draft.makeLine(b, c) App.ActiveDocument.recompute() if not App.GuiUp: _no_gui("DraftFillet") self.assertTrue(True) return import DraftFillet radius = 4 _msg(" Fillet") _msg(" radius={}".format(radius)) obj = DraftFillet.makeFillet([L1, L2], radius) self.assertTrue(obj, "'{}' failed".format(operation)) def test_circle(self): """Create a circle.""" operation = "Draft Circle" _msg(" Test '{}'".format(operation)) radius = 3 _msg(" radius={}".format(radius)) obj = Draft.makeCircle(radius) self.assertTrue(obj, "'{}' failed".format(operation)) def test_arc(self): """Create a circular arc.""" operation = "Draft Arc" _msg(" Test '{}'".format(operation)) radius = 2 start_angle = 0 end_angle = 90 _msg(" radius={}".format(radius)) _msg(" startangle={0}, endangle={1}".format(start_angle, end_angle)) obj = Draft.makeCircle(radius, start_angle, end_angle) self.assertTrue(obj, "'{}' failed".format(operation)) def test_arc_3points(self): """Create a circular arc from three points.""" operation = "Draft Arc 3Points" _msg(" Test '{}'".format(operation)) a = Vector(5, 0, 0) b = Vector(4, 3, 0) c = Vector(0, 5, 0) _msg(" a={0}, b={1}".format(a, b)) _msg(" c={}".format(c)) import draftobjects.arc_3points as arc3 obj = arc3.make_arc_3points([a, b, c]) self.assertTrue(obj, "'{}' failed".format(operation)) def test_ellipse(self): """Create an ellipse.""" operation = "Draft Ellipse" _msg(" Test '{}'".format(operation)) a = 5 b = 3 _msg(" major_axis={0}, minor_axis={1}".format(a, b)) obj = Draft.makeEllipse(a, b) self.assertTrue(obj, "'{}' failed".format(operation)) def test_polygon(self): """Create a regular polygon.""" operation = "Draft Polygon" _msg(" Test '{}'".format(operation)) n_faces = 6 radius = 5 _msg(" n_faces={0}, radius={1}".format(n_faces, radius)) obj = Draft.makePolygon(n_faces, radius) self.assertTrue(obj, "'{}' failed".format(operation)) def test_rectangle(self): """Create a rectangle.""" operation = "Draft Rectangle" _msg(" Test '{}'".format(operation)) length = 5 width = 2 _msg(" length={0}, width={1}".format(length, width)) obj = Draft.makeRectangle(length, width) self.assertTrue(obj, "'{}' failed".format(operation)) def test_text(self): """Create a text object.""" operation = "Draft Text" _msg(" Test '{}'".format(operation)) text = "Testing testing" _msg(" text='{}'".format(text)) obj = Draft.makeText(text) self.assertTrue(obj, "'{}' failed".format(operation)) def test_dimension_linear(self): """Create a linear dimension.""" operation = "Draft Dimension" _msg(" Test '{}'".format(operation)) _msg(" Occasionally crashes") a = Vector(0, 0, 0) b = Vector(9, 0, 0) c = Vector(4, -1, 0) _msg(" a={0}, b={1}".format(a, b)) _msg(" c={}".format(c)) obj = Draft.makeDimension(a, b, c) self.assertTrue(obj, "'{}' failed".format(operation)) def test_dimension_radial(self): """Create a radial dimension. NOT IMPLEMENTED CURRENTLY.""" operation = "Draft Dimension Radial" _msg(" Test '{}'".format(operation)) a = Vector(5, 0, 0) b = Vector(4, 3, 0) c = Vector(0, 5, 0) _msg(" a={0}, b={1}".format(a, b)) _msg(" c={}".format(c)) Draft.make_dimension_radial = _fake_function obj = Draft.make_dimension_radial(a, b, c) self.assertTrue(obj, "'{}' failed".format(operation)) def test_bspline(self): """Create a BSpline of three points.""" operation = "Draft BSpline" _msg(" Test '{}'".format(operation)) a = Vector(0, 0, 0) b = Vector(2, 0, 0) c = Vector(2, 2, 0) _msg(" a={0}, b={1}".format(a, b)) _msg(" c={}".format(c)) obj = Draft.makeBSpline([a, b, c]) self.assertTrue(obj, "'{}' failed".format(operation)) def test_point(self): """Create a point.""" operation = "Draft Point" _msg(" Test '{}'".format(operation)) p = Vector(5, 3, 2) _msg(" p.x={0}, p.y={1}, p.z={2}".format(p.x, p.y, p.z)) obj = Draft.makePoint(p.x, p.y, p.z) self.assertTrue(obj, "'{}' failed".format(operation)) def test_shapestring(self): """Create a ShapeString. NOT IMPLEMENTED CURRENTLY.""" operation = "Draft ShapeString" _msg(" Test '{}'".format(operation)) _msg(" In order to test this, a font file is needed.") text = "Testing Shapestring " font = None # TODO: get a font file here _msg(" text='{0}', font='{1}'".format(text, font)) Draft.makeShapeString = _fake_function obj = Draft.makeShapeString("Text", font) # Draft.makeShapeString("Text", FontFile="") self.assertTrue(obj, "'{}' failed".format(operation)) def test_facebinder(self): """Create a box, and then a facebinder from its faces.""" operation = "Draft Facebinder" _msg(" Test '{}'".format(operation)) _msg(" In order to test this, a selection is needed") _msg(" or an App::PropertyLinkSubList") _msg(" Box") box = App.ActiveDocument.addObject("Part::Box") App.ActiveDocument.recompute() # The facebinder function accepts a Gui selection set, # or a 'PropertyLinkSubList' # Gui selection set only works when the graphical interface is up # Gui.Selection.addSelection(box, ('Face1', 'Face6')) # selection_set = Gui.Selection.getSelectionEx() # elements = selection_set[0].SubElementNames # PropertyLinkSubList selection_set = [(box, ("Face1", "Face6"))] elements = selection_set[0][1] _msg(" object='{0}' ({1})".format(box.Shape.ShapeType, box.TypeId)) _msg(" sub-elements={}".format(elements)) obj = Draft.makeFacebinder(selection_set) self.assertTrue(obj, "'{}' failed".format(operation)) def test_cubicbezcurve(self): """Create a cubic bezier curve of four points.""" operation = "Draft CubBezCurve" _msg(" Test '{}'".format(operation)) a = Vector(0, 0, 0) b = Vector(2, 2, 0) c = Vector(5, 3, 0) d = Vector(9, 0, 0) _msg(" a={0}, b={1}".format(a, b)) _msg(" c={0}, d={1}".format(c, d)) obj = Draft.makeBezCurve([a, b, c, d], degree=3) self.assertTrue(obj, "'{}' failed".format(operation)) def test_bezcurve(self): """Create a bezier curve of six points, degree five.""" operation = "Draft BezCurve" _msg(" Test '{}'".format(operation)) a = Vector(0, 0, 0) b = Vector(2, 2, 0) c = Vector(5, 3, 0) d = Vector(9, 0, 0) e = Vector(12, 5, 0) f = Vector(12, 8, 0) _msg(" a={0}, b={1}".format(a, b)) _msg(" c={0}, d={1}".format(c, d)) _msg(" e={0}, f={1}".format(e, f)) obj = Draft.makeBezCurve([a, b, c, d, e, f]) self.assertTrue(obj, "'{}' failed".format(operation)) def test_label(self): """Create a label.""" operation = "Draft Label" _msg(" Test '{}'".format(operation)) _msg(" Occasionally crashes") target_point = Vector(0, 0, 0) distance = -25 placement = App.Placement(Vector(50, 50, 0), App.Rotation()) _msg(" target_point={0}, " "distance={1}".format(target_point, distance)) _msg(" placement={}".format(placement)) obj = Draft.makeLabel(targetpoint=target_point, distance=distance, placement=placement) App.ActiveDocument.recompute() self.assertTrue(obj, "'{}' failed".format(operation)) def tearDown(self): """Finish the test. This is executed after each test, so we close the document. """ App.closeDocument(self.doc_name)