Draft: move the TestDraft unit tests to separate modules
This results in better organization of the tests, avoiding extremely big files.
This commit is contained in:
vocx-fc
317
src/Mod/Draft/drafttests/test_creation.py
Normal file
317
src/Mod/Draft/drafttests/test_creation.py
Normal file
@@ -0,0 +1,317 @@
|
||||
"""Unit test for the Draft module, object creation tests.
|
||||
"""
|
||||
# ***************************************************************************
|
||||
# * (c) 2013 Yorik van Havre <yorik@uncreated.net> *
|
||||
# * (c) 2019 Eliud Cabrera Castillo <e.cabrera-castillo@tum.de> *
|
||||
# * *
|
||||
# * 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))
|
||||
Draft.make_arc_3points = _fake_function
|
||||
obj = Draft.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(" Occasionaly 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(" Occasionaly 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)
|
||||
Reference in New Issue
Block a user