def test_attribute_mixing_matrix_multigraph(self): mapping = {'one': 0, 'two': 1, 'red': 2, 'blue': 3} a_result = np.array([[4, 0, 0, 0], [0, 2, 0, 0], [0, 0, 0, 0], [0, 0, 0, 0]]) a = mixing.attribute_mixing_matrix(self.M, 'fish', mapping=mapping, normalized=False) npt.assert_equal(a, a_result) a = mixing.attribute_mixing_matrix(self.M, 'fish', mapping=mapping) npt.assert_equal(a, a_result / float(a_result.sum()))
def test_attribute_mixing_matrix_multigraph(self): mapping={'one':0,'two':1,'red':2,'blue':3} a_result=np.array([[4,0,0,0], [0,2,0,0], [0,0,0,0], [0,0,0,0]] ) a=mixing.attribute_mixing_matrix(self.M,'fish', mapping=mapping, normalized=False) npt.assert_equal(a,a_result) a=mixing.attribute_mixing_matrix(self.M,'fish', mapping=mapping) npt.assert_equal(a,a_result/float(a_result.sum()))