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_create_grid():
# create the grid
gr = Grid()
# 2 cells
points = numpy.array([(0., 0., 0.), (1., 0., 0.), (1., 1., 0.),
(0., 1., 0.), (1., 0., 0.), (2., 0., 0.),
(2., 1., 0.), (1., 1., 0.)]).reshape((2, 4, 3))
gr.setPoints(points)
# get a pointer to the points of the cell-by-cell mesh
pts = gr.getPoints()
print(pts)
with TemporaryDirectory() as d:
fname = str(Path(d) / Path('grid.vtk'))
gr.dump(fname)