def METHOD_NAME(
    self, num_points: int, key: KeyArray
) -> Float[Array, "N D"]:
    """
    Generate test points from the search space of the test function.
    Args:
        num_points (int): Number of points to sample.
        key (KeyArray): JAX PRNG key.
    Returns:
        Float[Array, 'N D']: Test points sampled from the search space.
    """
    return self.search_space.sample(num_points=num_points, key=key)