def test_reset1(self):
"Test the reset method."
x, y, z, v, s, src = self.get_data()
self.check_traits()
self.check_dataset()
# Call reset again with just a few things changed to see if it
# works correctly.
x, y, z = N.ogrid[-10:10:11j, -10:10:12j, -10:10:20j]
self.x, self.y, self.z = x, y, z
dims = (x.shape[0], y.shape[1], z.shape[2])
self.v = v = N.ones(dims + (3, ), float)
v[..., 2] = 2
v[..., 2] = 3
self.s = s = N.ones(dims, float)
src = sources.MArraySource()
src.reset(x=x,
y=y,
z=z,
u=v[..., 0],
v=v[..., 1],
w=v[..., 2],
scalars=s,
vectors=v)
self.check_traits()
self.check_dataset()
def setUp(self):
x, y, z = N.ogrid[-10:10:11j,
-10:10:12j,
-10:10:20j]
self.x, self.y, self.z = x, y, z
dims = (x.shape[0], y.shape[1], z.shape[2])
self.v = v = N.ones(dims + (3,), float)
v[...,2] = 2
v[...,2] = 3
self.s = s = N.ones(dims, float)
src = sources.MArraySource()
src.reset(x=x, y=y, z=z, u=v[...,0], v=v[...,1], w=v[...,2], scalars=s)
self.src = src
