c33a9f9aa2
changed the behavior to import division and print to be consistent across all app and similar with python 3.x
75 lines
2.6 KiB
Python
75 lines
2.6 KiB
Python
# coding=utf-8
|
|
from __future__ import absolute_import, division, print_function
|
|
from octoprint.printer.estimation import TimeEstimationHelper
|
|
|
|
__author__ = "Gina Häußge <osd@foosel.net>"
|
|
__license__ = 'GNU Affero General Public License http://www.gnu.org/licenses/agpl.html'
|
|
__copyright__ = "Copyright (C) 2014 The OctoPrint Project - Released under terms of the AGPLv3 License"
|
|
|
|
|
|
import unittest
|
|
from ddt import ddt, data, unpack
|
|
|
|
import octoprint.printer
|
|
|
|
@ddt
|
|
class EstimationTestCase(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
self.estimation_helper = type(TimeEstimationHelper)(TimeEstimationHelper.__name__, (TimeEstimationHelper,), {
|
|
'STABLE_THRESHOLD': 0.1,
|
|
'STABLE_ROLLING_WINDOW': 3,
|
|
'STABLE_COUNTDOWN': 1
|
|
})()
|
|
|
|
@data(
|
|
((1.0, 2.0, 3.0, 4.0, 5.0), 3.0),
|
|
((1.0, 2.0, 0.0, 1.0, 2.0), 1.2),
|
|
((1.0, -2.0, -1.0, -2.0, 3.0), -0.2)
|
|
)
|
|
@unpack
|
|
def test_average_total(self, estimates, expected):
|
|
for estimate in estimates:
|
|
self.estimation_helper.update(estimate)
|
|
|
|
self.assertEquals(self.estimation_helper.average_total, expected)
|
|
|
|
@data(
|
|
((1.0, 2.0), None), # not enough values, have 1, need 3
|
|
((1.0, 2.0, 3.0), None), # not enough values, have 2, need 3
|
|
((1.0, 2.0, 3.0, 4.0), 0.5), # average totals: 1.0, 1.5, 2.0, 2.5 => (3 * 0.5 / 3 = 0.5
|
|
((1.0, 2.0, 3.0, 4.0, 5.0), 0.5), # average totals: 1.0, 1.5, 2.0, 2.5, 3.0 => (0.5 + 0.5 + 0.5) / 3 = 0.5
|
|
((1.0, 2.0, 0.0, 1.0, 2.0), 0.7 / 3) # average totals: 1.0, 1.5, 1.0, 1.0, 1.2 => (0.5 + 0.0 + 0.2) / 3 = 0.7 / 3
|
|
)
|
|
@unpack
|
|
def test_average_distance(self, estimates, expected):
|
|
for estimate in estimates:
|
|
self.estimation_helper.update(estimate)
|
|
|
|
self.assertEquals(self.estimation_helper.average_distance, expected)
|
|
|
|
@data(
|
|
((1.0, 1.0), None),
|
|
((1.0, 1.0, 1.0), 1.0),
|
|
((1.0, 2.0, 3.0, 4.0, 5.0), 4.0),
|
|
)
|
|
@unpack
|
|
def test_average_total_rolling(self, estimates, expected):
|
|
for estimate in estimates:
|
|
self.estimation_helper.update(estimate)
|
|
|
|
self.assertEquals(self.estimation_helper.average_total_rolling, expected)
|
|
|
|
@data(
|
|
((1.0, 1.0, 1.0, 1.0), False), # average totals: 1.0, 1.0, 1.0, 1.0 => 3.0 / 3 = 1.0
|
|
((1.0, 1.0, 1.0, 1.0, 1.0), True), # average totals: 1.0, 1.0, 1.0, 1.0, 1.0 => 0.0 / 3 = 0.0
|
|
((1.0, 2.0, 3.0, 4.0, 5.0), False), # average totals: 1.0, 1.5, 2.0, 2.5, 3.0 => 1.5 / 3 = 0.5
|
|
((0.0, 0.09, 0.18, 0.27, 0.36), True) # average totals: 0.0, 0.045, 0.09, 0.135, 0.18 => (0.045 + 0.045 + 0.045) / 3 = 0.045
|
|
)
|
|
@unpack
|
|
def test_is_stable(self, estimates, expected):
|
|
for estimate in estimates:
|
|
self.estimation_helper.update(estimate)
|
|
|
|
self.assertEquals(self.estimation_helper.is_stable(), expected)
|
|
|