def test_mean_vertex_weights(self): tree, altitudes = TestAttributes.get_test_tree() leaf_data = np.asarray((0, 1, 2, 3, 4, 5, 6, 7, 8), dtype=np.float64) ref_attribute = np.asarray((0, 1, 2, 3, 4, 5, 6, 7, 8, 1. / 2, 7. / 2, 7. / 2, 13. / 2, 7., 2., 29. / 7, 4.)) attribute = hg.attribute_mean_vertex_weights(tree, vertex_weights=leaf_data) self.assertTrue(np.allclose(ref_attribute, attribute))
def test_mean_vertex_weights_vectorial(self): tree, altitudes = TestAttributes.get_test_tree() leaf_data = np.asarray(((0, 0), (1, 1), (2, 2), (3, 3), (4, 4), (5, 5), (6, 6), (7, 7), (8, 8)), dtype=np.float64) ref_attribute = np.asarray( ((0, 0), (1, 1), (2, 2), (3, 3), (4, 4), (5, 5), (6, 6), (7, 7), (8, 8), (1. / 2, 1. / 2), (7. / 2, 7. / 2), (7. / 2, 7. / 2), (13. / 2, 13. / 2), (7., 7.), (2., 2.), (29. / 7, 29. / 7), (4., 4.))) attribute = hg.attribute_mean_vertex_weights(tree, vertex_weights=leaf_data) self.assertTrue(np.allclose(ref_attribute, attribute))