diff options
author | Kevin May <kevin.may@arm.com> | 2021-02-03 17:38:41 +0000 |
---|---|---|
committer | Francis Murtagh <francis.murtagh@arm.com> | 2021-02-04 11:23:41 +0000 |
commit | 7d96b16acfdbdcef6739d59ba067a37c062aa03f (patch) | |
tree | bb0a93f9030e26568d6d7c44776993e433996c31 /src/armnnTfLiteParser/test/GetBuffer.cpp | |
parent | d0bb8eafef2a93356e435ccc4029d487a2cde9e4 (diff) | |
download | armnn-7d96b16acfdbdcef6739d59ba067a37c062aa03f.tar.gz |
IVGCVSW-5592 Implement Pimpl Idiom for Tf and TfLite Parsers
Signed-off-by: Kevin May <kevin.may@arm.com>
Change-Id: I4a82aca4a2c47b3c598b91bc0075c09397be728a
Diffstat (limited to 'src/armnnTfLiteParser/test/GetBuffer.cpp')
-rw-r--r-- | src/armnnTfLiteParser/test/GetBuffer.cpp | 25 |
1 files changed, 14 insertions, 11 deletions
diff --git a/src/armnnTfLiteParser/test/GetBuffer.cpp b/src/armnnTfLiteParser/test/GetBuffer.cpp index cccdbce7aa..0e72522c79 100644 --- a/src/armnnTfLiteParser/test/GetBuffer.cpp +++ b/src/armnnTfLiteParser/test/GetBuffer.cpp @@ -8,7 +8,7 @@ #include "../TfLiteParser.hpp" #include <sstream> -using armnnTfLiteParser::TfLiteParser; +using armnnTfLiteParser::TfLiteParserImpl; BOOST_AUTO_TEST_SUITE(TensorflowLiteParser) @@ -88,12 +88,12 @@ struct GetBufferFixture : public ParserFlatbuffersFixture ReadStringToBinary(); } - void CheckBufferContents(const TfLiteParser::ModelPtr& model, + void CheckBufferContents(const TfLiteParserImpl::ModelPtr& model, std::vector<int32_t> bufferValues, size_t bufferIndex) { for(long unsigned int i=0; i<bufferValues.size(); i++) { - BOOST_CHECK_EQUAL(TfLiteParser::GetBuffer(model, bufferIndex)->data[i], bufferValues[i]); + BOOST_CHECK_EQUAL(TfLiteParserImpl::GetBuffer(model, bufferIndex)->data[i], bufferValues[i]); } } }; @@ -101,7 +101,8 @@ struct GetBufferFixture : public ParserFlatbuffersFixture BOOST_FIXTURE_TEST_CASE(GetBufferCheckContents, GetBufferFixture) { //Check contents of buffer are correct - TfLiteParser::ModelPtr model = TfLiteParser::LoadModelFromBinary(m_GraphBinary.data(), m_GraphBinary.size()); + TfLiteParserImpl::ModelPtr model = TfLiteParserImpl::LoadModelFromBinary(m_GraphBinary.data(), + m_GraphBinary.size()); std::vector<int32_t> bufferValues = {2,1,0,6,2,1,4,1,2}; CheckBufferContents(model, bufferValues, 2); } @@ -109,18 +110,20 @@ BOOST_FIXTURE_TEST_CASE(GetBufferCheckContents, GetBufferFixture) BOOST_FIXTURE_TEST_CASE(GetBufferCheckEmpty, GetBufferFixture) { //Check if test fixture buffers are empty or not - TfLiteParser::ModelPtr model = TfLiteParser::LoadModelFromBinary(m_GraphBinary.data(), m_GraphBinary.size()); - BOOST_CHECK(TfLiteParser::GetBuffer(model, 0)->data.empty()); - BOOST_CHECK(TfLiteParser::GetBuffer(model, 1)->data.empty()); - BOOST_CHECK(!TfLiteParser::GetBuffer(model, 2)->data.empty()); - BOOST_CHECK(TfLiteParser::GetBuffer(model, 3)->data.empty()); + TfLiteParserImpl::ModelPtr model = TfLiteParserImpl::LoadModelFromBinary(m_GraphBinary.data(), + m_GraphBinary.size()); + BOOST_CHECK(TfLiteParserImpl::GetBuffer(model, 0)->data.empty()); + BOOST_CHECK(TfLiteParserImpl::GetBuffer(model, 1)->data.empty()); + BOOST_CHECK(!TfLiteParserImpl::GetBuffer(model, 2)->data.empty()); + BOOST_CHECK(TfLiteParserImpl::GetBuffer(model, 3)->data.empty()); } BOOST_FIXTURE_TEST_CASE(GetBufferCheckParseException, GetBufferFixture) { //Check if armnn::ParseException thrown when invalid buffer index used - TfLiteParser::ModelPtr model = TfLiteParser::LoadModelFromBinary(m_GraphBinary.data(), m_GraphBinary.size()); - BOOST_CHECK_THROW(TfLiteParser::GetBuffer(model, 4), armnn::Exception); + TfLiteParserImpl::ModelPtr model = TfLiteParserImpl::LoadModelFromBinary(m_GraphBinary.data(), + m_GraphBinary.size()); + BOOST_CHECK_THROW(TfLiteParserImpl::GetBuffer(model, 4), armnn::Exception); } BOOST_AUTO_TEST_SUITE_END() |