def METHOD_NAME(variant_scalar_rgb):
    assert(dr.allclose(mi.warp.square_to_cosine_hemisphere([0.5, 0.5]), [0,  0,  1]))
    assert(dr.allclose(mi.warp.square_to_cosine_hemisphere([0.5,   0]), [0, -1, 0], atol=1e-7))
    check_inverse(mi.warp.square_to_cosine_hemisphere, mi.warp.cosine_hemisphere_to_square)
    check_warp_vectorization("square_to_cosine_hemisphere", atol=1e-5)