1 // Copyright (C) 2018-2019 Intel Corporation
2 // SPDX-License-Identifier: Apache-2.0
5 #include <gtest/gtest.h>
7 #include <ie_builders.hpp>
8 #include <builders/ie_const_layer.hpp>
10 #include "builder_test.hpp"
12 using namespace testing;
13 using namespace InferenceEngine;
15 class ConstLayerBuilderTest : public BuilderTestCommon {};
17 TEST_F(ConstLayerBuilderTest, getExistsLayerFromNetworkBuilder) {
18 Builder::Network net("network");
19 Builder::ConstLayer layer("const layer");
20 layer.setData(generateBlob(Precision::FP32, {3}, Layout::C));
21 const size_t ind = net.addLayer(layer);
22 ASSERT_NO_THROW(net.getLayer(ind)->validate(false));
25 TEST_F(ConstLayerBuilderTest, cannotCreateLayerWithoutData) {
26 Builder::Network net("network");
27 Builder::ConstLayer layer("const layer");
28 ASSERT_THROW(net.addLayer(layer),
29 InferenceEngine::details::InferenceEngineException);