Files
create/src/Mod/Test/UnitTests.py
Stefan Brüns 271abecb53 Test: Provide more useful information when unit translation fails
Add the input quantity and the target scheme to the exception output.
Also make the output formatting more canonical and remove the unused
"s" temporary.
2022-03-06 11:20:09 -05:00

184 lines
8.0 KiB
Python

# -*- coding: utf-8 -*-
#***************************************************************************
#* Copyright (c) 2010 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
import unittest
import math
def tu(str):
return FreeCAD.Units.Quantity(str).Value
def ts(q):
return q.UserString
def ts2(q):
return FreeCAD.Units.Quantity(q.UserString).UserString
#---------------------------------------------------------------------------
# define the functions to test the FreeCAD UnitApi code
#---------------------------------------------------------------------------
def compare(x, y):
return math.fabs(x - y) < 0.00001
class UnitBasicCases(unittest.TestCase):
def setUp(self):
par = FreeCAD.ParamGet("User parameter:BaseApp/Preferences/Units")
dec = par.GetInt("Decimals")
self.delta = math.pow(10,-dec)
def testConversions(self):
#tu = FreeCAD.Units.translateUnit
self.failUnless(compare(tu('10 m'), 10000.0))
self.failUnless(compare(tu('3/8 in'), 9.525))
self.failUnless(compare(tu('100 km/h'), 27777.77777777))
self.failUnless(compare(tu('m^2*kg*s^-3*A^-2'), 1000000.0))
self.failUnless(compare(tu('(m^2*kg)/(A^2*s^3)'), 1000000.0))
self.failUnless(compare(tu('2*pi rad'), 360.0))
self.failUnless(compare(tu('2*pi rad') / tu('gon'), 400.0))
self.failUnless(compare(tu('999 kg') / tu('1 m^3'), 0.000009999))
def testImperial(self):
#tu = FreeCAD.Units.translateUnit
self.failUnless(compare(tu('3/8in'), 9.525))
#self.failUnless(compare(tu('1fo(3+7/16)in'),392.112500))thisgivesaparsersyntaxerror!!!
self.failUnless(compare(tu('1\'(3+7/16)"'), 392.112500))
psi = FreeCAD.Units.parseQuantity("1psi")
mpa = psi.getValueAs("MPa").Value
self.assertAlmostEqual(0.006894744825, mpa, delta=self.delta)
kpa = psi.getValueAs("kPa").Value
self.assertAlmostEqual(6.894744825494, kpa, delta=self.delta)
ksi = FreeCAD.Units.parseQuantity("1ksi")
mpa = ksi.getValueAs("MPa").Value
self.assertAlmostEqual(6.894744825494, mpa, delta=self.delta)
kpa = ksi.getValueAs("kPa").Value
self.assertAlmostEqual(6894.744825494, kpa, delta=self.delta)
def testSelfConsistency(self):
qu = FreeCAD.Units.Quantity("0.23 W/m/K")
self.assertTrue(ts(qu), ts2(qu))
qu = FreeCAD.Units.Quantity("237 mm*kg/(s^3*K)")
self.assertTrue(ts(qu), ts2(qu))
qu = FreeCAD.Units.Quantity("237.000 W/mm/K")
self.assertTrue(ts(qu), ts2(qu))
def testDivide(self):
qu1 = FreeCAD.Units.Quantity("1 m/s")
qu2 = FreeCAD.Units.Quantity("m/s")
self.assertTrue(qu1/qu2, 1)
def testSchemes(self):
schemes = FreeCAD.Units.listSchemas()
num = len(schemes)
psi = FreeCAD.Units.parseQuantity("1psi")
for i in range(num):
t = FreeCAD.Units.schemaTranslate(psi, i)
v = FreeCAD.Units.parseQuantity(t[0]).getValueAs("psi")
self.assertAlmostEqual(1, v.Value, msg="Failed with \"{0}\" scheme: {1} != 1 (delta: {2})".format(schemes[i], v.Value, self.delta), delta=self.delta)
ksi = FreeCAD.Units.parseQuantity("1ksi")
for i in range(num):
t = FreeCAD.Units.schemaTranslate(ksi, i)
v = FreeCAD.Units.parseQuantity(t[0]).getValueAs("ksi")
self.assertAlmostEqual(1, v.Value, msg="Failed with \"{0}\" scheme: {1} != 1 (delta: {2})".format(schemes[i], v.Value, self.delta), delta=self.delta)
def testSchemeTranslation(self):
quantities = []
for i in dir(FreeCAD.Units):
if issubclass(type(getattr(FreeCAD.Units, i)), FreeCAD.Units.Quantity):
quantities.append(i)
schemes = FreeCAD.Units.listSchemas()
for i in quantities:
q1 = getattr(FreeCAD.Units, i)
q1 = FreeCAD.Units.Quantity(q1)
q1.Format = {'Precision': 16}
for idx, val in enumerate(schemes):
[t, amountPerUnit, unit] = FreeCAD.Units.schemaTranslate(q1, idx)
try:
q2 = FreeCAD.Units.Quantity(t)
if math.fabs(q1.Value - q2.Value) > 0.01:
print (" {} : {} : {} : {} : {}".format(q1, q2, t, i, val).encode("utf-8").strip())
except Exception as e:
print ("{} : {} : {} : {}".format(q1, i, val, e).encode("utf-8").strip())
def testVoltage(self):
q1 = FreeCAD.Units.Quantity("1e20 V")
t = FreeCAD.Units.schemaTranslate(q1, 0) # Standard
q2 = FreeCAD.Units.Quantity(t[0])
self.assertAlmostEqual(q1.Value, q2.Value, delta=self.delta)
def testEnergy(self):
q1 = FreeCAD.Units.Quantity("1e20 J")
t = FreeCAD.Units.schemaTranslate(q1, 0) # Standard
q2 = FreeCAD.Units.Quantity(t[0])
self.assertAlmostEqual(q1.Value, q2.Value, delta=self.delta)
def testTrigonometric(self):
#tu=FreeCAD.Units.translateUnit
self.failUnless(compare(tu('sin(pi)'), math.sin(math.pi)))
self.failUnless(compare(tu('cos(pi)'), math.cos(math.pi)))
self.failUnless(compare(tu('tan(pi)'), math.tan(math.pi)))
def testQuantity(self):
length = FreeCAD.Units.Quantity(1, "m")
self.assertEqual(length.Value, 1000)
self.assertEqual(length.Unit, FreeCAD.Units.Length)
def testToString(self):
value = FreeCAD.Units.toNumber(1023, 'g', 2)
self.assertEqual(float(value), 1000)
value = FreeCAD.Units.toNumber(1023, 'g', 3)
self.assertEqual(float(value), 1020)
value = FreeCAD.Units.toNumber(1023, 'f', 2)
self.assertEqual(float(value), 1023)
value = FreeCAD.Units.toNumber(1023, 'e', 1)
self.assertEqual(float(value), 1000)
value = FreeCAD.Units.toNumber(1023, 'e', 2)
self.assertEqual(float(value), 1020)
value = FreeCAD.Units.toNumber(1023, 'e', 3)
self.assertEqual(float(value), 1023)
q = FreeCAD.Units.Quantity("1023")
value = FreeCAD.Units.toNumber(q, 'f', 2)
self.assertEqual(float(value), 1023)
with self.assertRaises(TypeError):
FreeCAD.Units.toNumber("1023", 'g', 2)
with self.assertRaises(ValueError):
FreeCAD.Units.toNumber(1023, 'gg', 2)
with self.assertRaises(ValueError):
FreeCAD.Units.toNumber(1023, 's', 2)