def testPartitionGraph(self): data = MEDLoaderDataForTest.buildACompleteMEDDataStructureWithFieldsOnCells_1( ) m = data.getMeshes().getMeshAtPos(0) graph = MEDPartitioner.Graph(m.getLevel0Mesh().generateGraph()) graph.partGraph(2) tool = MEDPartitioner(data, graph) data2 = tool.getMEDFileData() self.assertEqual(2, data2.getMeshes().getNumberOfMeshes())
def testPartition(self): fname = "PyPartitionTest.med" data = MEDLoaderDataForTest.buildACompleteMEDDataStructureWithFieldsOnCells_1( ) data.write(fname, 2) part_file = MEDPartitioner(fname, 2) part_data = MEDPartitioner(data, 2) part_file.write("splitted_PyPartitionTest_1") part_data.write("splitted_PyPartitionTest_2") part_file_xml = MEDPartitioner("splitted_PyPartitionTest_1.xml") part_data_xml = MEDPartitioner("splitted_PyPartitionTest_2.xml") data1 = part_file_xml.getMEDFileData() data2 = part_data_xml.getMEDFileData() m1d = data1.getMeshes().getMeshAtPos(0) m2d = data2.getMeshes().getMeshAtPos(0) self.assertTrue(m1d.isEqual(m2d, 1e-12))