def test_input_args(self): """ Check that scalar_scatter can take different input arguments """ # Check for a single number as position vectors. ss = sources.scalar_scatter(0, 0, 0, figure=None) self.check_positions(ss, 0, 0, 0) self.check_scalars(ss, None) self.check_vectors(ss, None, None, None) # Check for a single number as scalar data, and no position # vectors. ss = sources.scalar_scatter(0, figure=None) self.check_positions(ss, 0, 0, 0) self.check_scalars(ss, 0) self.check_vectors(ss, None, None, None) # Check for a list as position vectors. ss = sources.scalar_scatter([0, 1], [0, 1], [0, 1], figure=None) self.check_positions(ss, [0, 1], [0, 1], [0, 1]) self.check_scalars(ss, None) self.check_vectors(ss, None, None, None) # Check for a list as scalar data, and no position vectors. ss = sources.scalar_scatter([0, 1], figure=None) self.check_scalars(ss, [0, 1]) self.check_vectors(ss, None, None, None) # Check for a 1D array as position vectors. a = np.array([0, 1]) ss = sources.scalar_scatter(a, a, a, figure=None) self.check_positions(ss, a, a, a) self.check_scalars(ss, None) self.check_vectors(ss, None, None, None) # Check for a 1D array as a scalar data, and no position vectors. ss = sources.scalar_scatter(a, figure=None) self.check_scalars(ss, a) self.check_vectors(ss, None, None, None) # Check for a 2D array as position vectors. a = np.array([[0, 1], [2, 3]]) ss = sources.scalar_scatter(a, a, a, figure=None) self.check_positions(ss, a, a, a) self.check_scalars(ss, None) self.check_vectors(ss, None, None, None) # Check for a 2D array as scalar data, and no position vectors. ss = sources.scalar_scatter(a, figure=None) self.check_scalars(ss, a) self.check_vectors(ss, None, None, None) # Check for a 2D array as scalar data, and no position vectors. ss = sources.scalar_scatter(a, figure=None) self.check_scalars(ss, a) self.check_vectors(ss, None, None, None)
def test_input_args(self): """ Check that vector_scatter can take different input arguments """ # Check for a single number as a position vector. ss = sources.vector_scatter(0, 0, 0, 0, 0, 0, figure=None) self.check_positions(ss, 0, 0, 0) self.check_scalars(ss, None) self.check_vectors(ss, 0, 0, 0) # Check for no position vectors, and single numbers for vector # data. ss = sources.vector_scatter(0, 0, 0, figure=None) self.check_positions(ss, 0, 0, 0) self.check_scalars(ss, None) self.check_vectors(ss, 0, 0, 0) # Check for a list as a position vector. ss = sources.vector_scatter([0, 1], [0, 1], [0, 1], [0, 1], [0, 1], [0, 1], figure=None) self.check_positions(ss, [0, 1], [0, 1], [0, 1]) self.check_scalars(ss, None) self.check_vectors(ss, [0, 1], [0, 1], [0, 1]) # Check for a lists as a vector data, and no position vectors ss = sources.vector_scatter([0, 1], [0, 1], [0, 1], figure=None) self.check_scalars(ss, None) self.check_vectors(ss, [0, 1], [0, 1], [0, 1]) # Check for a 1D array as a position vector. a = np.array([0, 1]) ss = sources.vector_scatter(a, a, a, a, a, a, figure=None) self.check_positions(ss, a, a, a) self.check_scalars(ss, None) self.check_vectors(ss, a, a, a) # Check for a 1D array as vector data, and no position vectors. ss = sources.vector_scatter(a, a, a, figure=None) self.check_scalars(ss, None) self.check_vectors(ss, a, a, a) # Check for a 2D array as a position vector. a = np.array([[0, 1], [2, 3]]) ss = sources.vector_scatter(a, a, a, a, a, a, figure=None) self.check_positions(ss, a, a, a) self.check_scalars(ss, None) self.check_vectors(ss, a, a, a) # Check for a 2D array as vector data, and no position vectors. ss = sources.vector_scatter(a, a, a, figure=None) self.check_scalars(ss, None) self.check_vectors(ss, a, a, a) # Check for a 3D array as a position vector. x, y, z = np.mgrid[0:3, 0:3, 0:3] ss = sources.vector_scatter(x, y, z, x, y, z, figure=None) self.check_positions(ss, x, y, z) self.check_scalars(ss, None) self.check_vectors(ss, x, y, z) # Check for a 3D array as vector data, and no position vectors. x, y, z = np.mgrid[0:3, 0:3, 0:3] ss = sources.scalar_scatter(z, figure=None) self.check_scalars(ss, z) X, Y, Z = np.indices(z.shape) self.check_positions(ss, X, Y, Z)