143 lines
6.4 KiB
Python
143 lines
6.4 KiB
Python
# Unit test for the Draft module
|
|
|
|
#***************************************************************************
|
|
#* (c) Yorik van Havre <yorik@uncreated.net> 2013 *
|
|
#* *
|
|
#* 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 FreeCAD, os, unittest, FreeCADGui, Draft
|
|
|
|
class DraftTest(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
# setting a new document to hold the tests
|
|
if FreeCAD.ActiveDocument:
|
|
if FreeCAD.ActiveDocument.Name != "DraftTest":
|
|
FreeCAD.newDocument("DraftTest")
|
|
else:
|
|
FreeCAD.newDocument("DraftTest")
|
|
FreeCAD.setActiveDocument("DraftTest")
|
|
|
|
def testPivy(self):
|
|
FreeCAD.Console.PrintLog ('Checking Pivy...\n')
|
|
from pivy import coin
|
|
c = coin.SoCube()
|
|
FreeCADGui.ActiveDocument.ActiveView.getSceneGraph().addChild(c)
|
|
self.failUnless(c,"Pivy is not working properly")
|
|
|
|
# creation tools
|
|
|
|
def testLine(self):
|
|
FreeCAD.Console.PrintLog ('Checking Draft Line...\n')
|
|
Draft.makeLine(FreeCAD.Vector(0,0,0),FreeCAD.Vector(-2,0,0))
|
|
self.failUnless(FreeCAD.ActiveDocument.getObject("Line"),"Draft Line failed")
|
|
|
|
def testWire(self):
|
|
FreeCAD.Console.PrintLog ('Checking Draft Wire...\n')
|
|
Draft.makeWire([FreeCAD.Vector(0,0,0),FreeCAD.Vector(2,0,0),FreeCAD.Vector(2,2,0)])
|
|
self.failUnless(FreeCAD.ActiveDocument.getObject("DWire"),"Draft Wire failed")
|
|
|
|
def testBSpline(self):
|
|
FreeCAD.Console.PrintLog ('Checking Draft BSpline...\n')
|
|
Draft.makeBSpline([FreeCAD.Vector(0,0,0),FreeCAD.Vector(2,0,0),FreeCAD.Vector(2,2,0)])
|
|
self.failUnless(FreeCAD.ActiveDocument.getObject("BSpline"),"Draft BSpline failed")
|
|
|
|
def testRectangle(self):
|
|
FreeCAD.Console.PrintLog ('Checking Draft Rectangle...\n')
|
|
Draft.makeRectangle(4,2)
|
|
self.failUnless(FreeCAD.ActiveDocument.getObject("Rectangle"),"Draft Rectangle failed")
|
|
|
|
def testArc(self):
|
|
FreeCAD.Console.PrintLog ('Checking Draft Arc...\n')
|
|
Draft.makeCircle(2, startangle=0, endangle=90)
|
|
self.failUnless(FreeCAD.ActiveDocument.getObject("Arc"),"Draft Arc failed")
|
|
|
|
def testCircle(self):
|
|
FreeCAD.Console.PrintLog ('Checking Draft Circle...\n')
|
|
Draft.makeCircle(3)
|
|
self.failUnless(FreeCAD.ActiveDocument.getObject("Circle"),"Draft Circle failed")
|
|
|
|
def testPolygon(self):
|
|
FreeCAD.Console.PrintLog ('Checking Draft Polygon...\n')
|
|
Draft.makePolygon(5,5)
|
|
self.failUnless(FreeCAD.ActiveDocument.getObject("Polygon"),"Draft Polygon failed")
|
|
|
|
def testEllipse(self):
|
|
FreeCAD.Console.PrintLog ('Checking Draft Ellipse...\n')
|
|
Draft.makeEllipse(5,3)
|
|
self.failUnless(FreeCAD.ActiveDocument.getObject("Ellipse"),"Draft Ellipse failed")
|
|
|
|
def testPoint(self):
|
|
FreeCAD.Console.PrintLog ('Checking Draft Point...\n')
|
|
Draft.makePoint(5,3,2)
|
|
self.failUnless(FreeCAD.ActiveDocument.getObject("Point"),"Draft Point failed")
|
|
|
|
def testText(self):
|
|
FreeCAD.Console.PrintLog ('Checking Draft Text...\n')
|
|
Draft.makeText("Testing Draft")
|
|
self.failUnless(FreeCAD.ActiveDocument.getObject("Text"),"Draft Text failed")
|
|
|
|
#def testShapeString(self):
|
|
# not working ATM because it needs a font file
|
|
# FreeCAD.Console.PrintLog ('Checking Draft ShapeString...\n')
|
|
# Draft.makeShapeString("Testing Draft")
|
|
# self.failUnless(FreeCAD.ActiveDocument.getObject("ShapeString"),"Draft ShapeString failed")
|
|
|
|
def testDimension(self):
|
|
FreeCAD.Console.PrintLog ('Checking Draft Dimension...\n')
|
|
Draft.makeDimension(FreeCAD.Vector(0,0,0),FreeCAD.Vector(2,0,0),FreeCAD.Vector(1,-1,0))
|
|
self.failUnless(FreeCAD.ActiveDocument.getObject("Dimension"),"Draft Dimension failed")
|
|
|
|
# modification tools
|
|
|
|
def testMove(self):
|
|
FreeCAD.Console.PrintLog ('Checking Draft Move...\n')
|
|
l = Draft.makeLine(FreeCAD.Vector(0,0,0),FreeCAD.Vector(-2,0,0))
|
|
Draft.move(l,FreeCAD.Vector(2,0,0))
|
|
self.failUnless(l.Start == FreeCAD.Vector(2,0,0),"Draft Move failed")
|
|
|
|
def testCopy(self):
|
|
FreeCAD.Console.PrintLog ('Checking Draft Move with copy...\n')
|
|
l = Draft.makeLine(FreeCAD.Vector(0,0,0),FreeCAD.Vector(2,0,0))
|
|
l2 = Draft.move(l,FreeCAD.Vector(2,0,0),copy=True)
|
|
self.failUnless(l2,"Draft Move with copy failed")
|
|
|
|
def testRotate(self):
|
|
FreeCAD.Console.PrintLog ('Checking Draft Rotate...\n')
|
|
l = Draft.makeLine(FreeCAD.Vector(2,0,0),FreeCAD.Vector(4,0,0))
|
|
Draft.rotate(l,90)
|
|
self.failUnless(l.Start == FreeCAD.Vector(0,2,0),"Draft Rotate failed")
|
|
|
|
def testOffset(self):
|
|
FreeCAD.Console.PrintLog ('Checking Draft Offset...\n')
|
|
r = Draft.makeRectangle(4,2)
|
|
r2 = Draft.offset(r,FreeCAD.Vector(-1,-1,0),copy=True)
|
|
self.failUnless(r2,"Draft Offset failed")
|
|
|
|
# modification tools
|
|
|
|
def tearDown(self):
|
|
FreeCAD.closeDocument("DraftTest")
|
|
pass
|
|
|
|
|
|
|