def METHOD_NAME(variant_scalar_rgb):
    assert(dr.allclose(mi.warp.square_to_std_normal([0, 0]), [0, 0]))
    assert(dr.allclose(mi.warp.square_to_std_normal([0, 1]), [0, 0]))
    assert(dr.allclose(mi.warp.square_to_std_normal([0.39346, 0]), [1, 0], atol=1e-3))