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