inference-engine/tests/unit/builders/output_layer_test.cpp

   1 // Copyright (C) 2019 Intel Corporation
   2 // SPDX-License-Identifier: Apache-2.0
   3 //
   4
   5 #include <gtest/gtest.h>
   6 #include <string.h>
   7 #include <ie_builders.hpp>
   8 #include <builders/ie_output_layer.hpp>
   9
  10 #include "builder_test.hpp"
  11
  12 using namespace testing;
  13 using namespace InferenceEngine;
  14
  15 class OutputLayerBuilderTest : public BuilderTestCommon {};
  16
  17 TEST_F(OutputLayerBuilderTest, getExistsLayerFromNetworkBuilder) {
  18     Builder::Network network("network");
  19     Builder::OutputLayer layer("output layer");
  20     layer.setPort(Port({1, 1, 1, 1}));
  21     size_t ind = network.addLayer(layer);
  22     Builder::OutputLayer layerFromNet(network.getLayer(ind));
  23     ASSERT_EQ(layer.getPort().shape(), layerFromNet.getPort().shape());
  24     ASSERT_EQ(layer.getPort().shape(), Port({1, 1, 1, 1}).shape());
  25 }