def test_meshgrid(as_midp):
vec1 = (0, 1)
vec2 = (-1, 0, 1)
vec3 = (2, 3, 4, 5)
# Sparse meshgrid
mgx = np.array(vec1)[:, np.newaxis, np.newaxis]
mgy = np.array(vec2)[np.newaxis, :, np.newaxis]
mgz = np.array(vec3)[np.newaxis, np.newaxis, :]
grid = TensorGrid(vec1, vec2, vec3, as_midp=as_midp)
xx, yy, zz = grid.meshgrid()
assert all_equal(mgx, xx)
assert all_equal(mgy, yy)
assert all_equal(mgz, zz)
xx, yy, zz = grid.meshgrid()
assert all_equal(mgx, xx)
assert all_equal(mgy, yy)
assert all_equal(mgz, zz)
# Fortran ordering
grid = TensorGrid(vec1, vec2, vec3, order='F', as_midp=as_midp)
xx, yy, zz = grid.meshgrid()
assert all(arr.flags.f_contiguous for arr in (xx, yy, zz))
