diff options
author | Tim Rault <tim.rault@cengn.ca> | 2016-06-29 10:54:55 -0400 |
---|---|---|
committer | Tim Rault <tim.rault@cengn.ca> | 2016-07-06 10:26:51 -0400 |
commit | d7d5efb6fe647e117f54ce0923e4966014eaeb9a (patch) | |
tree | 2d5d7cc03be043d2f7aacf0b5b4308e76d45b418 /storperf/tests/utilities/math.py | |
parent | b3e8e5113c4bfb467f5257b20047feb2a0d73b4b (diff) |
Add Least Squares slope function
Added a math module in utilities that contains a slope function
able to compute the slope of the best Least Squares curve fit
given a series of [x,y] values.
Implemented a test harness for this math module in the tests/utilities
section.
Change-Id: If4d63af092d0904b2269c5ee0991e18ab84533c0
JIRA: STORPERF-54
JIRA: STORPERF-55
JIRA: STORPERF-51
Signed-off-by: Tim Rault <tim.rault@cengn.ca>
Diffstat (limited to 'storperf/tests/utilities/math.py')
-rw-r--r-- | storperf/tests/utilities/math.py | 66 |
1 files changed, 66 insertions, 0 deletions
diff --git a/storperf/tests/utilities/math.py b/storperf/tests/utilities/math.py new file mode 100644 index 0000000..c78538d --- /dev/null +++ b/storperf/tests/utilities/math.py @@ -0,0 +1,66 @@ +############################################################################## +# Copyright (c) 2016 CENGN and others. +# +# All rights reserved. This program and the accompanying materials +# are made available under the terms of the Apache License, Version 2.0 +# which accompanies this distribution, and is available at +# http://www.apache.org/licenses/LICENSE-2.0 +############################################################################## +import unittest +from storperf.utilities.math import math + +class MathTest(unittest.TestCase): + + def setUp(self): + unittest.TestCase.setUp(self) + pass + + def test_slope_empty_series(self): + expected = 0 + actual = math.slope([]) + self.assertEqual(expected, actual) + + def test_slope_integer_series(self): + expected = 1.4 + actual = math.slope([[1,6], [2,5], [3,7], [4,10]]) + self.assertEqual(expected, actual) + + def test_slope_decimal_series(self): + expected = 1.4 + actual = math.slope([[1.0,6.0], [2.0,5.0], [3.0,7.0], [4.0,10.0]]) + self.assertEqual(expected, actual) + + def test_slope_decimal_integer_mix(self): + expected = 1.4 + actual = math.slope([[1.0,6], [2,5.0], [3,7], [4.0,10]]) + self.assertEqual(expected, actual) + + def test_slope_negative_y_series(self): + expected = 2 + actual = math.slope([[1.0,-2], [2,2], [3,2]]) + self.assertEqual(expected, actual) + + def test_slope_negative_x_series(self): + expected = 1.4 + actual = math.slope([[-24,6.0], [-23,5], [-22,7.0], [-21,10]]) + self.assertEqual(expected, actual) + + def test_slope_out_of_order_series(self): + expected = 1.4 + actual = math.slope([[2,5.0], [4,10], [3.0,7], [1,6]]) + self.assertEqual(expected, actual) + + def test_slope_0_in_y(self): + expected = -0.5 + actual = math.slope([[15.5,1], [16.5,0], [17.5,0]]) + self.assertEqual(expected, actual) + + def test_slope_0_in_x(self): + expected = 1.4 + actual = math.slope([[0,6.0], [1,5], [2,7], [3,10]]) + self.assertEqual(expected, actual) + + def test_slope_0_in_x_and_y(self): + expected = 1.5 + actual = math.slope([[0.0,0], [1,1], [2,3]]) + self.assertEqual(expected, actual) |