def test_create_in_bbox():
bbox = [0, 0, 1, 5]
shape = (2, 5)
g = Grid2D(bbox=bbox, shape=shape)
assert g.x0 == 0.25
g = Grid2D(bbox)
assert g.nx == 10
assert g.ny == 50
dx = 0.5
g = Grid2D(bbox, dx=dx)
assert g.dx == dx
assert g.dy == dx
assert g.n == 20
dxdy = (0.5, 2.5)
g = Grid2D(bbox, dx=dxdy)
assert g.dx == dxdy[0]
assert g.dy == dxdy[1]
assert g.n == 4
g = Grid2D(bbox, dx=dx, dy=2.5)
assert g.dx == dx
assert g.dy == 2.5
assert g.n == 4
def test_xy_to_bbox():
bbox = [0, 0, 1, 5]
g = Grid2D(bbox)
bbox2 = Grid2D.xy_to_bbox(g.xy)
assert np.all(bbox == bbox2)
bbox2 = Grid2D.xy_to_bbox(g.xy, buffer=0.2)
assert bbox2[0] == -0.2
assert bbox2[3] == 5.2
def test_invalid_grid_not_possible():
bbox = [0, 0, -1, 1] # x0 > x1
shape = (2, 2)
with pytest.raises(ValueError):
Grid2D(bbox=bbox, shape=shape)
bbox = [0, 0, 1, -1] # y0 > y1
shape = (2, 2)
with pytest.raises(ValueError):
Grid2D(bbox=bbox, shape=shape)
def test_xy_to_bbox():
bbox = [0, 0, 1, 5]
g = Grid2D(bbox)
xy = np.array([[0, 0], [0, 1], [1, 0], [1, 1], [0, 5], [1, 5]],
dtype=float)
bbox2 = Grid2D.xy_to_bbox(xy)
assert bbox[0] == bbox2[0]
assert bbox[-1] == bbox2[-1]
bbox2 = Grid2D.xy_to_bbox(xy, buffer=0.2)
assert bbox2[0] == -0.2
assert bbox2[3] == 5.2
def test_create_in_bbox():
bbox = [0, 0, 1, 5]
shape = (2, 5)
g = Grid2D(bbox=bbox, shape=shape)
assert g.x0 == 0.25
g = Grid2D(bbox=bbox, shape=(2, None))
assert g.x0 == 0.25
g = Grid2D(bbox)
assert g.nx == 10
assert g.ny == 50
dx = 0.5
g = Grid2D(bbox, dx=dx)
assert g.dx == dx
assert g.dy == dx
assert g.n == 20
dxdy = (0.5, 2.5)
g = Grid2D(bbox, dx=dxdy)
assert g.dx == dxdy[0]
assert g.dy == dxdy[1]
assert g.n == 4
g = Grid2D(bbox, dx=dx, dy=2.5)
assert g.dx == dx
assert g.dy == 2.5
assert g.n == 4
with pytest.raises(ValueError):
Grid2D(bbox, shape=(12, 2, 2))
with pytest.raises(ValueError):
Grid2D(bbox, shape=(None, None))
def test_x_y():
x0 = 2.0
x1 = 8.0
nx = 4
dx = 2.0
x = np.linspace(x0, x1, nx)
y0 = 3.0
y1 = 5
ny = 3
dy = 1.0
y = np.linspace(y0, y1, ny)
g = Grid2D(x, y)
assert g.x0 == x0
assert g.x1 == x1
assert g.y0 == y0
assert g.y1 == y1
assert np.all(g.x == x)
assert np.sum(g.y - y) == 0
assert g.nx == nx
assert g.ny == ny
assert g.dx == dx
assert g.dy == dy
# BoundingBox(left, bottom, right, top)
# Is this test good, or just a copy of the implementation?
assert g.bbox == ((x0 - dx / 2), (y0 - dy / 2), (x1 + dx / 2),
(y1 + dy / 2))
def test_find_index():
bbox = [0, 0, 1, 5]
g = Grid2D(bbox, dx=0.2)
xy1 = [0.52, 1.52]
xy2 = [1.5, 0.5]
i1, j1 = g.find_index(xy1)
assert i1 == 2
assert j1 == 7
i2, j2 = g.find_index(xy2)
assert i2 == -1
assert j2 == -1
xy = np.vstack([xy1, xy2])
ii, jj = g.find_index(xy)
assert ii[0] == i1
assert ii[1] == i2
assert jj[0] == j1
assert jj[1] == j2
xy = np.vstack([xy1, xy2, xy2])
ii, jj = g.find_index(x=xy[:, 0], y=xy[:, 1])
assert ii[0] == i1
assert jj[0] == j1
assert ii[2] == i2
assert jj[2] == j2
def test_to_mesh():
outfilename = "temp.mesh"
g = Grid2D([0, 0, 1, 5])
g.to_mesh(outfilename)
assert os.path.exists(outfilename)
mesh = Mesh(outfilename)
assert mesh.n_elements == g.n
os.remove(outfilename) # clean up
nc = g.get_node_coordinates()
new_z = nc[:, 1] - 10
g.to_mesh(outfilename, z=new_z)
assert os.path.exists(outfilename)
mesh = Mesh(outfilename)
assert mesh.node_coordinates[0, 2] == new_z[2]
os.remove(outfilename) # clean up
new_z = -10
g.to_mesh(outfilename, z=new_z)
assert os.path.exists(outfilename)
mesh = Mesh(outfilename)
assert mesh.node_coordinates[0, 2] == -10
os.remove(outfilename) # clean up
def test_xx_yy():
nx = 4
ny = 3
x = np.linspace(1, 7, nx)
y = np.linspace(3, 5, ny)
g = Grid2D(x, y)
assert g.n == nx * ny
assert g.xx[0, 0] == 1.0
assert g.yy[-1, -1] == 5.0
assert np.all(g.xy[1] == [3.0, 3.0])
assert np.all(g.coordinates[1] == [3.0, 3.0])
g2 = Grid2D(x, y)
# Reverse order compared to above makes no difference
assert g2.yy[-1, -1] == 5.0
assert g2.xx[0, 0] == 1.0
def test_xx_yy():
nx = 4
ny = 3
x = np.linspace(1, 7, nx)
y = np.linspace(3, 5, ny)
g = Grid2D(x, y)
assert g.n == nx * ny
assert g.xx[0, 0] == 1.0
assert g.yy[-1, -1] == 5.0
assert np.all(g.xy[1] == [3.0, 3.0])
def test_to_mesh():
outfilename = "temp.mesh"
g = Grid2D([0, 0, 1, 5])
g.to_mesh(outfilename)
assert os.path.exists(outfilename)
mesh = Mesh(outfilename)
assert True
os.remove(outfilename) # clean up
def test_create_in_bbox():
bbox = [0, 0, 1, 5]
shape = (3, 6)
g = Grid2D(bbox=bbox, shape=shape)
assert g.x0 == 0.0
g = Grid2D(bbox)
assert g.nx == 10
assert g.ny == 50
dx = 1.0
g = Grid2D(bbox, dxdy=dx)
assert g.dx == dx
assert g.dy == dx
assert g.n == 12
dxdy = (0.5, 2.5)
g = Grid2D(bbox, dxdy=dxdy)
assert g.dx == dxdy[0]
assert g.dy == dxdy[1]
assert g.n == 9
def test_x_y():
x0 = 2.0
x1 = 8.0
nx = 4
dx = 2.0
x = np.linspace(x0, x1, nx)
y0 = 3.0
y1 = 5
ny = 3
dy = 1.0
y = np.linspace(y0, y1, ny)
g = Grid2D(x, y)
assert g.x0 == x0
assert g.x1 == x1
assert g.y0 == y0
assert g.y1 == y1
assert np.all(g.x == x)
assert np.sum(g.y - y) == 0
assert g.nx == nx
assert g.ny == ny
assert g.dx == dx
assert g.dy == dy
assert np.all(g.bbox == [x0, y0, x1, y1])
def test_no_parameters():
with pytest.raises(ValueError):
Grid2D()