def test_linear_regression_no_data(self):
m = CorTiltDataModel()
with self.assertRaises(ValueError):
m.linear_regression()
self.assertFalse(m.has_results)
def test_linear_regression_data(self):
m = CorTiltDataModel()
m._points = [
Point(1409, 1401),
Point(1386, 1400),
Point(1363, 1400),
Point(1340, 1401),
Point(1317, 1400),
Point(1294, 1399),
Point(1271, 1398),
Point(1248, 1400),
Point(1225, 1398),
Point(1202, 1400),
Point(1179, 1399),
Point(1156, 1399),
Point(1133, 1400),
Point(1110, 1402),
Point(1087, 1398),
Point(1064, 1398),
Point(1041, 1397),
Point(1018, 1399),
Point(995, 1398),
Point(972, 1401),
Point(949, 1397),
Point(926, 1398),
Point(903, 1398),
Point(880, 1398),
Point(857, 1396),
Point(834, 1397),
]
m.linear_regression()
self.assertAlmostEqual(m.gradient.value, 0.005292, places=6)
self.assertAlmostEqual(m.cor.value, 1392.99, places=2)
def test_linear_regression(self):
m = CorTiltDataModel()
m.add_point(None, 0, 5.0)
m.add_point(None, 10, 6.0)
m.add_point(None, 20, 7.0)
m.add_point(None, 30, 8.0)
m.linear_regression()
self.assertTrue(m.has_results)
self.assertAlmostEqual(m.gradient.value, 0.1)
self.assertAlmostEqual(m.cor.value, 5.0)
def test_clear_results(self):
m = CorTiltDataModel()
m.add_point(None, 0, 5.0)
m.add_point(None, 10, 6.0)
m.add_point(None, 20, 7.0)
m.add_point(None, 30, 8.0)
m.linear_regression()
m.clear_results()
self.assertFalse(m.has_results)
def test_get_cor_for_slice_from_regression(self):
m = CorTiltDataModel()
m.add_point(None, 10, 5.0)
m.add_point(None, 20, 6.0)
m.add_point(None, 30, 7.0)
m.add_point(None, 40, 8.0)
m.linear_regression()
self.assertEqual(m.get_cor_from_regression(0), 4.0)
self.assertEqual(m.get_cor_from_regression(10), 5.0)
self.assertEqual(m.get_cor_from_regression(50), 9.0)
def test_stack_properties(self):
m = CorTiltDataModel()
m.add_point(None, 0, 5.0)
m.add_point(None, 10, 6.0)
m.add_point(None, 20, 7.0)
m.add_point(None, 30, 8.0)
m.linear_regression()
d = m.stack_properties
self.assertEqual(len(d), 5)
self.assertEqual(d[const.COR_TILT_FITTED_GRADIENT], m.gradient.value)
self.assertEqual(d[const.COR_TILT_ROTATION_CENTRE], m.cor.value)
self.assertEqual(d[const.COR_TILT_SLICE_INDICES], m.slices)
self.assertEqual(d[const.COR_TILT_ROTATION_CENTRES], m.cors)
self.assertTrue(isinstance(d[const.COR_TILT_TILT_ANGLE_DEG], float))
def test_data_ordering(self):
m = CorTiltDataModel()
m.add_point(None, 30, 7.0)
m.add_point(None, 10, 5.0)
m.add_point(None, 40, 8.0)
m.add_point(None, 20, 6.0)
m.sort_points()
# Expect data to be sorted as it is inserted
self.assertEqual(m.slices, [10, 20, 30, 40])
self.assertEqual(m.cors, [5.0, 6.0, 7.0, 8.0])
m.linear_regression()
self.assertTrue(m.has_results)
self.assertAlmostEqual(m.gradient.value, 0.1)
self.assertAlmostEqual(m.cor.value, 4.0)