def METHOD_NAME(random_process: randprocs.RandomProcess, args0: np.ndarray):
    """Test whether the mean of the random process has the correct shape."""
    expected_shape = args0.shape[:-1] + random_process.output_shape
    assert random_process.mean(args0).shape == expected_shape