func.func @pad_1D_test() {
    %0 = util.unfoldable_constant dense<42> : tensor<2xi32>
    %1 = tosa.const_shape { values = dense<[3, 2]> : tensor<2xindex> } : ()  -> !tosa.shape<2>
    %cst = "tosa.const"() { values = dense<0> : tensor<1xi32> } : () -> tensor<1xi32>
    %result = tosa.pad %0, %1, %cst : (tensor<2xi32>, !tosa.shape<2>, tensor<1xi32>) -> (tensor<7xi32>)
    check.expect_eq_const(%result, dense<[0, 0, 0, 42, 42, 0, 0]> : tensor<7xi32>) : tensor<7xi32>
    return
}

func.func @pad_2D_test() {
    %0 = util.unfoldable_constant dense<42> : tensor<2x2xi32>
    %1 = tosa.const_shape { values = dense<[1, 1, 1, 1]> : tensor<4xindex> } : ()  -> !tosa.shape<4>
    %cst = "tosa.const"() { values = dense<0> : tensor<1xi32> } : () -> tensor<1xi32>
    %result = tosa.pad %0, %1, %cst : (tensor<2x2xi32>, !tosa.shape<4>, tensor<1xi32>) -> (tensor<4x4xi32>)
    check.expect_eq_const(%result, dense<[[0, 0, 0, 0], [0, 42, 42, 0], [0, 42, 42, 0], [0, 0, 0, 0]]> : tensor<4x4xi32>) : tensor<4x4xi32>
    return
}

func.func @pad_3D_test() {
    %0 = util.unfoldable_constant dense<42> : tensor<1x1x2xi32>
    %1 = tosa.const_shape { values = dense<[0, 1, 1, 0, 0, 0]> : tensor<6xindex> } : ()  -> !tosa.shape<6>
    %cst = "tosa.const"() { values = dense<0> : tensor<1xi32> } : () -> tensor<1xi32>
    %result = tosa.pad %0, %1, %cst : (tensor<1x1x2xi32>, !tosa.shape<6>, tensor<1xi32>) -> (tensor<2x2x2xi32>)
    check.expect_eq_const(%result, dense<[[[0, 0], [42, 42]], [[0, 0], [0, 0]]]> : tensor<2x2x2xi32>) : tensor<2x2x2xi32>
    return
}