summaryrefslogtreecommitdiff
path: root/sc/qa/uitest/statistics/regression.py
blob: 588b1bea77fc64839659962157859cae98bcc243 (plain)
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
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
# -*- tab-width: 4; indent-tabs-mode: nil; py-indent-offset: 4 -*-
#
# 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/.
#
from uitest.framework import UITestCase
from uitest.uihelper.common import get_state_as_dict, get_url_for_data_file
from uitest.uihelper.common import select_pos
from uitest.uihelper.calc import enter_text_to_cell
from libreoffice.calc.document import get_sheet_from_doc
from libreoffice.calc.document import get_cell_by_position
from libreoffice.uno.propertyvalue import mkPropertyValues

class regression(UITestCase):
    def test_regression_row(self):
        self._regression_check(data_groupedby_column = False)

    def test_regression_column(self):
        self._regression_check(data_groupedby_column = True)

    def _regression_check(self, data_groupedby_column = True):
        calc_doc = self.ui_test.load_file(get_url_for_data_file("regression.ods"))
        xCalcDoc = self.xUITest.getTopFocusWindow()
        gridwin = xCalcDoc.getChild("grid_window")
        document = self.ui_test.get_component()

        # Initially the final check status is "FALSE" (failed).
        self.assertEqual(get_cell_by_position(document, 14, 1, 6).getString(), "FALSE",
                         "Check status must be FALSE before the test")
        self._do_regression(regression_type = "LINEAR", data_groupedby_column = data_groupedby_column, calc_intercept = True)
        self._do_regression(regression_type = "LINEAR", data_groupedby_column = data_groupedby_column, calc_intercept = False)
        self._do_regression(regression_type = "LOG", data_groupedby_column = data_groupedby_column)
        self._do_regression(regression_type = "POWER", data_groupedby_column = data_groupedby_column)
        self.assertEqual(get_cell_by_position(document, 14, 1, 6).getString(), "TRUE",
                         "One of more of the checks failed for data_groupedby_column = {}, manually try with the document".
                         format(data_groupedby_column))
        self.ui_test.close_doc()

    def _do_regression(self, regression_type, data_groupedby_column = True, calc_intercept = True):
        assert(regression_type == "LINEAR" or regression_type == "LOG" or regression_type == "POWER")
        self.ui_test.execute_modeless_dialog_through_command(".uno:RegressionDialog")
        xDialog = self.xUITest.getTopFocusWindow()
        xvariable1rangeedit = xDialog.getChild("variable1-range-edit")
        xvariable2rangeedit = xDialog.getChild("variable2-range-edit")
        xoutputrangeedit = xDialog.getChild("output-range-edit")
        xwithlabelscheck = xDialog.getChild("withlabels-check")
        xgroupedbyrowsradio = xDialog.getChild("groupedby-rows-radio")
        xgroupedbycolumnsradio = xDialog.getChild("groupedby-columns-radio")
        xlinearradio = xDialog.getChild("linear-radio")
        xlogarithmicradio = xDialog.getChild("logarithmic-radio")
        xpowerradio = xDialog.getChild("power-radio")
        xnointerceptcheck = xDialog.getChild("nointercept-check")

        ## Set the X, Y and output ranges
        xvariable1rangeedit.executeAction("FOCUS", tuple()) # Without this the range parser does not kick in somehow
        xvariable1rangeedit.executeAction("TYPE", mkPropertyValues({"KEYCODE":"CTRL+A"}))
        xvariable1rangeedit.executeAction("TYPE", mkPropertyValues({"KEYCODE":"BACKSPACE"}))
        if data_groupedby_column:
            xvariable1rangeedit.executeAction("TYPE", mkPropertyValues({"TEXT":"$DataInColumns.$A$1:$C$11"}))
        else:
            xvariable1rangeedit.executeAction("TYPE", mkPropertyValues({"TEXT":"$DataInRows.$A$1:$K$3"}))

        xvariable2rangeedit.executeAction("FOCUS", tuple()) # Without this the range parser does not kick in somehow
        xvariable2rangeedit.executeAction("TYPE", mkPropertyValues({"KEYCODE":"CTRL+A"}))
        xvariable2rangeedit.executeAction("TYPE", mkPropertyValues({"KEYCODE":"BACKSPACE"}))
        if data_groupedby_column:
            xvariable2rangeedit.executeAction("TYPE", mkPropertyValues({"TEXT":"$DataInColumns.$D$1:$D$11"}))
        else:
            xvariable2rangeedit.executeAction("TYPE", mkPropertyValues({"TEXT":"$DataInRows.$A$4:$K$4"}))
        # The data ranges have labels in them
        if (get_state_as_dict(xwithlabelscheck)["Selected"]) == "false":
            xwithlabelscheck.executeAction("CLICK", tuple())

        xoutputrangeedit.executeAction("FOCUS", tuple()) # Without this the range parser does not kick in somehow
        xoutputrangeedit.executeAction("TYPE", mkPropertyValues({"KEYCODE":"CTRL+A"}))
        xoutputrangeedit.executeAction("TYPE", mkPropertyValues({"KEYCODE":"BACKSPACE"}))
        if regression_type == "LINEAR":
            if calc_intercept:
                xoutputrangeedit.executeAction("TYPE", mkPropertyValues({"TEXT":"$ActualLinear.$A$1"}))
            else:
                xoutputrangeedit.executeAction("TYPE", mkPropertyValues({"TEXT":"$ActualLinearNoIntercept.$A$1"}))
        elif regression_type == "LOG":
            xoutputrangeedit.executeAction("TYPE", mkPropertyValues({"TEXT":"$ActualLog.$A$1"}))
        else:
            xoutputrangeedit.executeAction("TYPE", mkPropertyValues({"TEXT":"$ActualPower.$A$1"}))

        if data_groupedby_column:
            xgroupedbycolumnsradio.executeAction("CLICK", tuple())
        else:
            xgroupedbyrowsradio.executeAction("CLICK", tuple())

        if regression_type == "LINEAR":
            xlinearradio.executeAction("CLICK", tuple())
        elif regression_type == "LOG":
            xlogarithmicradio.executeAction("CLICK", tuple())
        else:
            xpowerradio.executeAction("CLICK", tuple())

        if not calc_intercept:
            xnointerceptcheck.executeAction("CLICK", tuple())

        xOKBtn = xDialog.getChild("ok")
        self.ui_test.close_dialog_through_button(xOKBtn)

    def test_regression_cancel(self):
        calc_doc = self.ui_test.create_doc_in_start_center("calc")
        self.ui_test.execute_modeless_dialog_through_command(".uno:RegressionDialog")
        xDialog = self.xUITest.getTopFocusWindow()
        xCancelBtn = xDialog.getChild("cancel")
        self.ui_test.close_dialog_through_button(xCancelBtn)

        self.ui_test.close_doc()

# vim: set shiftwidth=4 softtabstop=4 expandtab: