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