def test_rmse_1(self):
points = np.array([[0,0], [1,1], [2,2]])
knees = np.array([0,1,2])
expected = np.array([[1,1], [2,2]])
result = evaluation.rmse(points, knees, expected, evaluation.Strategy.worst)
desired = math.sqrt(1/3)
self.assertAlmostEqual(result, desired)
def test_rmse_2(self):
points = np.array([[0,0], [1,1], [2,2]])
knees = np.array([1])
expected = np.array([[1,1], [2,2]])
result = evaluation.rmse(points, knees, expected)
desired = math.sqrt(1/2)
self.assertAlmostEqual(result, desired)
def test_rmse_0(self):
points = np.array([[0,0], [1,1], [2,2]])
knees = np.array([0,1,2])
expected = np.array([[0,0], [1,1], [2,2]])
result = evaluation.rmse(points, knees, expected)
desired = 0.0
self.assertAlmostEqual(result, desired)