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))
