//
// Copyright © 2017 Arm Ltd. All rights reserved.
// SPDX-License-Identifier: MIT
//

#include <boost/test/unit_test.hpp>
#include "armnnTfParser/ITfParser.hpp"
#include "ParserPrototxtFixture.hpp"

BOOST_AUTO_TEST_SUITE(TensorflowParser)

struct ResizeBilinearFixture : public armnnUtils::ParserPrototxtFixture<armnnTfParser::ITfParser>
{
    ResizeBilinearFixture()
    {
        m_Prototext = R"(
node {
  name: "graphInput"
  op: "Placeholder"
  attr {
    key: "dtype"
    value {
      type: DT_FLOAT
    }
  }
  attr {
    key: "value"
    value {
      tensor {
        dtype: DT_FLOAT
        tensor_shape {
          dim {
            size: 1
          }
          dim {
            size: 3
          }
          dim {
            size: 3
          }
          dim {
            size: 1
          }
        }
        tensor_content:
"\000\000\000\000\000\000\200?\000\000\000@\000\000@@\000\000\200@\000\000\240@\000\000\300@\000\000\340@\000\000\000A"
      }
    }
  }
}
node {
  name: "resizeBilinearLayer/size"
  op: "Const"
  attr {
    key: "dtype"
    value {
      type: DT_INT32
    }
  }
  attr {
    key: "value"
    value {
      tensor {
        dtype: DT_INT32
        tensor_shape {
          dim {
            size: 2
          }
        }
        tensor_content: "\005\000\000\000\005\000\000\000"
      }
    }
  }
}
node {
  name: "resizeBilinearLayer"
  op: "ResizeBilinear"
  input: "graphInput"
  input: "resizeBilinearLayer/size"
  attr {
    key: "T"
    value {
      type: DT_FLOAT
    }
  }
  attr {
    key: "align_corners"
    value {
      b: false
    }
  }
}
        )";

        SetupSingleInputSingleOutput({ 1, 3, 3, 1 }, "graphInput", "resizeBilinearLayer");
    }
};

BOOST_FIXTURE_TEST_CASE(ParseResizeBilinear, ResizeBilinearFixture)
{
    RunTest<4>(// Input data.
               { 0.0f, 1.0f, 2.0f,
                 3.0f, 4.0f, 5.0f,
                 6.0f, 7.0f, 8.0f },
               // Expected output data.
               { 0.0f, 0.6f, 1.2f, 1.8f, 2.0f,
                 1.8f, 2.4f, 3.0f, 3.6f, 3.8f,
                 3.6f, 4.2f, 4.8f, 5.4f, 5.6f,
                 5.4f, 6.0f, 6.6f, 7.2f, 7.4f,
                 6.0f, 6.6f, 7.2f, 7.8f, 8.0f });

}

BOOST_AUTO_TEST_SUITE_END()