def test_std_shape(random_process: randprocs.RandomProcess, args0: np.ndarray):
"""Test whether the standard deviation of the random process has the correct
shape."""
if random_process.output_dim == 1:
assert random_process.std(args0).ndim == 1
else:
assert random_process.std(args0).shape[1] == random_process.output_dim
def test_std_shape(random_process: randprocs.RandomProcess, args0: np.ndarray):
"""Test whether the standard deviation of the random process has the correct
shape."""
expected_shape = args0.shape[:-1] + random_process.output_shape
assert random_process.std(args0).shape == expected_shape