diff options
Diffstat (limited to 'src/armnnTfLiteParser/test/TfLiteParser.cpp')
-rw-r--r-- | src/armnnTfLiteParser/test/TfLiteParser.cpp | 41 |
1 files changed, 41 insertions, 0 deletions
diff --git a/src/armnnTfLiteParser/test/TfLiteParser.cpp b/src/armnnTfLiteParser/test/TfLiteParser.cpp new file mode 100644 index 0000000000..36827c0586 --- /dev/null +++ b/src/armnnTfLiteParser/test/TfLiteParser.cpp @@ -0,0 +1,41 @@ +// +// Copyright © 2020 Arm Ltd. All rights reserved. +// SPDX-License-Identifier: MIT +// + +#include <boost/test/unit_test.hpp> +#include "ParserFlatbuffersFixture.hpp" +#include "../TfLiteParser.hpp" + +BOOST_AUTO_TEST_SUITE(TensorflowLiteParser) + +BOOST_AUTO_TEST_CASE(ParseEmptyBinaryData) +{ + ITfLiteParser::TfLiteParserOptions options; + ITfLiteParserPtr m_Parser(ITfLiteParser::Create(armnn::Optional<ITfLiteParser::TfLiteParserOptions>(options))); + // Should throw armnn::ParseException: Buffer doesn't conform to the expected Tensorflow Lite flatbuffers format. + BOOST_CHECK_THROW(m_Parser->CreateNetworkFromBinary({0}), armnn::ParseException); +} + +struct NoInputBindingsFixture : public ParserFlatbuffersFixture +{ + explicit NoInputBindingsFixture() + { + m_JsonString = R"( + { + "version": 3, + "operator_codes": [ { "builtin_code": "CONV_2D" } ], + "subgraphs": [ { } ] + } + )"; + SetupSingleInputSingleOutput("inputTensor", "outputTensor"); + } +}; + +BOOST_FIXTURE_TEST_CASE( ParseBadInputBindings, NoInputBindingsFixture ) +{ + // Should throw armnn::ParseException: No input binding found for subgraph:0 and name:inputTensor. + BOOST_CHECK_THROW( (RunTest<4, armnn::DataType::QAsymmU8>(0, { }, { 0 })), armnn::ParseException); +} + +BOOST_AUTO_TEST_SUITE_END() |