def test_centre2(self):
evecs = [[0, 0, 0], [-1, 0, 1], [1, 0, 1], [0, 4, 0]]
c1 = stratify.getCentroid(evecs, 0, [0, 1])
self.assertEqual(c1, -0.5)
def test_centre1(self):
evecs = [[0, 0, 0], [-1, 0, 1], [1, 0, 1], [0, 4, 0]]
c1 = stratify.getCentroid(evecs, 1, [0, 1, 2, 3])
self.assertEqual(c1, 1)
def test_centre1(self):
c1 = stratify.getCentroid(self.test_data[0], 1,
self.test_data[1]["KS3"])
self.assertAlmostEqual(c1, -0.014050000000)
