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)