def testConsistency(type,nonames=False): tempfile = cStringIO.StringIO() nodeNames=netio.writeNet_mat(self.simpleWeightedNet,tempfile,type=type) if nonames: nodeNames=[] string1=tempfile.getvalue() tempfile.seek(0) tempnet=netio.loadNet_mat(tempfile,nodeNames=nodeNames,type=type) tempfile = cStringIO.StringIO() netio.writeNet_mat(tempnet,tempfile,type=type) self.assertEqual(string1,tempfile.getvalue())
def test_writeNet_mat(self): tempfile = cStringIO.StringIO() #virtual file netio.writeNet_mat(self.simpleWeightedNet,tempfile,type="square") self.assertEqual(tempfile.getvalue(),"0\t1.0\t2.0\t3.0\n1.0\t0\t4.0\t0.0\n2.0\t4.0\t0\t0.0\n3.0\t0.0\t0.0\t0\n") tempfile = cStringIO.StringIO() netio.writeNet_mat(self.simpleWeightedNet,tempfile,type="upperdiag") self.assertEqual(tempfile.getvalue(),"0\t1.0\t2.0\t3.0\n\t0\t4.0\t0.0\n\t\t0\t0.0\n\t\t\t0\n") tempfile = cStringIO.StringIO() netio.writeNet_mat(self.simpleWeightedNet,tempfile,type="supperdiag") self.assertEqual(tempfile.getvalue(),"1.0\t2.0\t3.0\n\t4.0\t0.0\n\t\t0.0\n") tempfile = cStringIO.StringIO() netio.writeNet_mat(self.simpleWeightedNet,tempfile,type="lowerdiag") self.assertEqual(tempfile.getvalue(),"0\n1.0\t0\n2.0\t4.0\t0\n3.0\t0.0\t0.0\t0\n") tempfile = cStringIO.StringIO() netio.writeNet_mat(self.simpleWeightedNet,tempfile,type="slowerdiag") self.assertEqual(tempfile.getvalue(),"1.0\n2.0\t4.0\n3.0\t0.0\t0.0\n")