def view_surface():
from enthought.mayavi import mlab
g = Grid(surface_param)
xyz_grid = g[-1:1:201j, -1:1:101j]
x, y, z = xyz_grid.transposed_values
mlab.mesh(x, y, z)
mlab.draw()
def test_grid():
g = Grid(surface_param)
xyz_grid = g[-1:1:201j, -1:1:101j]
x, y, z = xyz_grid.transposed_values
yield assert_equal, x.shape, (201, 101)
yield assert_equal, y.shape, (201, 101)
yield assert_equal, z.shape, (201, 101)
def test_grid():
input = CoordinateSystem('ij', 'input')
output = CoordinateSystem('xy', 'output')
def f(ij):
i = ij[:,0]
j = ij[:,1]
return np.array([i**2+j,j**3+i]).T
cmap = CoordinateMap(input, output, f)
grid = Grid(cmap)
eval = ArrayCoordMap.from_shape(cmap, (50,40))
assert_true(np.allclose(grid[0:50,0:40].values, eval.values))
def test_grid32():
# Check that we can use a float32 input and output
uv32 = CoordinateSystem('uv', 'input', np.float32)
xyz32 = CoordinateSystem('xyz', 'output', np.float32)
surface32 = CoordinateMap(uv32, xyz32, parametric_mapping)
g = Grid(surface32)
xyz_grid = g[-1:1:201j, -1:1:101j]
x, y, z = xyz_grid.transposed_values
yield assert_equal, x.shape, (201, 101)
yield assert_equal, y.shape, (201, 101)
yield assert_equal, z.shape, (201, 101)
yield assert_equal, x.dtype, np.dtype(np.float32)
def test_grid32_c128():
# Check that we can use a float32 input and complex128 output
uv32 = CoordinateSystem('uv', 'input', np.float32)
xyz128 = CoordinateSystem('xyz', 'output', np.complex128)
def par_c128(x):
return parametric_mapping(x).astype(np.complex128)
surface = CoordinateMap(uv32, xyz128, par_c128)
g = Grid(surface)
xyz_grid = g[-1:1:201j, -1:1:101j]
x, y, z = xyz_grid.transposed_values
yield assert_equal, x.shape, (201, 101)
yield assert_equal, y.shape, (201, 101)
yield assert_equal, z.shape, (201, 101)
yield assert_equal, x.dtype, np.dtype(np.complex128)
def test_eval_slice():
input = CoordinateSystem('ij', 'input')
output = CoordinateSystem('xy', 'output')
def f(ij):
i = ij[:,0]
j = ij[:,1]
return np.array([i**2+j,j**3+i]).T
cmap = CoordinateMap(input, output, f)
cmap = CoordinateMap(input, output, f)
grid = Grid(cmap)
e = grid[0:50,0:40]
ee = e[0:20:3]
yield assert_equal, ee.shape, (7,40)
yield assert_equal, ee.values.shape, (280,2)
yield assert_equal, ee.transposed_values.shape, (2,7,40)
ee = e[0:20:2,3]
yield assert_equal, ee.values.shape, (10,2)
yield assert_equal, ee.transposed_values.shape, (2,10)
yield assert_equal, ee.shape, (10,)