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)