From 531ef94a0600665d723e64e7a47e01fcc0cadfc5 Mon Sep 17 00:00:00 2001 From: FrancisMurtagh Date: Fri, 1 Feb 2019 11:48:58 +0000 Subject: IVGCVSW-2552 Add unit tests for the input and output layers for Visitor Change-Id: I73ae9b43fa1631fdec423177d37fffc719c71f88 Signed-off-by: FrancisMurtagh --- src/armnn/test/TestInputOutputLayerVisitor.cpp | 53 ++++++++++++++++++++++++++ 1 file changed, 53 insertions(+) create mode 100644 src/armnn/test/TestInputOutputLayerVisitor.cpp (limited to 'src/armnn/test/TestInputOutputLayerVisitor.cpp') diff --git a/src/armnn/test/TestInputOutputLayerVisitor.cpp b/src/armnn/test/TestInputOutputLayerVisitor.cpp new file mode 100644 index 0000000000..35ffc55e55 --- /dev/null +++ b/src/armnn/test/TestInputOutputLayerVisitor.cpp @@ -0,0 +1,53 @@ +// +// Copyright © 2017 Arm Ltd. All rights reserved. +// SPDX-License-Identifier: MIT +// +#include "TestInputOutputLayerVisitor.hpp" +#include "Network.hpp" + +namespace armnn +{ + +BOOST_AUTO_TEST_SUITE(TestInputOutputLayerVisitor) + +BOOST_AUTO_TEST_CASE(CheckInputLayerVisitorBindingIdAndName) +{ + const char* layerName = "InputLayer"; + TestInputLayerVisitor visitor(1, layerName); + Network net; + + IConnectableLayer *const layer = net.AddInputLayer(1, layerName); + layer->Accept(visitor); +} + +BOOST_AUTO_TEST_CASE(CheckInputLayerVisitorBindingIdAndNameNull) +{ + TestInputLayerVisitor visitor(1); + Network net; + + IConnectableLayer *const layer = net.AddInputLayer(1); + layer->Accept(visitor); +} + +BOOST_AUTO_TEST_CASE(CheckOutputLayerVisitorBindingIdAndName) +{ + const char* layerName = "OutputLayer"; + TestOutputLayerVisitor visitor(1, layerName); + Network net; + + IConnectableLayer *const layer = net.AddOutputLayer(1, layerName); + layer->Accept(visitor); +} + +BOOST_AUTO_TEST_CASE(CheckOutputLayerVisitorBindingIdAndNameNull) +{ + TestOutputLayerVisitor visitor(1); + Network net; + + IConnectableLayer *const layer = net.AddOutputLayer(1); + layer->Accept(visitor); +} + +BOOST_AUTO_TEST_SUITE_END() + +} //namespace armnn \ No newline at end of file -- cgit v1.2.1