//
// Copyright © 2020, 2023-2024 Arm Ltd and Contributors. All rights reserved.
// SPDX-License-Identifier: MIT
//

#include "TransposeTestHelper.hpp"

#include <doctest/doctest.h>

namespace armnnDelegate
{

void TransposeFP32Test()
{
    // set test input data
    std::vector<int32_t> input0Shape {4, 2, 3};
    std::vector<int32_t> inputPermVecShape {3};
    std::vector<int32_t> outputShape {3, 4, 2};

    std::vector<float> input0Values = {0,  1,  2,  3,  4,  5,  6,  7,  8,  9,  10, 11,
                                       12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23};
    std::vector<int32_t> inputPermVec = {2, 0, 1};
    std::vector<float> expectedOutputValues = {0, 3, 6, 9, 12, 15, 18, 21, 1, 4, 7, 10,
                                               13, 16, 19, 22, 2, 5, 8, 11, 14, 17, 20, 23};

    TransposeTest<float>(input0Shape,
                         inputPermVecShape,
                         outputShape,
                         input0Values,
                         inputPermVec,
                         expectedOutputValues);
}

TEST_SUITE ("TransposeTests")
{

TEST_CASE ("Transpose_Float32_Test")
{
    TransposeFP32Test();
}

}

} // namespace armnnDelegate