def METHOD_NAME(
    model_setup: ModelSetup,
    input_dim: int,
    output_dim: int,
    N: int,
    Ntest: int,
    M: int,
    num_samples: int,
) -> None:
    samples_shape = (num_samples, Ntest, output_dim)
    X, Y, Z = rng.randn(N, input_dim), rng.randn(N, output_dim), rng.randn(M, input_dim)
    Xtest = rng.randn(Ntest, input_dim)
    model_gp = model_setup.get_model(Z, num_latent_gps=output_dim, data=(X, Y))
    samples = model_gp.predict_f_samples(Xtest, num_samples)
    assert samples.shape == samples_shape