diff --git a/src/Mod/Spreadsheet/TestSpreadsheet.py b/src/Mod/Spreadsheet/TestSpreadsheet.py index ea1354993..f7802080c 100644 --- a/src/Mod/Spreadsheet/TestSpreadsheet.py +++ b/src/Mod/Spreadsheet/TestSpreadsheet.py @@ -4,12 +4,14 @@ import os import sys +import math import unittest import FreeCAD import Part import Sketcher import tempfile from FreeCAD import Base +from Units import Unit,Quantity v = Base.Vector @@ -24,7 +26,207 @@ class SpreadsheetCases(unittest.TestCase): self.TempPath = tempfile.gettempdir() FreeCAD.Console.PrintLog( ' Using temp path: ' + self.TempPath + '\n') + def testAggregates(self): + """ Test all aggregate functions """ + sheet = self.doc.addObject('Spreadsheet::Sheet','Spreadsheet') + sheet.set('B3', '4') + sheet.set('B4', '5') + sheet.set('B5', '6') + sheet.set('A1', '=sum(1)') + sheet.set('A2', '=sum(1;2)') + sheet.set('A3', '=sum(1;2;3)') + sheet.set('A4', '=sum(1;2;3;B3)') + sheet.set('A5', '=sum(1;2;3;B3:B5)') + sheet.set('A6', '=stddev(1;2;3)') + sheet.set('A7', '=average(1;2;3)') + sheet.set('A8', '=count(1;2;3)') + sheet.set('A9', '=min(1;2;3)') + sheet.set('A10', '=max(1;2;3)') + self.doc.recompute() + self.assertEqual(sheet.A1, 1) + self.assertEqual(sheet.A2, 3) + self.assertEqual(sheet.A3, 6) + self.assertEqual(sheet.A4, 10) + self.assertEqual(sheet.A5, 21) + self.assertEqual(sheet.A6, 1) + self.assertEqual(sheet.A7, 2) + self.assertEqual(sheet.A8, 3) + self.assertEqual(sheet.A9, 1) + self.assertEqual(sheet.A10, 3) + + def assertMostlyEqual(self, a, b): + if type(a) is Quantity: + self.assertTrue( math.fabs(a.Value - b.Value) < 1e-14) + self.assertTrue( a.Unit == b.Unit) + else: + self.assertTrue( math.fabs(a - b) < 1e-14) + + def testFunctions(self): + """ Test all built-in simple functions """ + doc = FreeCAD.newDocument() + sheet = self.doc.addObject('Spreadsheet::Sheet','Spreadsheet') + sheet.set('A1', '=cos(60)') # Cos + sheet.set('B1', '=cos(60deg)') + sheet.set('C1', '=cos(pi / 2 * 1rad)') + sheet.set('A2', '=sin(30)') # Sin + sheet.set('B2', '=sin(30deg)') + sheet.set('C2', '=sin(pi / 6 * 1rad)') + sheet.set('A3', '=tan(45)') # Tan + sheet.set('B3', '=tan(45deg)') + sheet.set('C3', '=tan(pi / 4 * 1rad)') + sheet.set('A4', '=abs(3)') # Abs + sheet.set('B4', '=abs(-3)') + sheet.set('C4', '=abs(-3mm)') + sheet.set('A5', '=exp(3)') # Exp + sheet.set('B5', '=exp(-3)') + sheet.set('C5', '=exp(-3mm)') + sheet.set('A6', '=log(3)') # Log + sheet.set('B6', '=log(-3)') + sheet.set('C6', '=log(-3mm)') + sheet.set('A7', '=log10(10)') # Log10 + sheet.set('B7', '=log10(-3)') + sheet.set('C7', '=log10(-3mm)') + sheet.set('A8', '=round(3.4)')# Round + sheet.set('B8', '=round(3.6)') + sheet.set('C8', '=round(-3.4)') + sheet.set('D8', '=round(-3.6)') + sheet.set('E8', '=round(3.4mm)') + sheet.set('F8', '=round(3.6mm)') + sheet.set('G8', '=round(-3.4mm)') + sheet.set('H8', '=round(-3.6mm)') + sheet.set('A9', '=trunc(3.4)')# Trunc + sheet.set('B9', '=trunc(3.6)') + sheet.set('C9', '=trunc(-3.4)') + sheet.set('D9', '=trunc(-3.6)') + sheet.set('E9', '=trunc(3.4mm)') + sheet.set('F9', '=trunc(3.6mm)') + sheet.set('G9', '=trunc(-3.4mm)') + sheet.set('H9', '=trunc(-3.6mm)') + sheet.set('A10', '=ceil(3.4)') # Ceil + sheet.set('B10', '=ceil(3.6)') + sheet.set('C10', '=ceil(-3.4)') + sheet.set('D10', '=ceil(-3.6)') + sheet.set('E10', '=ceil(3.4mm)') + sheet.set('F10', '=ceil(3.6mm)') + sheet.set('G10', '=ceil(-3.4mm)') + sheet.set('H10', '=ceil(-3.6mm)') + sheet.set('A11', '=floor(3.4)')# Floor + sheet.set('B11', '=floor(3.6)') + sheet.set('C11', '=floor(-3.4)') + sheet.set('D11', '=floor(-3.6)') + sheet.set('E11', '=floor(3.4mm)') + sheet.set('F11', '=floor(3.6mm)') + sheet.set('G11', '=floor(-3.4mm)') + sheet.set('H11', '=floor(-3.6mm)') + sheet.set('A12', '=asin(0.5)') # Asin + sheet.set('B12', '=asin(0.5mm)') + sheet.set('A13', '=acos(0.5)') # Acos + sheet.set('B13', '=acos(0.5mm)') + sheet.set('A14', '=atan(sqrt(3))') # Atan + sheet.set('B14', '=atan(0.5mm)') + sheet.set('A15', '=sinh(0.5)') # Sinh + sheet.set('B15', '=sinh(0.5mm)') + sheet.set('A16', '=cosh(0.5)') # Cosh + sheet.set('B16', '=cosh(0.5mm)') + sheet.set('A17', '=tanh(0.5)') # Tanh + sheet.set('B17', '=tanh(0.5mm)') + sheet.set('A18', '=sqrt(4)') # Sqrt + sheet.set('B18', '=sqrt(4mm^2)') + sheet.set('A19', '=mod(7; 4)') # Mod + sheet.set('B19', '=mod(-7; 4)') + sheet.set('C19', '=mod(7mm; 4)') + sheet.set('D19', '=mod(7mm; 4mm)') + sheet.set('A20', '=atan2(3; 3)') # Atan2 + sheet.set('B20', '=atan2(-3; 3)') + sheet.set('C20', '=atan2(3mm; 3)') + sheet.set('D20', '=atan2(3mm; 3mm)') + sheet.set('A21', '=pow(7; 4)') # Pow + sheet.set('B21', '=pow(-7; 4)') + sheet.set('C21', '=pow(7mm; 4)') + sheet.set('D21', '=pow(7mm; 4mm)') + self.doc.recompute() + self.assertMostlyEqual(sheet.A1, 0.5) # Cos + self.assertMostlyEqual(sheet.B1, 0.5) + self.assertMostlyEqual(sheet.C1, 0) + self.assertMostlyEqual(sheet.A2, 0.5) # Sin + self.assertMostlyEqual(sheet.B2, 0.5) + self.assertMostlyEqual(sheet.C2, 0.5) + self.assertMostlyEqual(sheet.A3, 1) # Tan + self.assertMostlyEqual(sheet.B3, 1) + self.assertMostlyEqual(sheet.C3, 1 ) + self.assertMostlyEqual(sheet.A4, 3) # Abs + self.assertMostlyEqual(sheet.B4, 3) + self.assertMostlyEqual(sheet.C4, Quantity('3 mm')) + self.assertMostlyEqual(sheet.A5, math.exp(3)) # Exp + self.assertMostlyEqual(sheet.B5, math.exp(-3)) + self.assertEqual(sheet.C5, u'ERR: Unit must be empty.') + self.assertMostlyEqual(sheet.A6, math.log(3)) # Log + self.assertTrue(math.isnan(sheet.B6)) + self.assertEqual(sheet.C6, u'ERR: Unit must be empty.') + self.assertMostlyEqual(sheet.A7, math.log10(10)) # Log10 + self.assertTrue(math.isnan(sheet.B7)) + self.assertEqual(sheet.C7, u'ERR: Unit must be empty.') + self.assertMostlyEqual(sheet.A8, 3) # Round + self.assertMostlyEqual(sheet.B8, 4) + self.assertMostlyEqual(sheet.C8, -3) + self.assertMostlyEqual(sheet.D8, -4) + self.assertEqual(sheet.E8, Quantity('3 mm')) + self.assertEqual(sheet.F8, Quantity('4 mm')) + self.assertEqual(sheet.G8, Quantity('-3 mm')) + self.assertEqual(sheet.H8, Quantity('-4 mm')) + self.assertMostlyEqual(sheet.A9, 3)# Trunc + self.assertMostlyEqual(sheet.B9, 3) + self.assertMostlyEqual(sheet.C9, -3) + self.assertMostlyEqual(sheet.D9, -3) + self.assertEqual(sheet.E9, Quantity('3 mm')) + self.assertEqual(sheet.F9, Quantity('3 mm')) + self.assertEqual(sheet.G9, Quantity('-3 mm')) + self.assertEqual(sheet.H9, Quantity('-3 mm')) + self.assertMostlyEqual(sheet.A10, 4) # Ceil + self.assertMostlyEqual(sheet.B10, 4) + self.assertMostlyEqual(sheet.C10, -3) + self.assertMostlyEqual(sheet.D10, -3) + self.assertMostlyEqual(sheet.E10, Quantity('4 mm')) + self.assertMostlyEqual(sheet.F10, Quantity('4 mm')) + self.assertMostlyEqual(sheet.G10, Quantity('-3 mm')) + self.assertMostlyEqual(sheet.H10, Quantity('-3 mm')) + self.assertMostlyEqual(sheet.A11, 3)# Floor + self.assertMostlyEqual(sheet.B11, 3) + self.assertMostlyEqual(sheet.C11, -4) + self.assertMostlyEqual(sheet.D11, -4) + self.assertMostlyEqual(sheet.E11, Quantity('3 mm')) + self.assertMostlyEqual(sheet.F11, Quantity('3 mm')) + self.assertMostlyEqual(sheet.G11, Quantity('-4 mm')) + self.assertMostlyEqual(sheet.H11, Quantity('-4 mm')) + self.assertMostlyEqual(sheet.A12, Quantity('30 deg')) # Asin + self.assertEqual(sheet.B12, u'ERR: Unit must be empty.') + self.assertMostlyEqual(sheet.A13, Quantity('60 deg')) # Acos + self.assertEqual(sheet.B13, u'ERR: Unit must be empty.') + self.assertMostlyEqual(sheet.A14, Quantity('60 deg')) # Atan + self.assertEqual(sheet.B14, u'ERR: Unit must be empty.') + self.assertMostlyEqual(sheet.A15, math.sinh(0.5)) # Sinh + self.assertEqual(sheet.B15, u'ERR: Unit must be empty.') + self.assertMostlyEqual(sheet.A16, math.cosh(0.5)) # Cosh + self.assertEqual(sheet.B16, u'ERR: Unit must be empty.') + self.assertMostlyEqual(sheet.A17, math.tanh(0.5)) # Tanh + self.assertEqual(sheet.B17, u'ERR: Unit must be empty.') + self.assertMostlyEqual(sheet.A18, 2) # Sqrt + self.assertMostlyEqual(sheet.B18, Quantity('2 mm')) + self.assertMostlyEqual(sheet.A19, 3) # Mod + self.assertMostlyEqual(sheet.B19, -3) + self.assertMostlyEqual(sheet.C19, Quantity('3 mm')) + self.assertEqual(sheet.D19, u'ERR: Second argument must have empty unit.') + self.assertMostlyEqual(sheet.A20, Quantity('45 deg')) # Atan2 + self.assertMostlyEqual(sheet.B20, Quantity('-45 deg')) + self.assertEqual(sheet.C20, u'ERR: Units must be equal') + self.assertMostlyEqual(sheet.D20, Quantity('45 deg')) + self.assertMostlyEqual(sheet.A21, 2401) # Pow + self.assertMostlyEqual(sheet.B21, 2401) + self.assertMostlyEqual(sheet.C21, Quantity('2401mm^4')) + self.assertEqual(sheet.D21, u'ERR: Exponent is not allowed to have a unit.') + def testRelationalOperators(self): + """ Test relational operators """ sheet = self.doc.addObject('Spreadsheet::Sheet','Spreadsheet') # All should be 1 as result sheet.set('A1', '=1 == 1 ? 1 : 0') @@ -78,12 +280,74 @@ class SpreadsheetCases(unittest.TestCase): self.assertEqual(sheet.A23, 1) self.assertEqual(sheet.A24, 1) + def testUnits(self): + """ Units -- test unit calculations. """ + sheet = self.doc.addObject('Spreadsheet::Sheet','Spreadsheet') + sheet.set('A1', '=2mm + 3mm') + sheet.set('A2', '=2mm - 3mm') + sheet.set('A3', '=2mm * 3mm') + sheet.set('A4', '=4mm / 2mm') + sheet.set('A5', '=(4mm)^2') + sheet.set('A6', '=5(mm^2)') + sheet.set('A7', '=5mm^2') # ^2 operates on whole number + sheet.set('A8', '=5') + sheet.set('A9', '=5*1/K') # Currently fails + sheet.set('A10', '=5 K^-1') # Currently fails + sheet.set('A11', '=9.8 m/s^2') # Currently fails + sheet.setDisplayUnit('A8', '1/K') + self.doc.recompute() + self.assertEqual(sheet.A1, Quantity('5mm')) + self.assertEqual(sheet.A2, Quantity('-1 mm')) + self.assertEqual(sheet.A3, Quantity('6 mm^2')) + self.assertEqual(sheet.A4, Quantity('2')) + self.assertEqual(sheet.A5, Quantity('16 mm^2')) + self.assertEqual(sheet.A6, Quantity('5 mm^2')) + self.assertEqual(sheet.A7, Quantity('5 mm^2')) + self.assertEqual(sheet.A8, Quantity('5')) + self.assertEqual(sheet.A9, Quantity('5 K^-1')) + self.assertEqual(sheet.A10, Quantity('5 K^-1')) + self.assertEqual(sheet.A11, Quantity('9.8 m/s^2')) + def testPrecedence(self): """ Precedence -- test precedence for relational operators and conditional operator. """ sheet = self.doc.addObject('Spreadsheet::Sheet','Spreadsheet') sheet.set('A1', '=1 < 2 ? 3 : 4') sheet.set('A2', '=1 + 2 < 3 + 4 ? 5 + 6 : 7 + 8') sheet.set('A3', '=1 + 2 * 1 < 3 + 4 ? 5 * 2 + 6 * 3 + 2 ^ 4 : 7 * 2 + 8 * 3 + 2 ^ 3') + sheet.set('A4', '=123') + sheet.set('A5', '=123 + 321') + sheet.set('A6', '=123 * 2 + 321') + sheet.set('A7', '=123 * 2 + 333 / 3') + sheet.set('A8', '=123 * (2 + 321)') + sheet.set('A9', '=3 ^ 4') + sheet.set('A10', '=3 ^ 4 * 2') + sheet.set('A11', '=3 ^ (4 * 2)') + sheet.set('A12', '=3 ^ 4 + 4') + sheet.set('A13', '=1 + 4 / 2 + 5') + sheet.set('A14', '=(3 + 6) / (1 + 2)') + sheet.set('A15', '=1 * 2 / 3 * 4') + sheet.set('A16', '=(1 * 2) / (3 * 4)') + # Test associativity + sheet.set('A17', '=3 ^ 4 ^ 2') # exponentiation is left-associative; to follow excel, openoffice, matlab, octave + sheet.set('A18', '=3 ^ (4 ^ 2)') # exponentiation is left-associative + sheet.set('A19', '=(3 ^ 4) ^ 2') # exponentiation is left-associative + sheet.set('A20', '=3 + 4 + 2') + sheet.set('A21', '=3 + (4 + 2)') + sheet.set('A22', '=(3 + 4) + 2') + sheet.set('A23', '=3 - 4 - 2') + sheet.set('A24', '=3 - (4 - 2)') + sheet.set('A25', '=(3 - 4) - 2') + sheet.set('A26', '=3 * 4 * 2') + sheet.set('A27', '=3 * (4 * 2)') + sheet.set('A28', '=(3 * 4) * 2') + sheet.set('A29', '=3 / 4 / 2') + sheet.set('A30', '=3 / (4 / 2)') + sheet.set('A31', '=(3 / 4) / 2') + sheet.set('A32', '=pi * 3') + sheet.set('A33', '=A32 / 3') + sheet.set('A34', '=1 < 2 ? <> : <>') + sheet.set('A35', '=min(A32:A33)') + sheet.set('A36', '=(1 < 2 ? 0 : 1) * 3') self.doc.recompute() self.assertEqual(sheet.getContents("A1"), "=1 < 2 ? 3 : 4") self.assertEqual(sheet.getContents("A2"), "=1 + 2 < 3 + 4 ? 5 + 6 : 7 + 8") @@ -91,7 +355,112 @@ class SpreadsheetCases(unittest.TestCase): self.assertEqual(sheet.A1, 3) self.assertEqual(sheet.A2, 11) self.assertEqual(sheet.A3, 44) + self.assertEqual(sheet.A4, 123) + self.assertEqual(sheet.A5, 444) + self.assertEqual(sheet.A6, 567) + self.assertEqual(sheet.A7, 357) + self.assertEqual(sheet.A8, 39729) + self.assertEqual(sheet.A9, 81) + self.assertEqual(sheet.A10, 162) + self.assertEqual(sheet.A11, 6561) + self.assertEqual(sheet.A12, 85) + self.assertEqual(sheet.A13, 8) + self.assertEqual(sheet.A14, 3) + self.assertEqual(sheet.A15, 8.0/3) + self.assertEqual(sheet.A16, 1.0/6) + self.assertEqual(sheet.A17, 6561) + self.assertEqual(sheet.A18, 43046721) + self.assertEqual(sheet.A19, 6561) + self.assertEqual(sheet.A20, 9) + self.assertEqual(sheet.A21, 9) + self.assertEqual(sheet.A22, 9) + self.assertEqual(sheet.A23, -3) + self.assertEqual(sheet.A24, 1) + self.assertEqual(sheet.A25, -3) + self.assertEqual(sheet.A26, 24) + self.assertEqual(sheet.A27, 24) + self.assertEqual(sheet.A28, 24) + self.assertEqual(sheet.A29, 3.0/8) + self.assertEqual(sheet.A30, 3.0/2) + self.assertEqual(sheet.A31, 3.0/8) + self.assertEqual(sheet.getContents('A1'), '=1 < 2 ? 3 : 4') + self.assertEqual(sheet.getContents('A2'), '=1 + 2 < 3 + 4 ? 5 + 6 : 7 + 8') + self.assertEqual(sheet.getContents('A3'), '=1 + 2 * 1 < 3 + 4 ? 5 * 2 + 6 * 3 + 2 ^ 4 : 7 * 2 + 8 * 3 + 2 ^ 3') + self.assertEqual(sheet.getContents('A4'), '123') + self.assertEqual(sheet.getContents('A5'), '=123 + 321') + self.assertEqual(sheet.getContents('A6'), '=123 * 2 + 321') + self.assertEqual(sheet.getContents('A7'), '=123 * 2 + 333 / 3') + self.assertEqual(sheet.getContents('A8'), '=123 * (2 + 321)') + self.assertEqual(sheet.getContents('A9'), '=3 ^ 4') + self.assertEqual(sheet.getContents('A10'), '=3 ^ 4 * 2') + self.assertEqual(sheet.getContents('A11'), '=3 ^ (4 * 2)') + self.assertEqual(sheet.getContents('A12'), '=3 ^ 4 + 4') + self.assertEqual(sheet.getContents('A13'), '=1 + 4 / 2 + 5') + self.assertEqual(sheet.getContents('A14'), '=(3 + 6) / (1 + 2)') + self.assertEqual(sheet.getContents('A15'), '=1 * 2 / 3 * 4') + self.assertEqual(sheet.getContents('A16'), '=1 * 2 / 3 * 4') + self.assertEqual(sheet.getContents('A17'), '=3 ^ 4 ^ 2') + self.assertEqual(sheet.getContents('A18'), '=3 ^ (4 ^ 2)') + self.assertEqual(sheet.getContents('A19'), '=3 ^ 4 ^ 2') + self.assertEqual(sheet.getContents('A20'), '=3 + 4 + 2') + self.assertEqual(sheet.getContents('A21'), '=3 + 4 + 2') + self.assertEqual(sheet.getContents('A22'), '=3 + 4 + 2') + self.assertEqual(sheet.getContents('A23'), '=3 - 4 - 2') + self.assertEqual(sheet.getContents('A24'), '=3 - (4 - 2)') + self.assertEqual(sheet.getContents('A25'), '=3 - 4 - 2') + self.assertEqual(sheet.getContents('A26'), '=3 * 4 * 2') + self.assertEqual(sheet.getContents('A27'), '=3 * 4 * 2') + self.assertEqual(sheet.getContents('A28'), '=3 * 4 * 2') + self.assertEqual(sheet.getContents('A29'), '=3 / 4 / 2') + self.assertEqual(sheet.getContents('A30'), '=3 / (4 / 2)') + self.assertEqual(sheet.getContents('A31'), '=3 / 4 / 2') + self.assertEqual(sheet.getContents('A32'), '=pi * 3') + self.assertEqual(sheet.getContents('A33'), '=A32 / 3') + self.assertEqual(sheet.getContents('A34'), '=1 < 2 ? <> : <>') + self.assertEqual(sheet.getContents('A35'), '=min(A32:A33)') + self.assertEqual(sheet.getContents('A36'), '=(1 < 2 ? 0 : 1) * 3') + def testNumbers(self): + """ Test different numbers """ + sheet = self.doc.addObject('Spreadsheet::Sheet','Spreadsheet') + sheet.set('A1', '1') + sheet.set('A2', '1.5') + sheet.set('A3', '.5') + sheet.set('A4', '1e2') + sheet.set('A5', '1E2') + sheet.set('A6', '1e-2') + sheet.set('A7', '1E-2') + sheet.set('A8', '1.5e2') + sheet.set('A9', '1.5E2') + sheet.set('A10', '1.5e-2') + sheet.set('A11', '1.5E-2') + sheet.set('A12', '.5e2') + sheet.set('A13', '.5E2') + sheet.set('A14', '.5e-2') + sheet.set('A15', '.5E-2') + sheet.set('A16', '1/1') + sheet.set('A17', '1/2') + sheet.set('A18', '2/4') + self.doc.recompute() + self.assertEqual(sheet.A1, 1) + self.assertEqual(sheet.A2, 1.5) + self.assertEqual(sheet.A3, 0.5) + self.assertEqual(sheet.A4, 1e2) + self.assertEqual(sheet.A5, 1e2) + self.assertEqual(sheet.A6, 1e-2) + self.assertEqual(sheet.A7, 1e-2) + self.assertEqual(sheet.A8, 1.5e2) + self.assertEqual(sheet.A9, 1.5e2) + self.assertEqual(sheet.A10, 1.5e-2) + self.assertEqual(sheet.A11, 1.5e-2) + self.assertEqual(sheet.A12, 0.5e2) + self.assertEqual(sheet.A13, 0.5e2) + self.assertEqual(sheet.A14, 0.5e-2) + self.assertEqual(sheet.A15, 0.5e-2) + self.assertEqual(sheet.A16, 1) + self.assertEqual(sheet.A17, 0.5) + self.assertEqual(sheet.A18, 0.5) + def testRemoveRows(self): """ Removing rows -- check renaming of internal cells """ sheet = self.doc.addObject('Spreadsheet::Sheet','Spreadsheet')