GetBuffer.cpp File Reference

#include <boost/test/unit_test.hpp>
#include "ParserFlatbuffersFixture.hpp"
#include "../TfLiteParser.hpp"
#include <sstream>

Go to the source code of this file.

Functions
	BOOST_FIXTURE_TEST_CASE (GetBufferCheckContents, GetBufferFixture)
	BOOST_FIXTURE_TEST_CASE (GetBufferCheckEmpty, GetBufferFixture)
	BOOST_FIXTURE_TEST_CASE (GetBufferCheckParseException, GetBufferFixture)

Function Documentation

BOOST_FIXTURE_TEST_CASE() [1/3]

BOOST_FIXTURE_TEST_CASE	(	GetBufferCheckContents	,
		GetBufferFixture
	)

Definition at line 101 of file GetBuffer.cpp.

{
    //Check contents of buffer are correct
    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);
}

BOOST_FIXTURE_TEST_CASE() [2/3]

BOOST_FIXTURE_TEST_CASE	(	GetBufferCheckEmpty	,
		GetBufferFixture
	)

Definition at line 110 of file GetBuffer.cpp.

{
    //Check if test fixture buffers are empty or not
    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() [3/3]

BOOST_FIXTURE_TEST_CASE	(	GetBufferCheckParseException	,
		GetBufferFixture
	)

Definition at line 121 of file GetBuffer.cpp.

References BOOST_AUTO_TEST_SUITE_END().

{
    //Check if armnn::ParseException thrown when invalid buffer index used
    TfLiteParserImpl::ModelPtr model = TfLiteParserImpl::LoadModelFromBinary(m_GraphBinary.data(),
                                                                             m_GraphBinary.size());
    BOOST_CHECK_THROW(TfLiteParserImpl::GetBuffer(model, 4), armnn::Exception);
}