1 files changed, 47 insertions, 0 deletions
diff --git a/src/armnnCaffeParser/test/TestSplit.cpp b/src/armnnCaffeParser/test/TestSplit.cpp
new file mode 100644
index 0000000000..c2f29fb4f3
--- /dev/null
+++ b/src/armnnCaffeParser/test/TestSplit.cpp
@@ -0,0 +1,47 @@
+//
+// Copyright © 2017 Arm Ltd. All rights reserved.
+// See LICENSE file in the project root for full license information.
+//
+#include <boost/test/unit_test.hpp>
+#include "armnnCaffeParser/ICaffeParser.hpp"
+#include "ParserPrototxtFixture.hpp"
+
+BOOST_AUTO_TEST_SUITE(CaffeParser)
+
+struct SplitFixture : public ParserPrototxtFixture<armnnCaffeParser::ICaffeParser>
+{
+    SplitFixture()
+    {
+        m_Prototext = R"(
+name: "Split"
+layer {
+  name: "data"
+  type: "Input"
+  top: "data"
+  input_param { shape: { dim: 1 dim: 1 dim: 1 dim: 1 } }
+}
+layer {
+    name: "split"
+    type: "Split"
+    bottom: "data"
+    top: "split_top0"
+    top: "split_top1"
+}
+layer {
+    bottom: "split_top0"
+    bottom: "split_top1"
+    top: "add"
+    name: "add"
+    type: "Eltwise"
+}
+        )";
+        SetupSingleInputSingleOutput("data", "add");
+    }
+};
+
+BOOST_FIXTURE_TEST_CASE(Split, SplitFixture)
+{
+    RunTest<1>({ 1.0f }, { 2.0f });
+}
+
+BOOST_AUTO_TEST_SUITE_END()