def test_load_grid(): gr = Grid() filename = str(DATA_DIR / Path('test_create_grid.vtk')) gr.load(filename) ncells = gr.getNumberOfCells() print(f'ncells = {ncells}') assert (ncells == 2) num_bad_cells = gr.check() assert (num_bad_cells == 0) points = gr.getPoints() assert (points.shape[0] == ncells) assert (points.shape[1] == 4) assert (points.shape[2] == 3)
def test_load_from_ugrid_file(): gr = Grid() gr.setFlags(1, 1) filename = str(DATA_DIR / Path('cs_4.nc')) gr.loadFromUgrid2DFile(f'{filename}$physics') nedges = gr.getNumberOfEdges() print(f'nedges = {nedges}') assert (nedges == 192) ncells = gr.getNumberOfCells() for icell in range(ncells): for iedge in range(4): edgeId, edgeSign = gr.getEdgeId(icell, iedge) nodeIds = gr.getNodeIds(icell, iedge) print(f"cell {icell} edge {iedge}: " + f"edgeId = {edgeId}, {edgeSign} nodeIds = {nodeIds}") # attaching a 3 components field to the grid data = numpy.array(range(ncells * 4 * 3), numpy.float64) gr.attach('myData', data) num_bad_cells = gr.check() assert (num_bad_cells == 0)