1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
|
#!/usr/bin/env python3
########################################################################
#
# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at http://mozilla.org/MPL/2.0/.
#
########################################################################
import unittest
import os
import os.path
import mmap
from orcus import ods, FormulaTokenType
import file_load_common as common
class TestCase(unittest.TestCase):
@classmethod
def setUpClass(cls):
# base directory for ods test files.
basedir = os.path.join(os.path.dirname(__file__), "..", "ods")
cls.basedir = os.path.normpath(basedir)
def test_import(self):
test_dirs = ("raw-values-1", "formula-1", "formula-2")
for test_dir in test_dirs:
test_dir = os.path.join(self.basedir, test_dir)
common.run_test_dir(self, test_dir, common.DocLoader(ods))
def test_formula_tokens(self):
filepath = os.path.join(self.basedir, "formula-1", "input.ods")
with open(filepath, "rb") as f:
doc = ods.read(f, recalc=False)
self.assertEqual(len(doc.sheets), 1)
# The 'Formula' sheet contains 4 formula cells in A1:A4.
sheet = doc.sheets[0]
self.assertEqual(sheet.name, "Formula")
rows = [row for row in sheet.get_rows()]
self.assertEqual(len(rows), 4)
expected = ("1*2", "12/3", "AVERAGE($A1:A$2)", "SUM($A$1:$A$3)")
for row, expected_formula in zip(sheet.get_rows(), expected):
c = row[0]
self.assertEqual(c.formula, expected_formula)
expected = (
(("1", FormulaTokenType.VALUE), ("*", FormulaTokenType.OPERATOR), ("2", FormulaTokenType.VALUE)),
(("12", FormulaTokenType.VALUE), ("/", FormulaTokenType.OPERATOR), ("3", FormulaTokenType.VALUE)),
(("AVERAGE", FormulaTokenType.FUNCTION), ("(", FormulaTokenType.OPERATOR), ("$A1:A$2", FormulaTokenType.REFERENCE), (")", FormulaTokenType.OPERATOR)),
(("SUM", FormulaTokenType.FUNCTION), ("(", FormulaTokenType.OPERATOR), ("$A$1:$A$3", FormulaTokenType.REFERENCE), (")", FormulaTokenType.OPERATOR)),
)
for row, expected_formula_tokens in zip(sheet.get_rows(), expected):
c = row[0]
iter = c.get_formula_tokens()
for token, expected_token in zip(iter, expected_formula_tokens):
self.assertEqual(str(token), expected_token[0])
self.assertEqual(token.type, expected_token[1])
if __name__ == '__main__':
unittest.main()
|
