# -*- coding: utf-8 -*- #*************************************************************************** #* Copyright (c) 2010 Juergen Riegel * #* * #* 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 = FreeCAD.Units.schemaTranslate(q1, idx) try: q2 = FreeCAD.Units.Quantity(t[0]) if math.fabs(q1.Value - q2.Value) > 0.01: print (" {} : {} : {} : {} : {}".format(q1, q2, t, i, val).encode("utf-8").strip()) except Exception as e: s = "{}: {}".format(e, t[0]) print (" ".join(str(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)