Test: apply clang format

This commit is contained in:
wmayer
2023-09-03 18:14:40 +02:00
committed by Chris Hennes
parent 19a1204a5a
commit e48964fad5
23 changed files with 3065 additions and 2745 deletions

View File

@@ -1,29 +1,30 @@
#***************************************************************************
#* Copyright (c) 2004 Juergen Riegel <juergen.riegel@web.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 *
#* *
#***************************************************************************/
# ***************************************************************************
# * Copyright (c) 2004 Juergen Riegel <juergen.riegel@web.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 FreeCAD, os, unittest, tempfile, math
from FreeCAD import Base
class ConsoleTestCase(unittest.TestCase):
def setUp(self):
self.count = 0
@@ -40,20 +41,21 @@ class ConsoleTestCase(unittest.TestCase):
import _thread as thread, time
except Exception:
import thread, time
def adder():
lock.acquire()
self.count=self.count+1
self.count = self.count + 1
# call of Console method is thread-safe
FreeCAD.Console.PrintMessage("Call from Python thread: count="+str(self.count)+"\n")
FreeCAD.Console.PrintMessage("Call from Python thread: count=" + str(self.count) + "\n")
lock.release()
lock=thread.allocate_lock()
lock = thread.allocate_lock()
for i in range(10):
thread.start_new(adder,())
thread.start_new(adder, ())
time.sleep(3)
self.assertEqual(self.count, 10, "Synchronization of threads failed")
FreeCAD.Console.PrintMessage(str(self.count)+"\n")
FreeCAD.Console.PrintMessage(str(self.count) + "\n")
def testAsynchronPrintFromThread(self):
# http://python-kurs.eu/threads.php
@@ -61,43 +63,46 @@ class ConsoleTestCase(unittest.TestCase):
import _thread as thread, time
except Exception:
import thread, time
def adder():
self.count=self.count+1
# call of Console method is thread-safe
FreeCAD.Console.PrintMessage("Call from Python thread (not synchronized): count="+str(self.count)+"\n")
lock=thread.allocate_lock()
def adder():
self.count = self.count + 1
# call of Console method is thread-safe
FreeCAD.Console.PrintMessage(
"Call from Python thread (not synchronized): count=" + str(self.count) + "\n"
)
lock = thread.allocate_lock()
for i in range(10):
thread.start_new(adder,())
thread.start_new(adder, ())
time.sleep(3)
FreeCAD.Console.PrintMessage(str(self.count)+"\n")
FreeCAD.Console.PrintMessage(str(self.count) + "\n")
# def testStatus(self):
# SLog = FreeCAD.GetStatus("Console","Log")
# SErr = FreeCAD.GetStatus("Console","Err")
# SWrn = FreeCAD.GetStatus("Console","Wrn")
# SMsg = FreeCAD.GetStatus("Console","Msg")
# FreeCAD.SetStatus("Console","Log",1)
# FreeCAD.SetStatus("Console","Err",1)
# FreeCAD.SetStatus("Console","Wrn",1)
# FreeCAD.SetStatus("Console","Msg",1)
# self.assertEqual(FreeCAD.GetStatus("Console","Msg"),1,"Set and read status failed (Console,Msg)")
# self.assertEqual(FreeCAD.GetStatus("Console","Err"),1,"Set and read status failed (Console,Err)")
# self.assertEqual(FreeCAD.GetStatus("Console","Wrn"),1,"Set and read status failed (Console,Wrn)")
# self.assertEqual(FreeCAD.GetStatus("Console","Log"),1,"Set and read status failed (Console,Log)")
# FreeCAD.SetStatus("Console","Log",0)
# FreeCAD.SetStatus("Console","Err",0)
# FreeCAD.SetStatus("Console","Wrn",0)
# FreeCAD.SetStatus("Console","Msg",0)
# self.assertEqual(FreeCAD.GetStatus("Console","Msg"),0,"Set and read status failed (Console,Msg)")
# self.assertEqual(FreeCAD.GetStatus("Console","Err"),0,"Set and read status failed (Console,Err)")
# self.assertEqual(FreeCAD.GetStatus("Console","Wrn"),0,"Set and read status failed (Console,Wrn)")
# self.assertEqual(FreeCAD.GetStatus("Console","Log"),0,"Set and read status failed (Console,Log)")
# FreeCAD.SetStatus("Console","Log",SLog)
# FreeCAD.SetStatus("Console","Err",SErr)
# FreeCAD.SetStatus("Console","Wrn",SWrn)
# FreeCAD.SetStatus("Console","Msg",SMsg)
# def testStatus(self):
# SLog = FreeCAD.GetStatus("Console","Log")
# SErr = FreeCAD.GetStatus("Console","Err")
# SWrn = FreeCAD.GetStatus("Console","Wrn")
# SMsg = FreeCAD.GetStatus("Console","Msg")
# FreeCAD.SetStatus("Console","Log",1)
# FreeCAD.SetStatus("Console","Err",1)
# FreeCAD.SetStatus("Console","Wrn",1)
# FreeCAD.SetStatus("Console","Msg",1)
# self.assertEqual(FreeCAD.GetStatus("Console","Msg"),1,"Set and read status failed (Console,Msg)")
# self.assertEqual(FreeCAD.GetStatus("Console","Err"),1,"Set and read status failed (Console,Err)")
# self.assertEqual(FreeCAD.GetStatus("Console","Wrn"),1,"Set and read status failed (Console,Wrn)")
# self.assertEqual(FreeCAD.GetStatus("Console","Log"),1,"Set and read status failed (Console,Log)")
# FreeCAD.SetStatus("Console","Log",0)
# FreeCAD.SetStatus("Console","Err",0)
# FreeCAD.SetStatus("Console","Wrn",0)
# FreeCAD.SetStatus("Console","Msg",0)
# self.assertEqual(FreeCAD.GetStatus("Console","Msg"),0,"Set and read status failed (Console,Msg)")
# self.assertEqual(FreeCAD.GetStatus("Console","Err"),0,"Set and read status failed (Console,Err)")
# self.assertEqual(FreeCAD.GetStatus("Console","Wrn"),0,"Set and read status failed (Console,Wrn)")
# self.assertEqual(FreeCAD.GetStatus("Console","Log"),0,"Set and read status failed (Console,Log)")
# FreeCAD.SetStatus("Console","Log",SLog)
# FreeCAD.SetStatus("Console","Err",SErr)
# FreeCAD.SetStatus("Console","Wrn",SWrn)
# FreeCAD.SetStatus("Console","Msg",SMsg)
def tearDown(self):
pass
@@ -105,20 +110,22 @@ class ConsoleTestCase(unittest.TestCase):
def testILoggerBlocker(self):
if FreeCAD.GuiUp:
import QtUnitGui
QtUnitGui.testILoggerBlocker()
class ParameterTestCase(unittest.TestCase):
def setUp(self):
self.TestPar = FreeCAD.ParamGet("System parameter:Test")
def testGroup(self):
#FreeCAD.Console.PrintLog("Base::ParameterTestCase::testGroup\n")
# FreeCAD.Console.PrintLog("Base::ParameterTestCase::testGroup\n")
# check on Group creation
Temp = self.TestPar.GetGroup("44")
self.assertTrue(self.TestPar.HasGroup("44"),"Test on created group failed")
self.assertTrue(self.TestPar.HasGroup("44"), "Test on created group failed")
# check on Deletion
self.TestPar.RemGroup("44")
self.assertTrue(self.TestPar.HasGroup("44"),"A referenced group must not be deleted")
self.assertTrue(self.TestPar.HasGroup("44"), "A referenced group must not be deleted")
Temp = 0
def testGroupNames(self):
@@ -132,205 +139,203 @@ class ParameterTestCase(unittest.TestCase):
# check on special conditions
def testInt(self):
#FreeCAD.Console.PrintLog("Base::ParameterTestCase::testInt\n")
#Temp = FreeCAD.ParamGet("System parameter:Test/44")
# FreeCAD.Console.PrintLog("Base::ParameterTestCase::testInt\n")
# Temp = FreeCAD.ParamGet("System parameter:Test/44")
# check on Int
self.TestPar.SetInt("44",4711)
self.assertEqual(self.TestPar.GetInt("44"), 4711,"In and out error at Int")
self.TestPar.SetInt("44", 4711)
self.assertEqual(self.TestPar.GetInt("44"), 4711, "In and out error at Int")
# check on Deletion
self.TestPar.RemInt("44")
self.assertEqual(self.TestPar.GetInt("44",1), 1,"Deletion error at Int")
self.assertEqual(self.TestPar.GetInt("44", 1), 1, "Deletion error at Int")
def testBool(self):
#FreeCAD.Console.PrintLog("Base::ParameterTestCase::testBool\n")
# FreeCAD.Console.PrintLog("Base::ParameterTestCase::testBool\n")
# check on Int
self.TestPar.SetBool("44",1)
self.assertEqual(self.TestPar.GetBool("44"), 1,"In and out error at Bool")
self.TestPar.SetBool("44", 1)
self.assertEqual(self.TestPar.GetBool("44"), 1, "In and out error at Bool")
# check on Deletion
self.TestPar.RemBool("44")
self.assertEqual(self.TestPar.GetBool("44",0), 0,"Deletion error at Bool")
self.assertEqual(self.TestPar.GetBool("44", 0), 0, "Deletion error at Bool")
def testFloat(self):
#FreeCAD.Console.PrintLog("Base::ParameterTestCase::testFloat\n")
#Temp = FreeCAD.ParamGet("System parameter:Test/44")
# FreeCAD.Console.PrintLog("Base::ParameterTestCase::testFloat\n")
# Temp = FreeCAD.ParamGet("System parameter:Test/44")
# check on Int
self.TestPar.SetFloat("44",4711.4711)
self.assertEqual(self.TestPar.GetFloat("44"), 4711.4711,"In and out error at Float")
self.TestPar.SetFloat("44", 4711.4711)
self.assertEqual(self.TestPar.GetFloat("44"), 4711.4711, "In and out error at Float")
# check on Deletion
self.TestPar.RemFloat("44")
self.assertEqual(self.TestPar.GetFloat("44",1.1), 1.1,"Deletion error at Float")
self.assertEqual(self.TestPar.GetFloat("44", 1.1), 1.1, "Deletion error at Float")
def testString(self):
#FreeCAD.Console.PrintLog("Base::ParameterTestCase::testFloat\n")
#Temp = FreeCAD.ParamGet("System parameter:Test/44")
# FreeCAD.Console.PrintLog("Base::ParameterTestCase::testFloat\n")
# Temp = FreeCAD.ParamGet("System parameter:Test/44")
# check on Int
self.TestPar.SetString("44","abcdefgh")
self.assertEqual(self.TestPar.GetString("44"), "abcdefgh","In and out error at String")
self.TestPar.SetString("44", "abcdefgh")
self.assertEqual(self.TestPar.GetString("44"), "abcdefgh", "In and out error at String")
# check on Deletion
self.TestPar.RemString("44")
self.assertEqual(self.TestPar.GetString("44","hallo"), "hallo","Deletion error at String")
self.assertEqual(self.TestPar.GetString("44", "hallo"), "hallo", "Deletion error at String")
def testNesting(self):
# Parameter testing
#FreeCAD.Console.PrintLog("Base::ParameterTestCase::testNesting\n")
# FreeCAD.Console.PrintLog("Base::ParameterTestCase::testNesting\n")
for i in range(50):
self.TestPar.SetFloat(str(i),4711.4711)
self.TestPar.SetInt(str(i),4711)
self.TestPar.SetBool(str(i),1)
self.TestPar.SetFloat(str(i), 4711.4711)
self.TestPar.SetInt(str(i), 4711)
self.TestPar.SetBool(str(i), 1)
Temp = self.TestPar.GetGroup(str(i))
for l in range(50):
Temp.SetFloat(str(l),4711.4711)
Temp.SetInt(str(l),4711)
Temp.SetBool(str(l),1)
Temp.SetFloat(str(l), 4711.4711)
Temp.SetInt(str(l), 4711)
Temp.SetBool(str(l), 1)
Temp = 0
def testExportImport(self):
# Parameter testing
#FreeCAD.Console.PrintLog("Base::ParameterTestCase::testNesting\n")
self.TestPar.SetFloat("ExTest",4711.4711)
self.TestPar.SetInt("ExTest",4711)
self.TestPar.SetString("ExTest","4711")
self.TestPar.SetBool("ExTest",1)
# FreeCAD.Console.PrintLog("Base::ParameterTestCase::testNesting\n")
self.TestPar.SetFloat("ExTest", 4711.4711)
self.TestPar.SetInt("ExTest", 4711)
self.TestPar.SetString("ExTest", "4711")
self.TestPar.SetBool("ExTest", 1)
Temp = self.TestPar.GetGroup("ExTest")
Temp.SetFloat("ExTest",4711.4711)
Temp.SetInt("ExTest",4711)
Temp.SetString("ExTest","4711")
Temp.SetBool("ExTest",1)
Temp.SetFloat("ExTest", 4711.4711)
Temp.SetInt("ExTest", 4711)
Temp.SetString("ExTest", "4711")
Temp.SetBool("ExTest", 1)
TempPath = tempfile.gettempdir() + os.sep + "ExportTest.FCExport"
self.TestPar.Export(TempPath)
Temp = self.TestPar.GetGroup("ImportTest")
Temp.Import(TempPath)
self.assertEqual(Temp.GetFloat("ExTest"), 4711.4711,"ExportImport error")
self.assertEqual(Temp.GetFloat("ExTest"), 4711.4711, "ExportImport error")
Temp = 0
def tearDown(self):
#remove all
# remove all
TestPar = FreeCAD.ParamGet("System parameter:Test")
TestPar.Clear()
class AlgebraTestCase(unittest.TestCase):
def setUp(self):
pass
def testAngle(self):
v1 = FreeCAD.Vector(0,0,0.000001)
v2 = FreeCAD.Vector(0,0.000001,0)
self.assertAlmostEqual(v1.getAngle(v2), math.pi/2)
self.assertAlmostEqual(v2.getAngle(v1), math.pi/2)
v1 = FreeCAD.Vector(0, 0, 0.000001)
v2 = FreeCAD.Vector(0, 0.000001, 0)
self.assertAlmostEqual(v1.getAngle(v2), math.pi / 2)
self.assertAlmostEqual(v2.getAngle(v1), math.pi / 2)
def testVector2d(self):
v = FreeCAD.Base.Vector2d(1.0, 1.0)
v.rotate(math.pi/2)
v.rotate(math.pi / 2)
self.assertAlmostEqual(v.x, -1.0)
self.assertAlmostEqual(v.y, 1.0)
def testAngleWithNullVector(self):
v1 = FreeCAD.Vector(0,0,0)
v2 = FreeCAD.Vector(0,1,0)
v1 = FreeCAD.Vector(0, 0, 0)
v2 = FreeCAD.Vector(0, 1, 0)
self.assertTrue(math.isnan(v1.getAngle(v2)))
self.assertTrue(math.isnan(v2.getAngle(v1)))
def testMatrix(self):
m=FreeCAD.Matrix(4,2,1,0,1,1,1,0,0,0,1,0,0,0,0,1)
u=m.multiply(m.inverse())
self.assertEqual(u, FreeCAD.Matrix(),"Invalid inverse of matrix")
m = FreeCAD.Matrix(4, 2, 1, 0, 1, 1, 1, 0, 0, 0, 1, 0, 0, 0, 0, 1)
u = m.multiply(m.inverse())
self.assertEqual(u, FreeCAD.Matrix(), "Invalid inverse of matrix")
def testRotAndMoveMatrix(self):
m1=FreeCAD.Matrix()
m1.move(10,5,-3)
m1.rotateY(.2)
m2=FreeCAD.Matrix()
m2.rotateY(.2)
m2.move(10,5,-3)
m3=FreeCAD.Matrix()
m3.move(10,5,-3)
m4=FreeCAD.Matrix()
m4.rotateY(.2)
self.assertNotEqual(m1, m3*m4, "Wrong multiplication order")
self.assertEqual(m1, m4*m3 , "Wrong multiplication order")
self.assertEqual(m2, m3*m4 , "Wrong multiplication order")
self.assertNotEqual(m2, m4*m3, "Wrong multiplication order")
m1 = FreeCAD.Matrix()
m1.move(10, 5, -3)
m1.rotateY(0.2)
m2 = FreeCAD.Matrix()
m2.rotateY(0.2)
m2.move(10, 5, -3)
m3 = FreeCAD.Matrix()
m3.move(10, 5, -3)
m4 = FreeCAD.Matrix()
m4.rotateY(0.2)
self.assertNotEqual(m1, m3 * m4, "Wrong multiplication order")
self.assertEqual(m1, m4 * m3, "Wrong multiplication order")
self.assertEqual(m2, m3 * m4, "Wrong multiplication order")
self.assertNotEqual(m2, m4 * m3, "Wrong multiplication order")
def testRotationFromMatrix(self):
m1 = FreeCAD.Matrix()
m1.rotateZ(.2)
m1.rotateZ(0.2)
m1.scale(0.5)
m1.rotateY(.2)
m1.rotateY(0.2)
m1.scale(3)
m1.move(10,5,-3)
m1.move(10, 5, -3)
r1 = FreeCAD.Rotation(m1)
m2 = FreeCAD.Matrix()
m2.rotateZ(.2)
m2.rotateY(.2)
m2.move(10,5,-3)
r2 = FreeCAD.Rotation(m2)
self.assertTrue(r1.isSame(r2, 1e-7), 'Scale on matrix influenced rotation')
m2.rotateZ(0.2)
m2.rotateY(0.2)
m2.move(10, 5, -3)
r2 = FreeCAD.Rotation(m2)
self.assertTrue(r1.isSame(r2, 1e-7), "Scale on matrix influenced rotation")
m3 = FreeCAD.Matrix()
m3.scale(-1)
r3 = FreeCAD.Rotation(m3)
r0 = FreeCAD.Rotation()
self.assertTrue(r3.isSame(r0, 1e-7), 'Scale on matrix influenced rotation')
self.assertTrue(r3.isSame(r0, 1e-7), "Scale on matrix influenced rotation")
m4 = FreeCAD.Matrix()
m4.scale(1.25,1.0,0.25)
m4.move(4,5,6)
r24 = FreeCAD.Rotation(m2*m4)
r42 = FreeCAD.Rotation(m4*m2)
self.assertTrue(r2.isSame(r24, 1e-7), 'Scale on matrix influenced rotation')
self.assertTrue(r2.isSame(r42, 1e-7), 'Scale on matrix influenced rotation')
m4.scale(1.25, 1.0, 0.25)
m4.move(4, 5, 6)
r24 = FreeCAD.Rotation(m2 * m4)
r42 = FreeCAD.Rotation(m4 * m2)
self.assertTrue(r2.isSame(r24, 1e-7), "Scale on matrix influenced rotation")
self.assertTrue(r2.isSame(r42, 1e-7), "Scale on matrix influenced rotation")
def testRotation(self):
r=FreeCAD.Rotation(1,0,0,0) # 180 deg around (1,0,0)
self.assertEqual(r.Axis, FreeCAD.Vector(1,0,0))
r = FreeCAD.Rotation(1, 0, 0, 0) # 180 deg around (1,0,0)
self.assertEqual(r.Axis, FreeCAD.Vector(1, 0, 0))
self.assertAlmostEqual(math.fabs(r.Angle), math.fabs(math.pi))
r=r.multiply(r) # identity
self.assertEqual(r.Axis, FreeCAD.Vector(0,0,1))
r = r.multiply(r) # identity
self.assertEqual(r.Axis, FreeCAD.Vector(0, 0, 1))
self.assertAlmostEqual(r.Angle, 0)
r=FreeCAD.Rotation(1,0,0,0)
r.Q=(0,0,0,1) # update axis and angle
s=FreeCAD.Rotation(0,0,0,1)
r = FreeCAD.Rotation(1, 0, 0, 0)
r.Q = (0, 0, 0, 1) # update axis and angle
s = FreeCAD.Rotation(0, 0, 0, 1)
self.assertEqual(r.Axis, s.Axis)
self.assertAlmostEqual(r.Angle, s.Angle)
self.assertTrue(r.isSame(s))
r=FreeCAD.Rotation(1,0,0,0)
r.Matrix=FreeCAD.Matrix() # update axis and angle
s=FreeCAD.Rotation(0,0,0,1)
r = FreeCAD.Rotation(1, 0, 0, 0)
r.Matrix = FreeCAD.Matrix() # update axis and angle
s = FreeCAD.Rotation(0, 0, 0, 1)
self.assertEqual(r.Axis, s.Axis)
self.assertAlmostEqual(r.Angle, s.Angle)
self.assertTrue(r.isSame(s))
r=FreeCAD.Rotation(1,0,0,0)
r.Axes=(FreeCAD.Vector(0,0,1),FreeCAD.Vector(0,0,1)) # update axis and angle
s=FreeCAD.Rotation(0,0,0,1)
r = FreeCAD.Rotation(1, 0, 0, 0)
r.Axes = (FreeCAD.Vector(0, 0, 1), FreeCAD.Vector(0, 0, 1)) # update axis and angle
s = FreeCAD.Rotation(0, 0, 0, 1)
self.assertEqual(r.Axis, s.Axis)
self.assertAlmostEqual(r.Angle, s.Angle)
self.assertTrue(r.isSame(s))
#add 360 deg to angle
r=FreeCAD.Rotation(FreeCAD.Vector(1,0,0),270)
s=FreeCAD.Rotation(FreeCAD.Vector(1,0,0),270+360)
# add 360 deg to angle
r = FreeCAD.Rotation(FreeCAD.Vector(1, 0, 0), 270)
s = FreeCAD.Rotation(FreeCAD.Vector(1, 0, 0), 270 + 360)
self.assertEqual(r.Axis, s.Axis)
#self.assertAlmostEqual(r.Angle, s.Angle + 2*math.pi)
# self.assertAlmostEqual(r.Angle, s.Angle + 2*math.pi)
self.assertTrue(r.isSame(s))
#subtract 360 deg from angle using Euler angles
r=FreeCAD.Rotation(0,0,180)
# subtract 360 deg from angle using Euler angles
r = FreeCAD.Rotation(0, 0, 180)
r.invert()
s=FreeCAD.Rotation(0,0,-180)
s = FreeCAD.Rotation(0, 0, -180)
self.assertTrue(r.isSame(s))
#subtract 360 deg from angle using quaternion
r=FreeCAD.Rotation(1,0,0,0)
s=FreeCAD.Rotation(-1,0,0,0)
#angles have the same sign
# subtract 360 deg from angle using quaternion
r = FreeCAD.Rotation(1, 0, 0, 0)
s = FreeCAD.Rotation(-1, 0, 0, 0)
# angles have the same sign
if r.Angle * s.Angle > 0:
self.assertEqual(r.Axis, s.Axis*(-1))
self.assertEqual(r.Axis, s.Axis * (-1))
else:
self.assertAlmostEqual(r.Angle, -s.Angle)
self.assertTrue(r.isSame(s))
@@ -338,10 +343,10 @@ class AlgebraTestCase(unittest.TestCase):
self.assertTrue(r.isSame(s))
# gimbal lock (north pole)
r=FreeCAD.Rotation()
r = FreeCAD.Rotation()
r.setYawPitchRoll(20, 90, 10)
a=r.getYawPitchRoll()
s=FreeCAD.Rotation()
a = r.getYawPitchRoll()
s = FreeCAD.Rotation()
s.setYawPitchRoll(*a)
self.assertAlmostEqual(a[0], 0.0)
self.assertAlmostEqual(a[1], 90.0)
@@ -349,10 +354,10 @@ class AlgebraTestCase(unittest.TestCase):
self.assertTrue(r.isSame(s, 1e-12))
# gimbal lock (south pole)
r=FreeCAD.Rotation()
r = FreeCAD.Rotation()
r.setYawPitchRoll(20, -90, 10)
a=r.getYawPitchRoll()
s=FreeCAD.Rotation()
a = r.getYawPitchRoll()
s = FreeCAD.Rotation()
s.setYawPitchRoll(*a)
self.assertAlmostEqual(a[0], 0.0)
self.assertAlmostEqual(a[1], -90.0)
@@ -382,20 +387,20 @@ class AlgebraTestCase(unittest.TestCase):
plm = FreeCAD.Placement()
rot.Matrix = FreeCAD.Matrix(1, 1, 0, 0, 1, 0, 0, 0, 1)
def testYawPitchRoll(self):
def getYPR1(yaw, pitch, roll):
r = FreeCAD.Rotation()
r.setYawPitchRoll(yaw, pitch, roll)
return r
def getYPR2(yaw, pitch, roll):
rx = FreeCAD.Rotation()
ry = FreeCAD.Rotation()
rz = FreeCAD.Rotation()
rx.Axis = FreeCAD.Vector(1,0,0)
ry.Axis = FreeCAD.Vector(0,1,0)
rz.Axis = FreeCAD.Vector(0,0,1)
rx.Axis = FreeCAD.Vector(1, 0, 0)
ry.Axis = FreeCAD.Vector(0, 1, 0)
rz.Axis = FreeCAD.Vector(0, 0, 1)
rx.Angle = math.radians(roll)
ry.Angle = math.radians(pitch)
@@ -404,9 +409,9 @@ class AlgebraTestCase(unittest.TestCase):
return rz.multiply(ry).multiply(rx)
angles = []
angles.append((10,10,10))
angles.append((13,45,-24))
angles.append((10,-90,20))
angles.append((10, 10, 10))
angles.append((13, 45, -24))
angles.append((10, -90, 20))
for i in angles:
r = getYPR1(*i)
@@ -414,21 +419,31 @@ class AlgebraTestCase(unittest.TestCase):
self.assertTrue(r.isSame(s, 1e-12))
def testBounding(self):
b=FreeCAD.BoundBox()
b = FreeCAD.BoundBox()
b.setVoid()
self.assertFalse(b.isValid(),"Bbox is not invalid")
b.add(0,0,0)
self.assertFalse(b.isValid(), "Bbox is not invalid")
b.add(0, 0, 0)
self.assertTrue(b.isValid(), "Bbox is invalid")
self.assertEqual(b.XLength, 0, "X length > 0")
self.assertEqual(b.YLength, 0, "Y length > 0")
self.assertEqual(b.ZLength, 0, "Z length > 0")
self.assertEqual(b.Center, FreeCAD.Vector(0,0,0), "Center is not at (0,0,0)")
self.assertEqual(b.Center, FreeCAD.Vector(0, 0, 0), "Center is not at (0,0,0)")
self.assertTrue(b.isInside(b.Center), "Center is not inside Bbox")
b.add(2,2,2)
self.assertTrue(b.isInside(b.getIntersectionPoint(b.Center,FreeCAD.Vector(0,1,0))),"Intersection point is not inside Bbox")
self.assertTrue(b.intersect(b),"Bbox doesn't intersect with itself")
self.assertFalse(b.intersected(FreeCAD.BoundBox(4,4,4,6,6,6)).isValid(),"Bbox should not intersect with Bbox outside")
self.assertEqual(b.intersected(FreeCAD.BoundBox(-2,-2,-2,2,2,2)).Center, b.Center,"Bbox is not a full subset")
b.add(2, 2, 2)
self.assertTrue(
b.isInside(b.getIntersectionPoint(b.Center, FreeCAD.Vector(0, 1, 0))),
"Intersection point is not inside Bbox",
)
self.assertTrue(b.intersect(b), "Bbox doesn't intersect with itself")
self.assertFalse(
b.intersected(FreeCAD.BoundBox(4, 4, 4, 6, 6, 6)).isValid(),
"Bbox should not intersect with Bbox outside",
)
self.assertEqual(
b.intersected(FreeCAD.BoundBox(-2, -2, -2, 2, 2, 2)).Center,
b.Center,
"Bbox is not a full subset",
)
def testMultLeftOrRight(self):
doc = FreeCAD.newDocument()
@@ -451,12 +466,13 @@ class AlgebraTestCase(unittest.TestCase):
FreeCAD.closeDocument(doc.Name)
class MatrixTestCase(unittest.TestCase):
def setUp(self):
self.mat = FreeCAD.Matrix()
def testOrder(self):
self.mat = FreeCAD.Matrix(1.0,2.0,3.0,4.0)
self.mat = FreeCAD.Matrix(1.0, 2.0, 3.0, 4.0)
self.assertEqual(self.mat.A11, 1.0)
self.assertEqual(self.mat.A12, 2.0)
self.assertEqual(self.mat.A13, 3.0)
@@ -507,18 +523,20 @@ class MatrixTestCase(unittest.TestCase):
def testMatrixPlacementMatrix(self):
# Example taken from https://forum.freecad.org/viewtopic.php?f=3&t=61000
mat = FreeCAD.Matrix(-0.470847778020266,
0.8150598976807029,
0.3376088628746235,
-11.25290913640202,
-0.8822144756796808,
-0.4350066260577338,
-0.180185641360483,
-2876.45492562325,
1.955470978815492e-9,
-0.3826834326750831,
0.923879538425552,
941.3822018176414)
mat = FreeCAD.Matrix(
-0.470847778020266,
0.8150598976807029,
0.3376088628746235,
-11.25290913640202,
-0.8822144756796808,
-0.4350066260577338,
-0.180185641360483,
-2876.45492562325,
1.955470978815492e-9,
-0.3826834326750831,
0.923879538425552,
941.3822018176414,
)
plm = FreeCAD.Placement(mat)
mat = plm.toMatrix()
self.assertEqual(mat.hasScale(), FreeCAD.ScaleType.NoScaling)
@@ -528,60 +546,60 @@ class MatrixTestCase(unittest.TestCase):
self.mat * "string"
def testUnity(self):
mat = FreeCAD.Matrix(2,0,0,0, 0,1,0,0, 0,0,2,0, 0,0,0,-1)
mat = FreeCAD.Matrix(2, 0, 0, 0, 0, 1, 0, 0, 0, 0, 2, 0, 0, 0, 0, -1)
self.assertFalse(mat.isUnity())
mat.unity()
self.assertTrue(mat.isUnity())
def testPower(self):
mat = FreeCAD.Matrix(2,0,0,0, 0,1,0,0, 0,0,2,0, 0,0,0,-1)
mat = FreeCAD.Matrix(2, 0, 0, 0, 0, 1, 0, 0, 0, 0, 2, 0, 0, 0, 0, -1)
with self.assertRaises(NotImplementedError):
mat ** "string"
mat2 = mat ** 0
mat2 = mat**0
self.assertTrue(mat2.isUnity())
self.assertEqual(mat ** -1, mat.inverse())
self.assertEqual(mat ** 1, mat)
self.assertEqual(mat ** 2, mat * mat)
self.assertEqual(mat ** 3, mat * mat * mat)
self.assertEqual(mat**-1, mat.inverse())
self.assertEqual(mat**1, mat)
self.assertEqual(mat**2, mat * mat)
self.assertEqual(mat**3, mat * mat * mat)
mat.nullify()
with self.assertRaises(RuntimeError):
mat ** -1
mat**-1
def testScale(self):
self.mat.scale(1., 2., 3.)
self.mat.scale(1.0, 2.0, 3.0)
self.assertEqual(self.mat.determinant(), 6.0)
self.assertEqual(self.mat.hasScale(), FreeCAD.ScaleType.NonUniformLeft)
self.mat.unity()
self.assertEqual(self.mat.hasScale(), FreeCAD.ScaleType.NoScaling)
self.mat.scale(2., 2., 2.)
self.mat.scale(2.0, 2.0, 2.0)
self.assertEqual(self.mat.hasScale(), FreeCAD.ScaleType.Uniform)
self.mat.rotateX(1.0)
self.assertEqual(self.mat.hasScale(), FreeCAD.ScaleType.Uniform)
self.mat.scale(1., 2., 3.)
self.mat.scale(1.0, 2.0, 3.0)
self.assertEqual(self.mat.hasScale(), FreeCAD.ScaleType.NonUniformLeft)
self.mat.unity()
self.mat.scale(1., 2., 3.)
self.mat.scale(1.0, 2.0, 3.0)
self.mat.rotateX(1.0)
self.assertEqual(self.mat.hasScale(), FreeCAD.ScaleType.NonUniformRight)
self.mat.unity()
self.mat.setCol(0, FreeCAD.Vector(1,2,3))
self.mat.setCol(1, FreeCAD.Vector(1,2,3))
self.mat.setCol(2, FreeCAD.Vector(1,2,3))
self.mat.setCol(0, FreeCAD.Vector(1, 2, 3))
self.mat.setCol(1, FreeCAD.Vector(1, 2, 3))
self.mat.setCol(2, FreeCAD.Vector(1, 2, 3))
self.assertEqual(self.mat.hasScale(), FreeCAD.ScaleType.Other)
self.mat.unity()
self.mat.setRow(0, FreeCAD.Vector(1,2,3))
self.mat.setRow(1, FreeCAD.Vector(1,2,3))
self.mat.setRow(2, FreeCAD.Vector(1,2,3))
self.mat.setRow(0, FreeCAD.Vector(1, 2, 3))
self.mat.setRow(1, FreeCAD.Vector(1, 2, 3))
self.mat.setRow(2, FreeCAD.Vector(1, 2, 3))
self.assertEqual(self.mat.hasScale(), FreeCAD.ScaleType.Other)
self.mat.unity()
self.mat.scale(-1.)
self.mat.scale(-1.0)
self.assertEqual(self.mat.hasScale(), FreeCAD.ScaleType.Uniform)
self.mat.scale(-2.)
self.mat.scale(-2.0)
self.assertEqual(self.mat.hasScale(), FreeCAD.ScaleType.Uniform)
def testShearing(self):
self.mat.setRow(1, FreeCAD.Vector(0,1,1))
self.mat.setRow(1, FreeCAD.Vector(0, 1, 1))
self.assertEqual(self.mat.hasScale(), FreeCAD.ScaleType.Other)
def testMultLeftOrRight(self):
@@ -589,7 +607,7 @@ class MatrixTestCase(unittest.TestCase):
mat1.rotateX(1.0)
mat2 = FreeCAD.Matrix()
mat2.scale(1., 2., 3.)
mat2.scale(1.0, 2.0, 3.0)
self.assertEqual((mat1 * mat2).hasScale(), FreeCAD.ScaleType.NonUniformRight)
self.assertEqual((mat2 * mat1).hasScale(), FreeCAD.ScaleType.NonUniformLeft)
@@ -612,12 +630,12 @@ class MatrixTestCase(unittest.TestCase):
self.mat.row("string")
self.assertEqual(type(self.mat.col(0)), FreeCAD.Vector)
self.assertEqual(type(self.mat.row(0)), FreeCAD.Vector)
self.mat.setCol(0, FreeCAD.Vector(1,0,0))
self.mat.setRow(0, FreeCAD.Vector(1,0,0))
self.mat.setCol(0, FreeCAD.Vector(1, 0, 0))
self.mat.setRow(0, FreeCAD.Vector(1, 0, 0))
def testTrace(self):
self.mat.scale(2., 2., 2.)
self.assertEqual(self.mat.trace(), FreeCAD.Vector(2., 2., 2.))
self.mat.scale(2.0, 2.0, 2.0)
self.assertEqual(self.mat.trace(), FreeCAD.Vector(2.0, 2.0, 2.0))
def testNumberProtocol(self):
with self.assertRaises(NotImplementedError):