InputOutputTensorNames.cpp File Reference

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

Go to the source code of this file.

Functions
	BOOST_FIXTURE_TEST_CASE (EmptyNetworkHasNoInputsAndOutputs, EmptyNetworkFixture)
	BOOST_FIXTURE_TEST_CASE (MissingTensorsThrowException, MissingTensorsFixture)
	BOOST_FIXTURE_TEST_CASE (InvalidTensorsThrowException, InvalidTensorsFixture)
	BOOST_FIXTURE_TEST_CASE (GetValidInputOutputTensorNames, ValidTensorsFixture)
	BOOST_FIXTURE_TEST_CASE (ThrowIfSubgraphIdInvalidForInOutNames, ValidTensorsFixture)
	BOOST_FIXTURE_TEST_CASE (Rank0Tensor, Rank0TensorFixture)

Function Documentation

BOOST_FIXTURE_TEST_CASE() [1/6]

BOOST_FIXTURE_TEST_CASE	(	EmptyNetworkHasNoInputsAndOutputs	,
		EmptyNetworkFixture
	)

Definition at line 24 of file InputOutputTensorNames.cpp.

References ParserFlatbuffersFixture::Setup().

{
    Setup();
    BOOST_TEST(m_Parser->GetSubgraphCount() == 1);
    BOOST_TEST(m_Parser->GetSubgraphInputTensorNames(0).size() == 0);
    BOOST_TEST(m_Parser->GetSubgraphOutputTensorNames(0).size() == 0);
}

BOOST_FIXTURE_TEST_CASE() [2/6]

BOOST_FIXTURE_TEST_CASE	(	MissingTensorsThrowException	,
		MissingTensorsFixture
	)

Definition at line 48 of file InputOutputTensorNames.cpp.

{
    // this throws because it cannot do the input output tensor connections
    BOOST_CHECK_THROW(Setup(), armnn::ParseException);
}

BOOST_FIXTURE_TEST_CASE() [3/6]

BOOST_FIXTURE_TEST_CASE	(	InvalidTensorsThrowException	,
		InvalidTensorsFixture
	)

Definition at line 81 of file InputOutputTensorNames.cpp.

References armnn::MaxNumOfTensorDimensions.

{
    // Tensor numDimensions must be less than or equal to MaxNumOfTensorDimensions
    static_assert(armnn::MaxNumOfTensorDimensions == 5, "Please update InvalidTensorsFixture");
    BOOST_CHECK_THROW(Setup(), armnn::InvalidArgumentException);
}

BOOST_FIXTURE_TEST_CASE() [4/6]

BOOST_FIXTURE_TEST_CASE	(	GetValidInputOutputTensorNames	,
		ValidTensorsFixture
	)

Definition at line 131 of file InputOutputTensorNames.cpp.

{
    Setup();
    BOOST_CHECK_EQUAL(m_Parser->GetSubgraphInputTensorNames(0).size(), 1u);
    BOOST_CHECK_EQUAL(m_Parser->GetSubgraphOutputTensorNames(0).size(), 1u);
    BOOST_CHECK_EQUAL(m_Parser->GetSubgraphInputTensorNames(0)[0], "In");
    BOOST_CHECK_EQUAL(m_Parser->GetSubgraphOutputTensorNames(0)[0], "Out");
}

BOOST_FIXTURE_TEST_CASE() [5/6]

BOOST_FIXTURE_TEST_CASE	(	ThrowIfSubgraphIdInvalidForInOutNames	,
		ValidTensorsFixture
	)

Definition at line 140 of file InputOutputTensorNames.cpp.

{
    Setup();

    // these throw because of the invalid subgraph id
    BOOST_CHECK_THROW(m_Parser->GetSubgraphInputTensorNames(1), armnn::ParseException);
    BOOST_CHECK_THROW(m_Parser->GetSubgraphOutputTensorNames(1), armnn::ParseException);
}

BOOST_FIXTURE_TEST_CASE() [6/6]

BOOST_FIXTURE_TEST_CASE	(	Rank0Tensor	,
		Rank0TensorFixture
	)

Definition at line 188 of file InputOutputTensorNames.cpp.

References BOOST_AUTO_TEST_SUITE_END().

{
    Setup();
    BOOST_CHECK_EQUAL(m_Parser->GetSubgraphInputTensorNames(0).size(), 2u);
    BOOST_CHECK_EQUAL(m_Parser->GetSubgraphOutputTensorNames(0).size(), 1u);
    BOOST_CHECK_EQUAL(m_Parser->GetSubgraphInputTensorNames(0)[0], "In0");
    BOOST_CHECK_EQUAL(m_Parser->GetSubgraphInputTensorNames(0)[1], "In1");
    BOOST_CHECK_EQUAL(m_Parser->GetSubgraphOutputTensorNames(0)[0], "Out");
}