42 lines
1.2 KiB
Python
42 lines
1.2 KiB
Python
# coding=utf-8
|
|
from __future__ import absolute_import
|
|
|
|
__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 = octoprint.printer.TimeEstimationHelper()
|
|
|
|
@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, 3.0, 4.0, 5.0), 0.5), # average totals: 1.0, 1.5, 2.0, 2.5, 3.0
|
|
((1.0, 2.0, 0.0, 1.0, 2.0), 0.3) # average totals: 1.0, 1.5, 1.0, 1.0, 1.2
|
|
)
|
|
@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)
|
|
|