IVGCVSW-2552 Add unit tests for the input and output layers for Visitor

Change-Id: I73ae9b43fa1631fdec423177d37fffc719c71f88
Signed-off-by: FrancisMurtagh <francis.murtagh@arm.com>

1 files changed, 53 insertions, 0 deletions

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