From bfedee88b2cb4b0f8a90ebd89cbd10b6eac25d8a Mon Sep 17 00:00:00 2001
From: Anton Pankratv <anton.pankratov@intel.com>
Date: Sat, 15 Aug 2020 14:44:13 +0300
Subject: [PATCH] Added split test into template plugin (#1669)

* Added split test into template plugin

* Fixed split test

* Fixed test parameters style
---
 .../single_layer_tests/split.cpp              | 22 +++++++++++++++++++
 .../tests/functional/skip_tests_config.cpp    |  3 ++-
 .../shared/src/single_layer_tests/split.cpp   |  2 +-
 3 files changed, 25 insertions(+), 2 deletions(-)
 create mode 100644 docs/template_plugin/tests/functional/shared_tests_instances/single_layer_tests/split.cpp

diff --git a/docs/template_plugin/tests/functional/shared_tests_instances/single_layer_tests/split.cpp b/docs/template_plugin/tests/functional/shared_tests_instances/single_layer_tests/split.cpp
new file mode 100644
index 00000000000..704278d4dc5
--- /dev/null
+++ b/docs/template_plugin/tests/functional/shared_tests_instances/single_layer_tests/split.cpp
@@ -0,0 +1,22 @@
+// Copyright (C) 2020 Intel Corporation
+// SPDX-License-Identifier: Apache-2.0
+
+
+#include <vector>
+
+#include "single_layer_tests/split.hpp"
+#include "common_test_utils/test_constants.hpp"
+
+using namespace LayerTestsDefinitions;
+
+namespace {
+
+INSTANTIATE_TEST_CASE_P(NumSplitsCheck, SplitLayerTest,
+                        ::testing::Combine(
+                                ::testing::Values(1, 2, 3, 5, 6, 10, 30),
+                                ::testing::Values(0, 1, 2, 3),
+                                ::testing::Values(InferenceEngine::Precision::FP32),
+                                ::testing::Values(std::vector<size_t >({30, 30, 30, 30})),
+                                ::testing::Values("TEMPLATE")),
+                        SplitLayerTest::getTestCaseName);
+}  // namespace
diff --git a/docs/template_plugin/tests/functional/skip_tests_config.cpp b/docs/template_plugin/tests/functional/skip_tests_config.cpp
index 0a9765777ba..f44a5608797 100644
--- a/docs/template_plugin/tests/functional/skip_tests_config.cpp
+++ b/docs/template_plugin/tests/functional/skip_tests_config.cpp
@@ -12,6 +12,7 @@ std::vector<std::string> disabledTestPatterns() {
         ".*ExclusiveAsyncRequests.*",
         ".*reusableCPUStreamsExecutor.*",
         ".*registerPlugin.*",
-        ".*IEClassGetAvailableDevices.*"
+        ".*IEClassGetAvailableDevices.*",
+        R"(.*SplitLayerTest.*numSplits\=30.*)"
     };
 }
\ No newline at end of file
diff --git a/inference-engine/tests/functional/plugin/shared/src/single_layer_tests/split.cpp b/inference-engine/tests/functional/plugin/shared/src/single_layer_tests/split.cpp
index 2bbc49ecdf3..a244f2c3da5 100644
--- a/inference-engine/tests/functional/plugin/shared/src/single_layer_tests/split.cpp
+++ b/inference-engine/tests/functional/plugin/shared/src/single_layer_tests/split.cpp
@@ -51,7 +51,7 @@ void SplitLayerTest::SetUp() {
                                                                                              ngPrc, numSplits, axis));
     ngraph::ResultVector results;
     for (int i = 0; i < numSplits; i++) {
-        results.push_back(std::make_shared<ngraph::opset1::Result>(split));
+        results.push_back(std::make_shared<ngraph::opset1::Result>(split->output(i)));
     }
     function = std::make_shared<ngraph::Function>(results, params, "split");
 }